oj.uz

답안 #780271

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
780271 2023-07-12T07:50:12 Z boris_mihov Maze (JOI23_ho_t3) C++17
0 / 100
 2000 ms 611100 KB 
#include <algorithm>
#include <iostream>
#include <numeric>
#include <cassert>
#include <vector>
#include <queue>

typedef long long llong;
const int MAXLOG = 23;
const int MAXN = 6000000 + 10;
const int INF  = 1e9;

struct BIT
{
    std::vector <int> tree;
    void build(int sz)
    {
        tree.resize(sz + 1, 0);
    }

    void update(int pos, int value)
    {
        for (int idx = pos ; idx < tree.size() ; idx += idx & (-idx))
        {
            tree[idx] += value;
        }
    }

    int query(int pos)
    {
        int res = 0;
        for (int idx = pos ; idx > 0 ; idx -= idx & (-idx))
        {
            res += tree[idx];
        }

        return pos - res;
    }

    int findKth(int k)
    {
        int idx = 0;
        for (int log = MAXLOG - 1 ; log >= 0 ; --log)
        {
            if (idx + (1 << log) < tree.size() && (1 << log) - tree[idx + (1 << log)] < k)
            {
                idx += (1 << log);
                k -= (1 << log) - tree[idx];
            }
        }

        return idx + 1;
    }
};

int r, c, n;
int sRow, sCol;
int eRow, eCol;
BIT byROW[MAXN];
BIT byCOL[MAXN];
std::pair <int,int> delta[] = {{-1, 0}, {0, -1}, {1, 0}, {0, 1}};
std::deque <std::pair <int,int>> dq;
std::vector <int> dist[MAXN];
std::string t[MAXN];

bool isOutside(int row, int col)
{
    return row == 0 || row == r + 1 || col == 0 || col == c + 1;
}

void setCELL(int row, int col)
{
    byROW[row].update(col, 1); 
    byCOL[col].update(row, 1); 
}

void addROW(int row, int colL, int colR, int currDist)
{
    int curr = byROW[row].query(colL - 1);
    // while (true)
    // {
    //     int search = byROW[row].findKth(curr + 1);
    //     if (search > colR)
    //     {
    //         break;
    //     }

    //     if (dist[row][search] > currDist + 1) dist[row][search] = currDist + 1;
    //     dq.push_back({row, search});
    //     setCELL(row, search);
    // }

    for (int i = colL ; i <= colR ; ++i)
    {
        if (dist[row][i] == INF)
        {
            if (byROW[row].findKth(curr + 1) != i)
            {
                int res = byROW[row].findKth(curr + 1); assert(res != i);
                // assert(byROW[row].query(res) - byROW[row].query(res - 1) == 1);
                while (res < i){}
                while (res > i){}
                while (true) assert(res != i);
            }

            setCELL(row, i);
            dist[row][i] = currDist + 1;
            dq.push_back({row, i});
        }
    }
}

void addCOL(int col, int rowL, int rowR, int currDist)
{
    int curr = byCOL[col].query(rowL - 1);
    // while (true)
    // {
    //     int search = byCOL[col].findKth(curr + 1);
    //     if (search > rowR)
    //     {
    //         break;
    //     }

    //     if (dist[search][col] > currDist + 1) dist[search][col] = currDist + 1;
    //     dq.push_back({search, col});
    //     setCELL(search, col);
    // }

    for (int i = rowL ; i <= rowR ; ++i)
    {
        if (dist[i][col] == INF)
        {
            if (byCOL[col].findKth(curr + 1) != i)
            {
                int res = byCOL[col].findKth(curr + 1); assert(res != i);
                // assert(byCOL[col].query(res) - byCOL[col].query(res - 1) == 1);
                while (res < i){}
                while (res > i){}
                while (true) assert(res != i);
            }

            setCELL(i, col);
            dist[i][col] = currDist + 1;
            dq.push_back({i, col});
        }
    }
}

