oj.uz

Submission #445886

# Submission time Handle Problem Language Result Execution time Memory
445886 2021-07-20T05:16:04 Z blue Flood (IOI07_flood) C++17
64 / 100
 587 ms 48336 KB 
#include <iostream>
#include <vector>
#include <deque>
#include <algorithm>
#include <bitset>
using namespace std;
 
const int maxN = 100'000;
const int maxW = 200'000;
vector<int> X(1+maxN), Y(1+maxN);
vector<int> A(1+maxW), B(1+maxW);
 
int N;
int W;
 
// vector<bool> T(1+maxN, 0), L(1+maxN, 0), D(1+maxN, 0), R(1+maxN, 0);
bitset<1+maxN> T;
bitset<1+maxN> L;
bitset<1+maxN> D;
bitset<1+maxN> R;
 
const int maxR = 4*maxN + 4;
vector< vector<int> > R_edge(1+maxR, vector<int>(0));
vector< vector<short> > R_wt(1+maxR, vector<short>(0));
 
void add_R_edge(int u, int v, int w)
{
    R_edge[u].push_back(v);
    R_wt[u].push_back(w);
 
    R_edge[v].push_back(u);
    R_wt[v].push_back(w);
}
 
/*
    0   1
      P
    3   2
 
    4*i + 0
    4*i + 1
    4*i + 2
    4*i + 3
*/
 
const int TL = 0;
const int TR = 1;
const int DL = 3;
const int DR = 2;
 
const int INF = 1e9;
 
int main()
{
    // bitset<maxN> H;
    // H.set(0, 1);
    // H.set(1, 1);
    // cout << H[0] + H[1] << '\n';
 
    cin >> N;
    for(int i = 1; i <= N; i++)
        cin >> X[i] >> Y[i];
 
 
    cin >> W;
    for(int j = 1; j <= W; j++)
    {
        cin >> A[j] >> B[j];
 
        if(X[ A[j] ] == X[ B[j] ])
        {
            if(Y[ A[j] ] > Y[ B[j] ])
                swap(A[j], B[j]);
 
            // D[ A[j] ] = 1;
            D.set(A[j], 1);
            // T[ B[j] ] = 1;
            T.set(B[j], 1);
 
            add_R_edge(4*A[j] + DL, 4*B[j] + TL, 0);
            add_R_edge(4*A[j] + DR, 4*B[j] + TR, 0);
        }
        else //if(Y[ A[j] ] == Y[ B[j] ])
        {
            if(X[ A[j] ] > X[ B[j] ])
                swap(A[j], B[j]);
 
            // R[ A[j] ] = 1;
            // L[ B[j] ] = 1;
            R.set(A[j], 1);
            L.set(B[j], 1);
 
            add_R_edge(4*A[j] + TR, 4*B[j] + TL, 0);
            add_R_edge(4*A[j] + DR, 4*B[j] + DL, 0);
        }
    }
 
 
 
 
 
 
 
    for(int i = 1; i <= N; i++)
    {
        add_R_edge(4*i + TL, 4*i + TR, T[i]);
        add_R_edge(4*i + TL, 4*i + DL, L[i]);
        add_R_edge(4*i + TR, 4*i + DR, R[i]);
        add_R_edge(4*i + DL, 4*i + DR, D[i]);
    }
 
    // cerr << "check\n";
 
 
 
 
 
    vector<unsigned short> outside_dist(1+maxR, 65000);
    deque<int> tbv;
 
    vector<int> O(N);
    for(int i = 0; i < N; i++) O[i] = i+1;
 
    sort(O.begin(), O.end(), [] (int a1, int a2)
    {
        return X[a1] < X[a2];
    });
 
 
    for(int U: O)
    {
        // cerr << "U = " << U << '\n';
        if(outside_dist[4*U + TL] != 65000) continue;
        outside_dist[4*U + TL] = 0;
 
        tbv.push_front(4*U + TL);
 
        while(!tbv.empty())
        {
            int u = tbv.front();
            tbv.pop_front();
 
            for(int e = 0; e < R_edge[u].size(); e++)
            {
                int v = R_edge[u][e];
                int w = R_wt[u][e];
 
                if(outside_dist[v] <= outside_dist[u] + w) continue;
 
                outside_dist[v] = outside_dist[u] + w;
 
                if(w == 0) tbv.push_front(v);
                else tbv.push_back(v);
            }
        }
    }
 
    // cerr << "check\n";
 
 
    vector<int> res;
 
    for(int j = 1; j <= W; j++)
    {
        if(X[ A[j] ] == X[ B[j] ])
        {
            if(outside_dist[4*A[j] + DL] == outside_dist[4*A[j] + DR])
            {
                res.push_back(j);
            }
        }
        else //if(Y[ A[j] ] == Y[ B[j] ])
        {
            if(outside_dist[4*A[j] + TR] == outside_dist[4*A[j] + DR])
            {
                res.push_back(j);
            }
        }
    }
 
    cout << res.size() << '\n';
    for(int r:res) cout << r << '\n';
}

Compilation message

flood.cpp: In function 'int main()':
flood.cpp:143:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  143 |             for(int e = 0; e < R_edge[u].size(); e++)
      |                            ~~^~~~~~~~~~~~~~~~~~

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 22240 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 22220 KB Output is correct
2 Correct 14 ms 22220 KB Output is correct
3 Correct 15 ms 22216 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 15 ms 22160 KB Output is correct

Subtask #4
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 22220 KB Output is correct
2 Correct 15 ms 22352 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 22216 KB Output is correct
2 Correct 14 ms 22216 KB Output is correct

Subtask #6
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 22272 KB Output is correct

Subtask #7
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 15 ms 22348 KB Output is correct

Subtask #8
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 15 ms 22220 KB Output is correct
2 Correct 16 ms 22336 KB Output is correct

Subtask #9
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 22360 KB Output is correct
2 Correct 15 ms 22348 KB Output is correct

Subtask #10
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 77 ms 28740 KB Output is correct

Subtask #11
0 / 9.0

# Verdict Execution time Memory Grader output
1 Runtime error 237 ms 44204 KB Memory limit exceeded

Subtask #12
0 / 9.0

# Verdict Execution time Memory Grader output
1 Runtime error 253 ms 45376 KB Memory limit exceeded
2 Halted 0 ms 0 KB -

Subtask #13
0 / 9.0

# Verdict Execution time Memory Grader output
1 Runtime error 587 ms 48208 KB Memory limit exceeded
2 Halted 0 ms 0 KB -

Subtask #14
0 / 9.0

# Verdict Execution time Memory Grader output
1 Runtime error 583 ms 48336 KB Memory limit exceeded
2 Halted 0 ms 0 KB -