Submission #369272

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
#define INF	100000000
#define MOD 1000000007
#define pb push_back
#define fi first
#define sc second
#define rep(i,n) for(int i = 0; i < n; i++)
typedef long long int ll;
typedef pair<int, int> P;
typedef pair<P,int> PP;

int n;
int ans;
vector<int> G[100000], val[100000];
int dp[100000][2];
bool used[100000];

void dfs(int s){
	used[s] = true;
	rep(i,G[s].size()){
		if(used[G[s][i]]) continue;
		dfs(G[s][i]);
		if(val[s][i] != 0) ans = max(ans,dp[s][1]+dp[G[s][i]][0]);
		if(val[s][i] != 1) ans = max(ans,dp[s][0]+dp[G[s][i]][1]);
		if(val[s][i] != 0) dp[s][0] = max(dp[s][0],dp[G[s][i]][0]+1);
		if(val[s][i] != 1) dp[s][1] = max(dp[s][1],dp[G[s][i]][1]+1);
	}
}

int main(){
	scanf("%d",&n);
	rep(i,n-1){
		int a, b, t;
		scanf("%d%d%d",&a,&b,&t);
		a--; b--;
		if(t == 2){
			G[a].pb(b);
			G[b].pb(a);
			val[a].pb(2);
			val[b].pb(2);
		} else{
			G[a].pb(b);
			G[b].pb(a);
			val[a].pb(1);
			val[b].pb(0);
		}
	}
	dfs(0);
	printf("%d\n",ans);
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:32:16: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d",&n);
                ^
./Main.cpp:35:27: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d%d%d",&a,&b,&t);
                           ^

Judge Result