Submission #535417

#TimeUsernameProblemLanguageResultExecution timeMemory
535417mario05092929Split the Attractions (IOI19_split)C++14
100 / 100
288 ms25112 KiB
#include "split.h" #include <bits/stdc++.h> #define x first #define y second using namespace std; typedef long long ll; typedef pair <int,int> pi; typedef vector <int> vec; vector <int> rv[100005]; vector <int> v[100005]; int n,m,A,B,C,a,b; int cnt,cntmx; int c[100005],sz[100005],p[100005],szp[100005]; int ch[100005],poham[100005]; pi po[3]; vector <int> ans; vector <int> tv[100005]; vector <int> com,dfscom; int Find(int x) {return (x == p[x] ? x : p[x] = Find(p[x]));} void init_union() { for(int i = 0;i < n;i++) { p[i] = i, szp[i] = 1; } } int get_sz(int x,int pp) { sz[x] = 1; for(int i : tv[x]) { if(i == pp) continue; sz[x] += get_sz(i,x); } return sz[x]; } int get_cent(int x,int pp,int cap) { for(int i : tv[x]) if(i != pp&&sz[i] > cap) return get_cent(i,x,cap); return x; } int make_cent() { return get_cent(0,-1,get_sz(0,-1)/2); } void dfs_fill(int x,int pp,int color) { if(cnt == cntmx||ans[x]) return; cnt++; ans[x] = color; for(int i : tv[x]) { if(i == pp) continue; dfs_fill(i,x,color); } } void dfs_pofill(int x,int color,int isok) { if(cnt == cntmx||ans[x]||poham[x] != isok) return; cnt++; ans[x] = color; for(int i : rv[x]) { dfs_pofill(i,color,isok); } } void canDo(int x,int wi) { cnt = 0, cntmx = a; dfs_fill(wi,x,po[0].y); cnt = 0, cntmx = b; dfs_fill(x,wi,po[1].y); for(int i = 0;i < n;i++) { if(!ans[i]) ans[i] = po[2].y; } } void merge(int x,int y) { if(Find(x) == Find(y)) return; x = Find(x), y = Find(y); p[x] = y; szp[y] += szp[x]; } void dfs_union(int x,int pp) { for(int i : tv[x]) { if(i == pp) continue; merge(i,x); dfs_union(i,x); } } void dfs_com(int x) { if(c[x]) return; c[x] = 1; dfscom.push_back(x); for(int i : v[x]) dfs_com(i); } vec find_split(int N,int AA,int BB,int CC,vec vec1,vec vec2) { n = N, A = AA, B = BB, C = CC; m = vec1.size(); po[0] = {A,1}, po[1] = {B,2}, po[2] = {C,3}; sort(po,po+3); a = po[0].x, b = po[1].x; ans.resize(n,0); init_union(); for(int i = 0;i < m;i++) { int x = vec1[i], y = vec2[i]; rv[x].push_back(y); rv[y].push_back(x); if(Find(x) == Find(y)) continue; merge(x,y); tv[x].push_back(y); tv[y].push_back(x); } int cent = make_cent(); get_sz(cent,-1); init_union(); for(int i : tv[cent]) { dfs_union(i,cent); if(szp[Find(i)] >= a) { canDo(cent,i); return ans; } } for(int i = 0;i < m;i++) { if(vec1[i] == cent||vec2[i] == cent) continue; int x = vec1[i], y = vec2[i]; if(Find(x) != Find(y)) { v[p[x]].push_back(p[y]); v[p[y]].push_back(p[x]); } } memset(c,0,sizeof(c)); for(int i = 0;i < n;i++) { if(i == cent||Find(i) != i||c[i]) continue; dfscom.clear(), com.clear(); dfs_com(i); int sum = 0; for(int j : dfscom) { sum += szp[j]; com.push_back(j); if(sum >= a) break; } if(sum >= a) { int st = -1; for(int j : com) ch[j] = 1; for(int j = 0;j < n;j++) { if(ch[Find(j)]) { poham[j] = 1; st = j; } } cnt = 0, cntmx = a; dfs_pofill(st,po[0].y,1); cnt = 0, cntmx = b; dfs_pofill(cent,po[1].y,0); for(int j = 0;j < n;j++) { if(!ans[j]) ans[j] = po[2].y; } return ans; } } return ans; }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...