HDU-6228 Tree

传送门：点击打开链接

题意：给出一棵有n个节点的树，现在你可以用k种颜色对节点染色，每种颜色对应一个集合，表示将树上所有这种颜色的点连起来经过的最小边。现在需要求所有集合取交集后的大小。(给一个数，k种颜色，让你给这个树染色，如果一个边的两边的点都含有k种颜色，那么这条边是好边，求可以得到的最多的好边的数量。 )，题目比较难懂。

分析：假设我们取定1为根节点。显然要是结果最大，相同颜色应该要尽可能分布在树的顶部和底部。虽然我们需要考虑的是边，但是我们可以转化为对每个节点去考虑。令在这个节点的两侧每种颜色均有分布，那么一定会有一条公共边。先用DFS求出每个点的子孙节点个数(包括自身)，假设为x，那么它的上面点的个数为n-x,只要x>=k&&(n-x)>=k即能满足上面的条件。我们只要枚举每个点，满足条件点的个数即为答案。

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<set>
#include<map>
#include<stack>
#include<queue>
using namespace std;
const int N = 2e5+10;
#define ll long long
#define pb push_back
#define inf 0x3f3f3f3f
#define eps 1e-10
int t,n,k,num[N],ans;
vector<int> eg[N];

void dfs(int u,int p) {
    num[u]=1;
    for(int i=0;i<eg[u].size();i++) {
        int v=eg[u][i];
        if(v==p) continue;///避免回搜
        dfs(v,u);
        num[u]+=num[v];///num[v]表示v节点的子节点数(包括本身)
        if(num[v]>=k && n-num[v]>=k) ans++;
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin>>t;
    while(t--) {
        cin>>n>>k;
        for(int i=1;i<=n;i++) eg[i].clear();
        memset(num,0,sizeof(num));
        ans=0;
        for(int i=0;i<n-1;i++) {
            int a,b;
            cin>>a>>b;
            eg[a].pb(b);
            eg[b].pb(a);
        }
        dfs(1,0);
        cout<<ans<<endl;
    }
    return 0;
}

参考博客： https://blog.csdn.net/my_sunshine26/article/details/78448923