結果

問題 No.1295 木と駒
ユーザー kotatsugamekotatsugame
提出日時 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()
      | ^~~~

ソースコード

diff #
プレゼンテーションモードにする

#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;
}
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0