Submission #968478

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
968478	2024-04-23T13:07:05 Z	hirayuu_oj	Flood (IOI07_flood)	C++17	64 / 100	75 ms	12092 KB

flood

#include<bits/stdc++.h>
using namespace std;
#define rep(i,n) for(int i=0; i<n; i++)
#define all(x) x.begin(),x.end()
using ll=long long;

int dir(pair<int,int> x,pair<int,int> y){
    if(x.first==y.first){
        if(x.second<y.second)return 0;
        return 2;
    }
    if(x.first<y.first)return 1;
    return 3;
}
int wall_ind(array<pair<int,int>,4> wall){
    int cnt=0;
    int pos=-1;
    rep(i,4){
        if(wall[i].first!=-1){
            cnt++;
            pos=i;
        }
    }
    if(cnt==1)return pos;
    return -1;
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N;
    cin>>N;
    vector<pair<int,int>> pnt(N);
    rep(i,N){
        cin>>pnt[i].first>>pnt[i].second;
    }
    vector<array<pair<int,int>,4>> gr(N,{pair(-1,-1),pair(-1,-1),pair(-1,-1),pair(-1,-1)});
    int W;
    cin>>W;
    vector<int> ok(W,0);
    vector<pair<int,int>> wall(W);
    rep(i,W){
        cin>>wall[i].first>>wall[i].second;
        wall[i].first--;
        wall[i].second--;
        gr[wall[i].first][dir(pnt[wall[i].first],pnt[wall[i].second])]=pair(i,wall[i].second);
        gr[wall[i].second][dir(pnt[wall[i].second],pnt[wall[i].first])]=pair(i,wall[i].first);
    }
    vector<int> ans;
    stack<int> del;
    rep(i,N){
        if(wall_ind(gr[i])!=-1){
            del.push(i);
        }
    }
    while(!del.empty()){
        int pos=del.top();
        del.pop();
        if(wall_ind(gr[pos])==-1)continue;
        int dir=wall_ind(gr[pos]);
        ok[gr[pos][dir].first]=1;
        ans.push_back(gr[pos][dir].first);
        gr[gr[pos][dir].second][(dir+2)%4]={-1,-1};
        if(wall_ind(gr[gr[pos][dir].second])!=-1){
            del.push(gr[pos][dir].second);
        }
        gr[pos][dir]={-1,-1};
    }
    vector<array<int,4>> vert(N,{-1,-1,-1,-1});
    int cnt=0;
    rep(i,N){
        rep(j,4){
            if(gr[i][j].first==-1)continue;
            if(vert[i][j]!=-1)continue;
            int pos=i;
            int dir=j;
            while(vert[pos][dir]==-1){
                vert[pos][dir]=cnt;
                pos=gr[pos][dir].second;
                dir=(dir+3)%4;
                while(gr[pos][dir].first==-1){
                    dir++;
                    dir%=4;
                }
            }
            cnt++;
        }
    }
    rep(i,N){
        if(wall_ind(gr[i])!=-1){
            assert(false);
        }
    }
    vector<vector<int>> zag(cnt);
    rep(i,W){
        if(ok[i])continue;
        int s=vert[wall[i].first][dir(pnt[wall[i].first],pnt[wall[i].second])];
        int t=vert[wall[i].second][dir(pnt[wall[i].second],pnt[wall[i].first])];
        zag[s].push_back(t);
        zag[t].push_back(s);
    }
    vector<pair<int,int>> order;
    rep(i,W){
        if(ok[i])continue;
        if(dir(pnt[wall[i].first],pnt[wall[i].second])==3){
            order.push_back({pnt[wall[i].first].second,vert[wall[i].first][3]});
        }
        if(dir(pnt[wall[i].second],pnt[wall[i].first])==3){
            order.push_back({pnt[wall[i].second].second,vert[wall[i].second][3]});
        }
    }
    sort(all(order));
    vector<int> dist(cnt,-1);
    for(auto &[_,p]:order){
        if(dist[p]!=-1)continue;
        dist[p]=0;
        stack<int> pos;
        pos.push(p);
        while(!pos.empty()){
            int np=pos.top();
            pos.pop();
            for(int i:zag[np]){
                if(dist[i]==-1){
                    dist[i]=dist[np]+1;
                    pos.push(i);
                }
            }
        }
    }
    rep(i,W){
        if(ok[i])continue;
        int s=vert[wall[i].first][dir(pnt[wall[i].first],pnt[wall[i].second])];
        int t=vert[wall[i].second][dir(pnt[wall[i].second],pnt[wall[i].first])];
        if(dist[s]==dist[t]){
            ans.push_back(i);
        }
    }
    cout<<ans.size()<<"\n";
    for(int i:ans)cout<<i+1<<"\n";
}

Subtask #1
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct

Subtask #2
10.0 / 10.0

#	Verdict	Memory	Grader output
1	Correct	348 KB	Output is correct
2	Correct	344 KB	Output is correct
3	Correct	348 KB	Output is correct

Subtask #3
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct

Subtask #4
0 / 3.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Incorrect	1 ms	348 KB	Output isn't correct
3	Halted	0 ms	0 KB	-

Subtask #5
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct

Subtask #6
3.0 / 3.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	344 KB	Output is correct

Subtask #7
3.0 / 3.0

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

Subtask #8
0 / 3.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Incorrect	1 ms	348 KB	Output isn't correct
3	Halted	0 ms	0 KB	-

Subtask #9
0 / 3.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	348 KB	Output is correct
2	Incorrect	1 ms	348 KB	Output isn't correct
3	Halted	0 ms	0 KB	-

Subtask #10
9.0 / 9.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	8 ms	2396 KB	Output is correct

Subtask #11
9.0 / 9.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	32 ms	7504 KB	Output is correct

Subtask #12
0 / 9.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	36 ms	9420 KB	Output is correct
2	Incorrect	71 ms	11980 KB	Output isn't correct
3	Halted	0 ms	0 KB	-

Subtask #13
0 / 9.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	65 ms	10576 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #14
0 / 9.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	75 ms	12092 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Submission #968478

Compilation message