【HDOJ6228】Tree（树）

题意：有一棵n个点的树，在树上的点涂色，每个点涂一种颜色，一共可以涂k种颜色，

然后把同一种颜色（比如说x）的点用最优方案连起来，在连线的边涂上x颜色，问涂k次的边最多有几条

k<=500

sigma n<=200000

思路：类似边分治的思路

考虑每一条边，如果以它为中心将树分割出的两部分点数都不小于k的话这条边必定对答案有贡献

这场都是队友写的，划水真爽

#include <stdio.h>
#include <vector>
#include <algorithm>
#include <string.h>
#include <limits.h>
#include <string>
#include <iostream>
#include <queue>
#include <math.h>
#include <stack>
#include <map>
#define left (now<<1)
#define right ((now<<1)+1)
#define mid ((l+r)>>1)
using namespace std;
typedef long long int lint;

const int MAXN = 2e5 + 10;
const int MOD = 1e9 + 7;
const int EPS = 1e-8;

int n,t,s[MAXN];
int ans1,ans2,ans,k;
vector<int> g[MAXN];

int dfs(int fa,int now){
    s[now] = 1;
    if(g[now].size() == 1 && now != 1){ return s[now];}
    for(int i = 0; i < g[now].size(); ++i){
        if(g[now][i] != fa){
            s[now] += dfs(now,g[now][i]);
        }
    }
    return s[now];
}

int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&k); memset(s,0,sizeof(s)); ans = 0;
        for(int i = 1; i <= n; ++i){g[i].clear();}
        for(int i = 1; i < n; ++i){
            int u,v; scanf("%d%d",&u,&v);
            g[u].push_back(v); g[v].push_back(u);
        }
        dfs(-1,1);
//        for(int i = 1; i <= n; ++i){
//            i != n ? printf("%d ",s[i]) : printf("%d
",s[i]);
//        }
        for(int i = 1; i <= n; ++i){
            int t1 = s[i]; int t2 = n - t1;
            if(t1 >= k && t2 >= k) ++ans;
        }
        printf("%d
",ans);
    }
    return 0;
}