Tree Gym - 101964C

http://codeforces.com/gym/101964/problem/C

给一棵树 每个点非黑即白 从所有黑点中选m个出来使这m个点之间最远距离最小 暴力枚举一个最远距离dis 把每个黑点向外dis层的点都加一 看是否有哪个点的值超过m即可 具体实现时考虑到奇偶 在每条边的两个点之间都再加一个点即可 和马拉车加*的意思差不多

问题可以转换为 平面上一些点 找一个最小半径 使平面上某一点被至少m个圆覆盖 既然该点被至少m个圆覆盖 那等价于超过m个圆交于一点

 

 

 

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

struct node
{
    int v,next;
};

node edge[4*maxn];
int clr[maxn],first[2*maxn],book[2*maxn];
int n,m,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 dep,int lim)
{
    int i,v;
    book[cur]++;
    if(dep==lim) return;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa){
            dfs(v,cur,dep+1,lim);
        }
    }
}

int main()
{
    int i,j,u,v,ans;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++){
        scanf("%d",&clr[i]);
    }
    memset(first,-1,sizeof(first));
    num=0;
    for(i=1;i<=n-1;i++){
        scanf("%d%d",&u,&v);
        addedge(u,n+i),addedge(n+i,u);
        addedge(v,n+i),addedge(n+i,v);
    }
    ans=-1;
    for(i=0;i<=n;i++){
        memset(book,0,sizeof(book));
        for(j=1;j<=n;j++){
            if(clr[j]) dfs(j,0,0,i);
        }
        for(j=1;j<=2*n;j++){
            if(book[j]>=m){
                ans=i;
                break;
            }
        }
        if(ans!=-1) break;
    }
    printf("%d\n",ans);
    return 0;
}

/*
6 3
1 1 0 1 1 1
1 2
1 3
1 4
3 5
3 6
*/

/*
9 4
1 0 1 0 1 0 0 1 1
1 2
2 4
2 3
4 5
1 6
6 7
6 8
7 9
*/