Submission #1870337
Source Code Expand
#include <iostream> #include <cstdio> #include <cstring> #include <cassert> #include <cctype> using namespace std; typedef long long lint; #define cout cerr #define ni (next_num<int>()) template<class T>inline T next_num(){ T i=0;char c; while(!isdigit(c=getchar())&&c!='-'); bool flag=c=='-'; flag?c=getchar():0; while(i=i*10-'0'+c,isdigit(c=getchar())); return flag?-i:i; } template<class T1,class T2>inline void apmax(T1 &a,const T2 &b){if(a<b)a=b;} template<class T1,class T2>inline void apmin(T1 &a,const T2 &b){if(b<a)a=b;} const int N=200010,INF=0x7f7f7f7f; lint ans=1; char s[N]; int onecnt=0; namespace T{ const int E=N<<1; int to[E],bro[E],head[N],e=0; inline void init(){ memset(head,-1,sizeof(head)); } inline void ae(int u,int v){ to[e]=v,bro[e]=head[u],head[u]=e++; } inline void add(int u,int v){ ae(u,v),ae(v,u); } int dep[E],cnt[E]; void dfs1(int x,int e){ if(~e){ dep[e]=1; cnt[e]=s[x]=='1'; } for(int i=head[x],v;~i;i=bro[i]){ if(i^e^1){ dfs1(v=to[i],i); if(~e){ cnt[e]+=cnt[i]; apmax(dep[e],dep[i]+1); } } } if(~e){ cnt[e^1]=onecnt-cnt[e]; } } void dfs2(int x,int fa){ int mx1=0,mx2=0,mn=s[x]=='1'?0:INF; int fr=0; for(int i=head[x];~i;i=bro[i]){ if(dep[i]>mx1){ mx2=mx1,mx1=dep[i]; fr=i; }else if(dep[i]>mx2){ mx2=dep[i]; } if(cnt[i]){ apmin(mn,dep[i]); } } ans+=max(min(mx1-1,mx2+1)-mn+1,0); for(int i=head[x],v;~i;i=bro[i]){ if((v=to[i])!=fa){ dep[i^1]=(fr!=i?mx1:mx2)+1; dfs2(v,x); } } } } inline lint Main(){ int n=ni; T::init(); for(int i=1;i<n;i++){ T::add(ni,ni); } scanf("%s",s+1); for(int i=1;i<=n;i++){ onecnt+=s[i]=='1'; } T::dfs1(1,-1),T::dfs2(1,0); return ans; } int main(){ printf("%lld\n",Main()); return 0; }
Submission Info
Submission Time | |
---|---|
Task | F - Black Radius |
User | sshockwave |
Language | C++14 (GCC 5.4.1) |
Score | 1900 |
Code Size | 1886 Byte |
Status | AC |
Exec Time | 52 ms |
Memory | 17024 KB |
Compile Error
./Main.cpp: In function ‘lint Main()’: ./Main.cpp:84:17: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result] scanf("%s",s+1); ^
Judge Result
Set Name | Sample | Subtask | All | ||||||
---|---|---|---|---|---|---|---|---|---|
Score / Max Score | 0 / 0 | 1300 / 1300 | 600 / 600 | ||||||
Status |
|
|
|
Set Name | Test Cases |
---|---|
Sample | 0_00.txt, 0_01.txt, 0_02.txt |
Subtask | 0_01.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt, 1_08.txt, 1_09.txt, 1_10.txt, 1_11.txt, 1_12.txt, 1_13.txt, 1_14.txt, 1_15.txt, 1_16.txt, 1_17.txt, 1_18.txt, 1_19.txt, 1_20.txt, 1_21.txt, 1_22.txt, 1_23.txt, 1_24.txt, 1_25.txt, 1_26.txt, 1_27.txt, 1_28.txt, 1_29.txt |
All | 0_00.txt, 0_01.txt, 0_02.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt, 1_08.txt, 1_09.txt, 1_10.txt, 1_11.txt, 1_12.txt, 1_13.txt, 1_14.txt, 1_15.txt, 1_16.txt, 1_17.txt, 1_18.txt, 1_19.txt, 1_20.txt, 1_21.txt, 1_22.txt, 1_23.txt, 1_24.txt, 1_25.txt, 1_26.txt, 1_27.txt, 1_28.txt, 1_29.txt, 2_00.txt, 2_01.txt, 2_02.txt, 2_03.txt, 2_04.txt, 2_05.txt, 2_06.txt, 2_07.txt, 2_08.txt, 2_09.txt, 2_10.txt, 2_11.txt, 2_12.txt, 2_13.txt, 2_14.txt, 2_15.txt, 2_16.txt, 2_17.txt, 2_18.txt, 2_19.txt, 2_20.txt, 2_21.txt, 2_22.txt, 2_23.txt, 2_24.txt, 2_25.txt, 2_26.txt, 2_27.txt, 2_28.txt, 2_29.txt, 2_30.txt, 2_31.txt, 2_32.txt, 2_33.txt, 2_34.txt, 2_35.txt, 2_36.txt, 2_37.txt, 2_38.txt, 2_39.txt, 2_40.txt, 2_41.txt, 2_42.txt, 2_43.txt, 2_44.txt, 2_45.txt, 2_46.txt, 2_47.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
0_00.txt | AC | 2 ms | 4352 KB |
0_01.txt | AC | 2 ms | 4352 KB |
0_02.txt | AC | 2 ms | 4352 KB |
1_00.txt | AC | 2 ms | 4352 KB |
1_01.txt | AC | 51 ms | 17024 KB |
1_02.txt | AC | 49 ms | 13568 KB |
1_03.txt | AC | 45 ms | 7552 KB |
1_04.txt | AC | 35 ms | 7424 KB |
1_05.txt | AC | 29 ms | 7424 KB |
1_06.txt | AC | 42 ms | 7424 KB |
1_07.txt | AC | 42 ms | 7552 KB |
1_08.txt | AC | 44 ms | 8320 KB |
1_09.txt | AC | 49 ms | 12288 KB |
1_10.txt | AC | 46 ms | 11520 KB |
1_11.txt | AC | 51 ms | 16896 KB |
1_12.txt | AC | 43 ms | 9600 KB |
1_13.txt | AC | 48 ms | 11904 KB |
1_14.txt | AC | 44 ms | 9344 KB |
1_15.txt | AC | 43 ms | 8448 KB |
1_16.txt | AC | 45 ms | 10112 KB |
1_17.txt | AC | 48 ms | 11264 KB |
1_18.txt | AC | 43 ms | 7424 KB |
1_19.txt | AC | 42 ms | 7424 KB |
1_20.txt | AC | 44 ms | 7552 KB |
1_21.txt | AC | 45 ms | 8064 KB |
1_22.txt | AC | 38 ms | 7424 KB |
1_23.txt | AC | 39 ms | 7424 KB |
1_24.txt | AC | 42 ms | 8192 KB |
1_25.txt | AC | 39 ms | 8960 KB |
1_26.txt | AC | 39 ms | 7424 KB |
1_27.txt | AC | 37 ms | 7424 KB |
1_28.txt | AC | 36 ms | 7808 KB |
1_29.txt | AC | 43 ms | 9600 KB |
2_00.txt | AC | 2 ms | 4352 KB |
2_01.txt | AC | 52 ms | 16640 KB |
2_02.txt | AC | 51 ms | 15104 KB |
2_03.txt | AC | 51 ms | 14720 KB |
2_04.txt | AC | 49 ms | 12032 KB |
2_05.txt | AC | 48 ms | 12160 KB |
2_06.txt | AC | 49 ms | 12928 KB |
2_07.txt | AC | 45 ms | 7552 KB |
2_08.txt | AC | 45 ms | 7552 KB |
2_09.txt | AC | 45 ms | 7552 KB |
2_10.txt | AC | 36 ms | 7424 KB |
2_11.txt | AC | 35 ms | 7424 KB |
2_12.txt | AC | 35 ms | 7424 KB |
2_13.txt | AC | 30 ms | 7424 KB |
2_14.txt | AC | 28 ms | 7424 KB |
2_15.txt | AC | 29 ms | 7424 KB |
2_16.txt | AC | 42 ms | 7424 KB |
2_17.txt | AC | 42 ms | 7424 KB |
2_18.txt | AC | 42 ms | 7424 KB |
2_19.txt | AC | 42 ms | 7552 KB |
2_20.txt | AC | 42 ms | 7552 KB |
2_21.txt | AC | 45 ms | 7424 KB |
2_22.txt | AC | 43 ms | 8192 KB |
2_23.txt | AC | 44 ms | 8576 KB |
2_24.txt | AC | 44 ms | 8448 KB |
2_25.txt | AC | 48 ms | 12544 KB |
2_26.txt | AC | 47 ms | 11648 KB |
2_27.txt | AC | 48 ms | 12544 KB |
2_28.txt | AC | 45 ms | 11136 KB |
2_29.txt | AC | 49 ms | 15488 KB |
2_30.txt | AC | 48 ms | 14080 KB |
2_31.txt | AC | 47 ms | 11264 KB |
2_32.txt | AC | 42 ms | 7808 KB |
2_33.txt | AC | 44 ms | 9472 KB |
2_34.txt | AC | 45 ms | 10880 KB |
2_35.txt | AC | 48 ms | 13184 KB |
2_36.txt | AC | 44 ms | 7424 KB |
2_37.txt | AC | 43 ms | 7552 KB |
2_38.txt | AC | 43 ms | 8320 KB |
2_39.txt | AC | 45 ms | 7680 KB |
2_40.txt | AC | 37 ms | 7424 KB |
2_41.txt | AC | 40 ms | 7424 KB |
2_42.txt | AC | 34 ms | 7424 KB |
2_43.txt | AC | 45 ms | 12032 KB |
2_44.txt | AC | 35 ms | 7424 KB |
2_45.txt | AC | 32 ms | 7424 KB |
2_46.txt | AC | 32 ms | 7680 KB |
2_47.txt | AC | 41 ms | 10112 KB |