Submission #1798886

---

Source Code Expand

#include<set>
#include<map>
#include<cmath>
#include<queue>
#include<bitset>
#include<string>
#include<cstdio>
#include<cctype>
#include<cassert>
#include<cstdlib>
#include<cstring>
#include<sstream>
#include<iostream>
#include<algorithm>

#define For(i,x,y) for (int i=x;i<y;i++)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define lf else if

#define dprintf(...) fprintf(stderr,__VA_ARGS__)
using namespace std;

typedef long long ll;
typedef double db;
typedef pair<int,int> pii;
typedef vector<int> Vi;

int IN(){
	int c,f,x;
	while (!isdigit(c=getchar())&&c!='-');c=='-'?(f=1,x=0):(f=0,x=c-'0');
	while (isdigit(c=getchar())) x=(x<<1)+(x<<3)+c-'0';return !f?x:-x;
}

const int N=200000+19;
const int oo=(1<<30)-1;

struct Edge{
	int y,nxt;
} E[N*2];
Vi V[N];
int D[N],U[N],top[N],low[N];
int n;
char s[N];
ll ans;

void Link(int x,int y){
	V[x].pb(y);
	V[y].pb(x);
}
void dfs(int x,int fa){
	int y;
	For(i,0,V[x].size()) if ((y=V[x][i])!=fa){
		dfs(y,x);
		D[x]=max(D[x],D[y]+1);
	}
}
void dfs2(int x,int fa){
	int mx=U[x],y;
	for (int y:V[x]) if (y!=fa){
		U[y]=max(U[y],mx+1);
		mx=max(mx,D[y]+1);
	}
	mx=U[x];
	for (int i=int(V[x].size())-1;~i;i--) if ((y=V[x][i])!=fa){
		U[y]=max(U[y],mx+1);
		mx=max(mx,D[y]+1);
	}
	for (int y:V[x]) if (y!=fa) dfs2(y,x);
}
void calc(int x,int fa){
	Vi T;
	top[x]=max(D[x],U[x])-1;
	low[x]=n;
	T.pb(0);
	if (~fa){
		T.pb(U[x]-1);
		low[x]=min(low[x],U[x]);
	}
	for (int y:V[x]) if (y!=fa){
		T.pb(D[y]);
		low[x]=min(low[x],D[y]);
		calc(y,x);
	}
	sort(T.begin(),T.end(),greater<int>());
	top[x]=min(top[x],T[1]+1);
}

int main(){
	n=IN();
	For(i,1,n) Link(IN(),IN());
	scanf("%s",s+1);
	if (n==1){
		puts("1");
		return 0;
	}
	dfs(1,-1);
	dfs2(1,-1);
	calc(1,-1);
	For(i,1,n+1)
		if (s[i]=='1'){
			ans+=max(top[i]+1,0);
		} else{
			ans+=max(top[i]-max(low[i],1)+1,0);
		}
	ans++;
	cout<<ans<<endl;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:94:17: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%s",s+1);
                 ^

Judge Result