Tree UVALive - 8212

题意：

有n个点，k个颜色，每个点都要被染色，相同颜色之间的边算是被该颜色覆盖，问有多少边被所有颜色覆盖

题解：

题目给的是无根树，我们可以将1默认为根然后求所有点的子树大小，
我们可以这么考虑，对于图中的红边，红边两侧如果点的数量大于等于K的数量，也就是说所有颜色均可以在红边两侧，说明红边可以被所有颜色覆盖，贡献为1。
如何知道红边两侧数量？就是求子树大小，一侧为子树的大小，另一侧为n-子树大小

代码：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define INF 0x3f3f3f3f
const int maxn=2e5+9;
vector<int>vec[maxn];
int siz[maxn];
int getsiz(int u,int fa){for(int i=0;i<vec[u].size();i++){int v=vec[u][i];if(v==fa)continue;siz[v]=1;siz[u]+=getsiz(v,u);}return siz[u];
}
int main(){int t;scanf("%d",&t);while(t--){memset(siz,0,sizeof(siz));int n,k;scanf("%d%d",&n,&k);for(int i=1;i<n;i++){int a,b;scanf("%d%d",&a,&b);vec[a].push_back(b);vec[b].push_back(a);}siz[1]=getsiz(1,0)+1;int tot=0;for(int i=1;i<=n;i++){int x,y;x=siz[i];y=n-x;if(x>=k&&y>=k)tot++;}printf("%d\n",tot);for(int i=1;i<=n;i++){vec[i].clear();}}return 0;
}