hdu6228

題意有一顆樹，在樹上的點塗色，每個點塗一種顏色，一共可以塗k種顏色，然後你需要把x顏色的點用最短路徑連起來，在連線的邊塗上x顏色，問塗k次的邊最多有幾條？
題意轉化就是問一條邊上的兩端，如果上端的點數大於k，並且下端的點數小於k，那麼這條邊將計數，（主要是題意轉化題)
心路歷程這個是今年瀋陽的題，一開始，我以爲是這個由祖先節點所決定，但是後來發現不是這樣，後來以爲是和子孫節點共同確定的，（主要是來了新地方課業多很久不寫題腦殘了）最後纔想明白，就是由子孫節點確定的，一段是n - sub[i] , 還有一端是sub[i];

#include <cstdio>
#include <queue>
#include <algorithm>
#include <queue>
#include <cmath>
#include <cstring>

using namespace std;

const int N = 2e5 + 10;

int all , ans;

int other[N * 2] , last[N * 2] , pre[N * 2] , n , k , num[N] , sub[N];

void build(int x , int y)
{
    pre[++all] = last[x];
    last[x] = all;
    other[all] = y;
}

void dfs(int x , int fa)
{
    int ed = last[x];
    int tmp = 0;
    //printf("x = %d\n",x);
    while(ed != -1){
        int dr = other[ed];
        if(dr != fa){
            num[dr] = num[x] + 1;
            dfs(dr , x);
            sub[x] += sub[dr];
        }
        ed = pre[ed];
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        memset(last , -1 , sizeof(last));
        all = -1;
        ans = 0;
        scanf("%d %d", &n , &k);
        for(int i = 1 ; i < n ; i++){
            int x , y;
            scanf("%d %d",&x , &y);
            build(x , y);
            build(y , x);
        }
        memset(num , 0 , sizeof(num));
        for(int i = 1 ; i <= n ; i++) sub[i] = 1;
        dfs(1 , -1);
        for(int i = 1 ; i <= n ; i++){
            //printf("**%d %d k = %d\n",num[i] , sub[i] , k);
            if(n - sub[i] >= k && sub[i] >= k) ans++ ;
        }
        printf("%d\n",ans);
    }
    return 0;
}

馬上就去南寧了，加油！