Submission #802249

#	Time	Username	Problem	Language	Result	Execution time	Memory
802249		Dan4Life	낙하산 고리들 (IOI12_rings)	C++17	52 / 100	1696 ms	262144 KiB

rings

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define sz(a) (int)a.size()
const int N = (int)1e6+9;
vector<int> v;
array<int,2> edges[N];
int n, s=-1, T=1, ok=1, totE;

struct Dsu{
    bitset<N> bad;
    vector<int> adj[N];
    int node, numBad, cycSz;
    int p[N], sz[N];
    short deg[N];
    void init(int N){
        node = -1; numBad=cycSz=0;
        for(int i = 0; i < N; i++){
            adj[i].clear(); 
            p[i]=i,sz[i]=1;
            bad[i]=deg[i]=0; 
        }
    }
    int findSet(int i){ return i==p[i]?i:p[i]=findSet(p[i]); }
    void unionSet(int i, int j){
        if(i==node or j==node) return;
        deg[i]++, deg[j]++, adj[i].pb(j), adj[j].pb(i);
        int x = findSet(i), y = findSet(j);
        if(x==y){ 
            if(!bad[x]) numBad++, bad[x]=1;
            cycSz = sz[x]; return;
        }
        numBad-=bad[x]+bad[y];
        if(sz[x]<sz[y]) swap(x,y);
        p[y] = x; sz[x]+=sz[y];
        if(bad[x]||bad[y]||deg[i]>2||deg[j]>2)
            bad[x]=bad[y]=1, numBad++;
    }
} dsu[5];

void Init(int n_) { n = n_; for(int i=0;i<5;i++) dsu[i].init(n); }

void Link(int a, int b) {
    for(int i=0;i<T;i++) dsu[i].unionSet(a,b);
    if(dsu[0].deg[a]==4 and sz(v)<=1) v.pb(a);
    if(dsu[0].deg[b]==4 and sz(v)<=1) v.pb(b);
    if(dsu[0].deg[a]==3 and s==-1) s=a;
    if(dsu[0].deg[b]==3 and s==-1) s=b;
    edges[totE++]={a,b};
}

int CountCritical(){
    if(sz(v)>=2 or dsu[0].numBad>=2) return 0;
    if(!dsu[0].numBad) return n;
    if(sz(v)==1){
        if(ok){
            dsu[T++].node = v[0]; ok=0;
            for(int i = 0; i < totE; i++)
                dsu[1].unionSet(edges[i][0],edges[i][1]);
        }
        return !dsu[1].numBad;
    }
    if(s==-1) return dsu[0].cycSz;
    if(ok){
        dsu[T++].node=s; ok = 0;
        for(auto u : dsu[0].adj[s]) dsu[T++].node=u;
        for(int j = 1; j < T; j++)
          for(int i = 0; i < totE; i++)
              dsu[j].unionSet(edges[i][0],edges[i][1]);
    }
    int ans=0;for(int i=1;i<T;i++)ans+=!dsu[i].numBad; return ans;
}

Compilation message (stderr)

rings.cpp: In member function 'void Dsu::init(int)':
rings.cpp:21:26: warning: suggest parentheses around assignment used as truth value [-Wparentheses]
   21 |             bad[i]=deg[i]=0;
      |                    ~~~~~~^~

• Subtask 1: 20 / 20
• Subtask 2: 0 / 17
• Subtask 3: 18 / 18
• Subtask 4: 14 / 14
• Subtask 5: 0 / 31