欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

hdu6228

程序员文章站 2022-05-19 20:58:05
...

题意有一颗树,在树上的点涂色,每个点涂一种颜色,一共可以涂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;
}

马上就去南宁了,加油!