結果
問題 | No.1769 Don't Stop the Game |
ユーザー | 👑 kmjp |
提出日時 | 2021-11-27 00:06:12 |
言語 | C++17 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 993 ms / 3,000 ms |
コード長 | 2,695 bytes |
コンパイル時間 | 2,624 ms |
コンパイル使用メモリ | 219,468 KB |
実行使用メモリ | 70,528 KB |
最終ジャッジ日時 | 2023-09-12 05:44:45 |
合計ジャッジ時間 | 12,808 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge15 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 8 ms
27,512 KB |
testcase_01 | AC | 8 ms
27,672 KB |
testcase_02 | AC | 8 ms
27,564 KB |
testcase_03 | AC | 8 ms
27,744 KB |
testcase_04 | AC | 8 ms
27,604 KB |
testcase_05 | AC | 8 ms
27,752 KB |
testcase_06 | AC | 14 ms
27,696 KB |
testcase_07 | AC | 9 ms
27,748 KB |
testcase_08 | AC | 993 ms
36,904 KB |
testcase_09 | AC | 125 ms
35,948 KB |
testcase_10 | AC | 200 ms
45,220 KB |
testcase_11 | AC | 283 ms
34,340 KB |
testcase_12 | AC | 209 ms
37,956 KB |
testcase_13 | AC | 212 ms
37,972 KB |
testcase_14 | AC | 325 ms
38,512 KB |
testcase_15 | AC | 975 ms
38,452 KB |
testcase_16 | AC | 778 ms
38,560 KB |
testcase_17 | AC | 245 ms
41,788 KB |
testcase_18 | AC | 291 ms
52,492 KB |
testcase_19 | AC | 301 ms
58,096 KB |
testcase_20 | AC | 317 ms
58,948 KB |
testcase_21 | AC | 311 ms
59,144 KB |
testcase_22 | AC | 314 ms
59,136 KB |
testcase_23 | AC | 127 ms
38,208 KB |
testcase_24 | AC | 154 ms
38,124 KB |
testcase_25 | AC | 107 ms
39,460 KB |
testcase_26 | AC | 233 ms
59,276 KB |
testcase_27 | AC | 158 ms
49,584 KB |
testcase_28 | AC | 271 ms
70,528 KB |
testcase_29 | AC | 262 ms
65,812 KB |
testcase_30 | AC | 266 ms
65,808 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef signed long long ll; #define _P(...) (void)printf(__VA_ARGS__) #define FOR(x,to) for(x=0;x<(to);x++) #define FORR(x,arr) for(auto& x:arr) #define FORR2(x,y,arr) for(auto& [x,y]:arr) #define ALL(a) (a.begin()),(a.end()) #define ZERO(a) memset(a,0,sizeof(a)) #define MINUS(a) memset(a,0xff,sizeof(a)) template<class T> bool chmax(T &a, const T &b) { if(a<b){a=b;return 1;}return 0;} template<class T> bool chmin(T &a, const T &b) { if(a>b){a=b;return 1;}return 0;} //------------------------------------------------------- int N; vector<pair<int,int>> E[202020]; int X[202020]; int L[202020],R[202020]; int id; map<int,vector<int>> V; int P[21][200005],D[200005]; template<int um> class UF { public: vector<int> par,rank,cnt; UF() {par=rank=vector<int>(um,0); cnt=vector<int>(um,1); for(int i=0;i<um;i++) par[i]=i;} void reinit(int num=um) {int i; FOR(i,num) rank[i]=0,cnt[i]=1,par[i]=i;} int operator[](int x) {return (par[x]==x)?(x):(par[x] = operator[](par[x]));} int count(int x) { return cnt[operator[](x)];} int operator()(int x,int y) { if((x=operator[](x))==(y=operator[](y))) return x; cnt[y]=cnt[x]=cnt[x]+cnt[y]; if(rank[x]>rank[y]) return par[x]=y; rank[x]+=rank[x]==rank[y]; return par[y]=x; } }; UF<202020> uf; void dfs(int cur,int x) { X[cur]=x; V[x].push_back(cur); L[cur]=id++; FORR2(e,o,E[cur]) if(e!=P[0][cur]) D[e]=D[cur]+1, P[0][e]=cur, dfs(e,x^o); R[cur]=id; } int getpar(int cur,int up) { int i; FOR(i,20) if(up&(1<<i)) cur=P[i][cur]; return cur; } void solve() { int i,j,k,l,r,x,y; string s; cin>>N; FOR(i,N-1) { cin>>x>>y>>k; E[x-1].push_back({y-1,k}); E[y-1].push_back({x-1,k}); } dfs(0,0); FOR(i,19) FOR(x,N) P[i+1][x]=P[i][P[i][x]]; ll ret=1LL*N*(N-1); FORR2(a,W,V) { if(W.size()<=500) { FORR(x,W) { vector<pair<int,int>> V; FORR(y,W) if(x!=y) { if(L[x]>L[y]&&L[x]<R[y]) { k=getpar(x,D[x]-D[y]-1); V.push_back({0,L[k]}); V.push_back({R[k],N}); } else { V.push_back({L[y],R[y]}); } } sort(ALL(V)); int r=0; FORR2(v1,v2,V) { if(v1<=r) { if(v2>r) { ret-=v2-r; r=v2; } } else { ret-=v2-v1; r=v2; } } } } else { uf.reinit(); FOR(i,N) if(X[i]!=a) FORR2(e,c,E[i]) if(X[e]!=a) uf(i,e); FORR(x,W) { ret-=N-1; FORR2(e,c,E[x]) if(X[e]!=a) ret+=uf.count(e); } } } cout<<ret<<endl; } int main(int argc,char** argv){ string s;int i; if(argc==1) ios::sync_with_stdio(false), cin.tie(0); FOR(i,argc-1) s+=argv[i+1],s+='\n'; FOR(i,s.size()) ungetc(s[s.size()-1-i],stdin); cout.tie(0); solve(); return 0; }