void solve()
{
    for (int i = 1 ; i <= r ; ++i)
    {
        std::fill(dist[i].begin(), dist[i].end(), INF);
    }

    for (int i = 1 ; i <= r ; ++i)
    {
        byROW[i].build(c);
    }
    
    for (int i = 1 ; i <= c ; ++i)
    {
        byCOL[i].build(r);
    }
    
    dq.push_back({sRow, sCol});
    dist[sRow][sCol] = 0;
    setCELL(sRow, sCol);

    while (!dq.empty())
    {
        auto [row, col] = dq.front();
        dq.pop_front();    

        if (row == eRow && col == eCol)
        {
            std::cout << dist[row][col] << '\n';
            break;
        }

        for (const auto &[dx, dy] : delta)
        {
            if (isOutside(row + dx, col + dy) || dist[row + dx][col + dy] <= dist[row][col])
            {
                continue;
            }

            if (t[row + dx][col + dy] == '.')
            {
                setCELL(row + dx, col + dy);
                dist[row + dx][col + dy] = dist[row][col]; 
                dq.push_front({row + dx, col + dy});
            }
        }

        if (abs(row - eRow) <= n && abs(col - eCol) <= n && dist[eRow][eCol] > dist[row][col])
        {
            setCELL(eRow, eCol);
            dist[eRow][eCol] = dist[row][col] + 1;
            dq.push_back({eRow, eCol});
        }

        addROW(std::max(1, row - n), std::max(1, col - n + 1), std::min(c, col + n - 1), dist[row][col]);
        addROW(std::min(r, row + n), std::max(1, col - n + 1), std::min(c, col + n - 1), dist[row][col]);
        addCOL(std::max(1, col - n), std::max(1, row - n + 1), std::min(r, row + n - 1), dist[row][col]);
        addCOL(std::min(c, col + n), std::max(1, row - n + 1), std::min(r, row + n - 1), dist[row][col]);
    }
}

void input()
{
    std::cin >> r >> c >> n;
    std::cin >> sRow >> sCol;
    std::cin >> eRow >> eCol;
    for (int i = 1 ; i <= r ; ++i)
    {
        std::cin >> t[i];
        t[i] = ' ' + t[i];
        dist[i].resize(c + 1, INF);
    }
}

void fastIOI()
{
    std::ios_base :: sync_with_stdio(0);
    std::cout.tie(nullptr);
    std::cin.tie(nullptr);
}

int main()
{
    fastIOI();
    input();
    solve();

    return 0;
}

Compilation message

Main.cpp: In member function 'void BIT::update(int, int)':
Main.cpp:23:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 |         for (int idx = pos ; idx < tree.size() ; idx += idx & (-idx))
      |                              ~~~~^~~~~~~~~~~~~
Main.cpp: In member function 'int BIT::findKth(int)':
Main.cpp:45:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |             if (idx + (1 << log) < tree.size() && (1 << log) - tree[idx + (1 << log)] < k)
      |                 ~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~

Subtask #1
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 260 ms 610864 KB Output is correct
2 Correct 241 ms 610828 KB Output is correct
3 Correct 246 ms 610848 KB Output is correct
4 Correct 243 ms 610764 KB Output is correct
5 Correct 245 ms 610780 KB Output is correct
6 Correct 245 ms 610828 KB Output is correct
7 Correct 257 ms 610876 KB Output is correct
8 Correct 243 ms 610892 KB Output is correct
9 Correct 248 ms 610748 KB Output is correct
10 Correct 244 ms 610992 KB Output is correct
11 Execution timed out 2093 ms 610788 KB Time limit exceeded
12 Halted 0 ms 0 KB -

Subtask #2
0 / 19.0

# 결과 실행 시간 메모리 Grader output
1 Correct 246 ms 610864 KB Output is correct
2 Correct 249 ms 610812 KB Output is correct
3 Correct 240 ms 610760 KB Output is correct
4 Correct 243 ms 610768 KB Output is correct
5 Correct 251 ms 610844 KB Output is correct
6 Correct 244 ms 610868 KB Output is correct
7 Correct 247 ms 610764 KB Output is correct
8 Correct 251 ms 611100 KB Output is correct
9 Correct 255 ms 610976 KB Output is correct
10 Correct 266 ms 610880 KB Output is correct
11 Correct 247 ms 610824 KB Output is correct
12 Correct 302 ms 610880 KB Output is correct
13 Correct 248 ms 610880 KB Output is correct
14 Correct 249 ms 610776 KB Output is correct
15 Correct 248 ms 610816 KB Output is correct
16 Correct 255 ms 611008 KB Output is correct
17 Correct 247 ms 610772 KB Output is correct
18 Correct 245 ms 610792 KB Output is correct
19 Execution timed out 2106 ms 610972 KB Time limit exceeded
20 Halted 0 ms 0 KB -

Subtask #3
0 / 16.0

# 결과 실행 시간 메모리 Grader output
1 Correct 258 ms 610768 KB Output is correct
2 Correct 250 ms 610928 KB Output is correct
3 Correct 252 ms 610860 KB Output is correct
4 Correct 246 ms 610856 KB Output is correct
5 Correct 249 ms 610820 KB Output is correct
6 Correct 257 ms 610836 KB Output is correct
7 Correct 250 ms 610828 KB Output is correct
8 Correct 282 ms 610760 KB Output is correct
9 Correct 250 ms 610924 KB Output is correct
10 Correct 250 ms 610816 KB Output is correct
11 Correct 249 ms 610780 KB Output is correct
12 Correct 261 ms 610864 KB Output is correct
13 Correct 247 ms 610760 KB Output is correct
14 Execution timed out 2102 ms 610764 KB Time limit exceeded
15 Halted 0 ms 0 KB -

