Tree HDU - 6228

http://acm.hdu.edu.cn/showproblem.php?pid=6228

考慮每條邊的貢獻 如果一條邊左右兩邊均有不少於k個點 則不管具體如何着色 該邊一定符合條件

 

#include <bits/stdc++.h>
using namespace std;
const int maxn=2e5+10;

struct node1
{
    int u,v;
};

struct node2
{
    int v,next;
};

node1 pre[maxn];
node2 edge[2*maxn];
int first[maxn],sum[maxn],deep[maxn];
int n,k,num;

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
}

void dfs(int cur,int fa)
{
    int i,v;
    sum[cur]=1;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa){
            deep[v]=deep[cur]+1;
            dfs(v,cur);
            sum[cur]+=sum[v];
        }
    }
}

int main()
{
    int t,i,ans;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&k);
        memset(first,-1,sizeof(first));
        num=0;
        for(i=1;i<=n-1;i++){
            scanf("%d%d",&pre[i].u,&pre[i].v);
            addedge(pre[i].u,pre[i].v),addedge(pre[i].v,pre[i].u);
        }
        deep[1]=1;
        dfs(1,0);
        ans=0;
        for(i=1;i<=n-1;i++){
            if(deep[pre[i].u]<deep[pre[i].v]) swap(pre[i].u,pre[i].v);
            if(sum[pre[i].u]>=k&&n-sum[pre[i].u]>=k) ans++;
        }
        printf("%d\n",ans);
    }
    return 0;
}