hdoj 5606 ||bc #68 div 2 B tree

http://acm.hdu.edu.cn/showproblem.php?pid=5606 题意:一棵树,边权为0或者1,问对于每个点,距离它最近的点(包括自身)的个数是多少。输出将所有点的答案异或后的值。 思路:由于包括自身,自己与自己距离为0,那么最近的点一定也距离为0,所以就是找对于每个点与它相连的边权为0 的点的个数**。建图的时候可以不管边权为1的点。。因为这样的点不会对任何点的答案有贡献。**正解貌似是冰茶几。。我就是dfs搞了下。。找到每一个联通快的点数。。然后把某个联通快的所有点的答案都更新成点的个数。。。

/* ***********************************************
Author :111qqz
Created Time :2016年01月02日 星期六 18时56分17秒
File Name :code/bc/#68/1002.cpp
************************************************ */
 1#include <cstdio>
 2#include <cstring>
 3#include <iostream>
 4#include <algorithm>
 5#include <vector>
 6#include <queue>
 7#include <set>
 8#include <map>
 9#include <string>
10#include <cmath>
11#include <cstdlib>
12#include <ctime>
13#define fst first
14#define sec second
15#define lson l,m,rt<<1
16#define rson m+1,r,rt<<1|1
17#define ms(a,x) memset(a,x,sizeof(a))
18typedef long long LL;
19#define pi pair < int ,int >
20#define MP make_pair
 1using namespace std;
 2const double eps = 1E-8;
 3const int dx4[4]={1,0,0,-1};
 4const int dy4[4]={0,-1,1,0};
 5const int inf = 0x3f3f3f3f;
 6const int N=1E5+7;
 7int n;
 8vector<int>Edge[N];
 9int cnt;
10bool vis[N];
11int ans[N];
12int sum[N];
13int path[N];
14void print()
15{
16    for ( int i =1 ; i <= n ; i++)
17    {
18	cout<<Edge[i].size()<<endl;
19    }
20}
 1void dfs(int x)
 2{
 3   // cout<<"x:"<<x<<endl;
 4    vis[x] = true;
 5    cnt++;
 6    path[cnt] = x;
 7    for ( int i = 0 ; i < Edge[x].size() ; i++)
 8    {
 9	int v = Edge[x][i];
10	if (!vis[v])
11	{
12	    dfs(v);
13	}
14    }
 1}
 2int main()
 3{
 4	#ifndef  ONLINE_JUDGE 
 5	freopen("code/in.txt","r",stdin);
 6  #endif
 7	int T;
 8	cin>>T;
 9	while (T--)
10	{
11	    ms(ans,0);
12	    ms(vis,false);
13	    scanf("%d",&n);
14	    for ( int i = 1 ; i <= n ; i++)
15	    {
16		Edge[i].clear();
17		Edge[i].push_back(i);
18	    }
19	    for ( int i = 1 ; i <= n -1 ; i++)
20	    {
21		int u,v,w;
22		scanf("%d %d %d",&u,&v,&w);
23		if (w==1) continue;
24		Edge[u].push_back(v);
25		Edge[v].push_back(u);
26	    }
27	    for ( int i = 1 ; i <= n ; i++)
28	    {
29		cnt = 0 ;
30		if (!vis[i])
31		{
32		    dfs(i);
33		}
34		for ( int j = 1 ; j <= cnt ; j ++) ans[path[j]] = cnt;
35	    }
36	  //  print();
37	    int total = 0;
38	    for ( int i = 1 ; i <= n ; i++)
39	    {
40	//	cout<<ans[i]<<endl;
41		total ^= ans[i];
42	    }
43	    printf("%d\n",total);
	}
1  #ifndef ONLINE_JUDGE  
2  fclose(stdin);
3  #endif
4    return 0;
5}