Subtask #4
0 / 19.0

# 결과 실행 시간 메모리 Grader output
1 Correct 246 ms 610864 KB Output is correct
2 Correct 249 ms 610812 KB Output is correct
3 Correct 240 ms 610760 KB Output is correct
4 Correct 243 ms 610768 KB Output is correct
5 Correct 251 ms 610844 KB Output is correct
6 Correct 244 ms 610868 KB Output is correct
7 Correct 247 ms 610764 KB Output is correct
8 Correct 251 ms 611100 KB Output is correct
9 Correct 255 ms 610976 KB Output is correct
10 Correct 266 ms 610880 KB Output is correct
11 Correct 247 ms 610824 KB Output is correct
12 Correct 302 ms 610880 KB Output is correct
13 Correct 248 ms 610880 KB Output is correct
14 Correct 249 ms 610776 KB Output is correct
15 Correct 248 ms 610816 KB Output is correct
16 Correct 255 ms 611008 KB Output is correct
17 Correct 247 ms 610772 KB Output is correct
18 Correct 245 ms 610792 KB Output is correct
19 Execution timed out 2106 ms 610972 KB Time limit exceeded
20 Halted 0 ms 0 KB -

Subtask #5
0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 246 ms 610864 KB Output is correct
2 Correct 249 ms 610812 KB Output is correct
3 Correct 240 ms 610760 KB Output is correct
4 Correct 243 ms 610768 KB Output is correct
5 Correct 251 ms 610844 KB Output is correct
6 Correct 244 ms 610868 KB Output is correct
7 Correct 247 ms 610764 KB Output is correct
8 Correct 251 ms 611100 KB Output is correct
9 Correct 255 ms 610976 KB Output is correct
10 Correct 266 ms 610880 KB Output is correct
11 Correct 247 ms 610824 KB Output is correct
12 Correct 302 ms 610880 KB Output is correct
13 Correct 248 ms 610880 KB Output is correct
14 Correct 249 ms 610776 KB Output is correct
15 Correct 248 ms 610816 KB Output is correct
16 Correct 255 ms 611008 KB Output is correct
17 Correct 247 ms 610772 KB Output is correct
18 Correct 245 ms 610792 KB Output is correct
19 Execution timed out 2106 ms 610972 KB Time limit exceeded
20 Halted 0 ms 0 KB -

Subtask #6
0 / 19.0

# 결과 실행 시간 메모리 Grader output
1 Correct 260 ms 610864 KB Output is correct
2 Correct 241 ms 610828 KB Output is correct
3 Correct 246 ms 610848 KB Output is correct
4 Correct 243 ms 610764 KB Output is correct
5 Correct 245 ms 610780 KB Output is correct
6 Correct 245 ms 610828 KB Output is correct
7 Correct 257 ms 610876 KB Output is correct
8 Correct 243 ms 610892 KB Output is correct
9 Correct 248 ms 610748 KB Output is correct
10 Correct 244 ms 610992 KB Output is correct
11 Execution timed out 2093 ms 610788 KB Time limit exceeded
12 Halted 0 ms 0 KB -

Subtask #7
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 260 ms 610864 KB Output is correct
2 Correct 241 ms 610828 KB Output is correct
3 Correct 246 ms 610848 KB Output is correct
4 Correct 243 ms 610764 KB Output is correct
5 Correct 245 ms 610780 KB Output is correct
6 Correct 245 ms 610828 KB Output is correct
7 Correct 257 ms 610876 KB Output is correct
8 Correct 243 ms 610892 KB Output is correct
9 Correct 248 ms 610748 KB Output is correct
10 Correct 244 ms 610992 KB Output is correct
11 Execution timed out 2093 ms 610788 KB Time limit exceeded
12 Halted 0 ms 0 KB -

Subtask #8
0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 260 ms 610864 KB Output is correct
2 Correct 241 ms 610828 KB Output is correct
3 Correct 246 ms 610848 KB Output is correct
4 Correct 243 ms 610764 KB Output is correct
5 Correct 245 ms 610780 KB Output is correct
6 Correct 245 ms 610828 KB Output is correct
7 Correct 257 ms 610876 KB Output is correct
8 Correct 243 ms 610892 KB Output is correct
9 Correct 248 ms 610748 KB Output is correct
10 Correct 244 ms 610992 KB Output is correct
11 Execution timed out 2093 ms 610788 KB Time limit exceeded
12 Halted 0 ms 0 KB -