結果
問題 | No.1295 木と駒 |
ユーザー |
|
提出日時 | 2020-11-20 22:29:57 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,782 bytes |
コンパイル時間 | 801 ms |
コンパイル使用メモリ | 77,848 KB |
実行使用メモリ | 26,708 KB |
最終ジャッジ日時 | 2024-07-23 13:23:23 |
合計ジャッジ時間 | 11,410 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 20 WA * 28 |
コンパイルメッセージ
main.cpp:107:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type] 107 | 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 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)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)dfs2(v,u,1);else dfs2(v,u,2);}else{if(A[u].back()==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());dp[0]=dfs(0,-1)!=2;pid[0]=-1;dfs2(0,-1,0);for(int i=0;i<N;i++){cout<<(dp[i]?"Yes":"No")<<endl;}}