HDU 6228 Tree 【dfs】

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6228

题意：给n个节点染k种颜色，每一种颜色以最少的边连在一起，求这些颜色共同经过的边最多有几条；

思路：

一个边要想满足同时被k种颜色经过，那么，这个边左边的点和右边的点一定大于或者等于k，否则不会满足条件，所以只要找出有几条这种边，就是答案，用dfs去遍历点；

 

#include<iostream>
#include<vector>
#include<cstring>

using namespace std;

const int Maxn = 2e5+10;

struct Edge {
	int u,v;
} edges[Maxn*2];

int n,k,ans,cnt[Maxn*2];
vector<int> G[Maxn*2];

void dfs(int u,int f) {  // f为u的父节点，保证u向下遍历的时候不会遍历到自己的父节点
	int m = G[u].size(),v;
	cnt[u] = 1;
	for (int i = 0; i < m; ++i) {
		v = edges[G[u][i]].v;
		if(v == f) continue; // v是父节点就跳过
		dfs(v,u);
		cnt[u]+=cnt[v]; // 累加 v 向下dfs的点
	}
	if(cnt[u] >= k && n-cnt[u] >= k) ans++; // 一边的点知道了，另一边只要n-cnt就知道了
}

int main (void)
{
	int T;
	scanf("%d",&T);
	while (T--) {
		scanf("%d%d",&n,&k);
		for (int i = 0; i <= n; ++i) G[i].clear();
		memset(cnt,0,sizeof(cnt));
		int u,v;
		for (int i = 1; i < n; ++i) {
			scanf("%d%d",&u,&v);
			edges[i].u = u; edges[i].v = v;
			G[u].push_back(i);
			edges[i+n].u = v; edges[i+n].v = u;
			G[v].push_back(i+n);
		}
		ans = 0;
		dfs(1,-1);
		printf("%d\n",ans);
	}
	return 0;
 }