結果
問題 | No.1295 木と駒 |
ユーザー |
|
提出日時 | 2020-11-20 23:53:25 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 255 ms / 2,000 ms |
コード長 | 1,914 bytes |
コンパイル時間 | 810 ms |
コンパイル使用メモリ | 78,424 KB |
実行使用メモリ | 26,616 KB |
最終ジャッジ日時 | 2024-07-23 14:03:17 |
合計ジャッジ時間 | 11,377 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 48 |
コンパイルメッセージ
main.cpp:113:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type] 113 | main() | ^~~~
ソースコード
#include<iostream> #include<algorithm> #include<vector> using namespace std; int N; vector<int>G[1<<17]; vector<int>A[1<<17]; int sum[1<<17]; int pid[1<<17]; int dfs(int u,int p) { for(int i=0;i<G[u].size();i++) { if(G[u][i]==p) { pid[u]=i; A[u].push_back(-1); } else { int t=dfs(G[u][i],u); sum[u]+=t; A[u].push_back(t); } } if(sum[u]==0) { if(G[u][0]==p||p==-1)return 0; else return 1; } else if(sum[u]==1) { if(A[u].back()==1)return 1; else if(A[u].back()==-1&&A[u][A[u].size()-2]==1)return 1; else if(A[u][0]==1)return 1; else return 2; } else return 2; } int dp[1<<17]; void dfs2(int u,int p,int a) { if(p!=-1) { sum[u]+=a; A[u][pid[u]]=a; } if(sum[u]==0) { dp[u]=true; } else if(sum[u]==1) { if(A[u].back()==1||A[u][0]==1)dp[u]=true; else dp[u]=false; } else dp[u]=false; for(int i=0;i<G[u].size();i++)if(i!=pid[u]) { int nsum=sum[u]-A[u][i]; int v=G[u][i]; if(nsum==0) { if(i==0)dfs2(v,u,0); else dfs2(v,u,1); } else if(nsum==1) { if(i==A[u].size()-1) { if(A[u][A[u].size()-2]==1||A[u][0]==1)dfs2(v,u,1); else dfs2(v,u,2); } else if(i==0) { if(A[u].back()==1)dfs2(v,u,1); else dfs2(v,u,2); } else { if(A[u].back()==1||A[u][0]==1)dfs2(v,u,1); else dfs2(v,u,2); } } else { dfs2(v,u,2); } } } /*int dfs_pre(int u,int p) { int ret=0; vector<int>A; for(int v:G[u])if(v!=p) { int t=dfs(v,u); A.push_back(t); ret+=t; } if(ret==0) { if(G[u][0]==p||p==-1)return 0; else return 1; } else if(ret==1) { if(A.back()==1)return 1; else return 2; } else return 2; }*/ main() { cin>>N; for(int i=1;i<N;i++) { int u,v;cin>>u>>v; u--,v--; G[u].push_back(v); G[v].push_back(u); } for(int i=0;i<N;i++)sort(G[i].begin(),G[i].end()); dfs(0,-1); pid[0]=-1; dfs2(0,-1,0); for(int i=0;i<N;i++) { cout<<(dp[i]?"Yes":"No")<<endl; } }