【NOIP2017提高A组模拟9.12】Shorten Diameter

题目

Description

给定一棵有n 个点的树，现要求不断删点直到树的直径<=K，求最少需要删除的点数。
一个点可以被删掉当且仅当该点的度数为1。

Input

第一行两个数n，k，意义如题所述。
接下来n -1 行描述这棵树。

Output

一个数表示答案。

Sample Input

6 2
1 2
3 2
4 2
1 6
5 6

Sample Output

2

Data Constraint

对于100% 的数据，K <= n <= 2000。
保证树的形态为随机生成（请勿过度解读）。

题解

我们可以枚举最终的树的直径的中点，如果k为偶数，那么我们就枚举一个点，如果k为奇数那么我们枚举其中一条边的中点，在确定了直径的中点之后，我们只要把所有与中点距离大于k/2的去掉就可以了
这样做完之后不仅过枚举点的路径的长度都<=k而且不过所枚举点的路径也满足这一限制，因为有dis[i]+dis[j]>=distance(i,j)
那么我们就可以把每一种情况都考虑到了

贴代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
#define fo(i,a,b) for(i=a;i<=b;i++)

using namespace std;

const int maxn=2005;

int dis[maxn],fi[maxn],ne[maxn*2],dui[maxn*2],qc[maxn],duii[maxn*2];
bool bz[maxn];
int i,j,k,l,m,n,x,y,z,ans,tot,now;

void add(int x,int y){
    if (fi[x]==0) fi[x]=++now; else ne[qc[x]]=++now;
    qc[x]=now; dui[now]=y; duii[now]=x;
}
void dfs(int x){
    bz[x]=true;
    int i=fi[x];
    while (i){
        if (bz[dui[i]]==true){
            i=ne[i];
            continue;
        }
        dis[dui[i]]=dis[x]+1;
        dfs(dui[i]);
    }
}
void work1(){
    k=k/2;
    ans=n;
    fo(i,1,n){
        memset(bz,false,sizeof(bz));
        dis[i]=0;
        dfs(i); tot=0;
        fo(j,1,n) if (dis[j]>k) tot++;
        ans=min(ans,tot);
    }
}
void work2(){
    k=(k+1)/2;
    ans=n;
    fo(i,1,now){
        memset(bz,false,sizeof(bz));
        dis[dui[i]]=dis[duii[i]]=1;
        bz[dui[i]]=bz[duii[i]]=true;
        dfs(dui[i]);
        dfs(duii[i]); tot=0;
        fo(j,1,n) if (dis[j]>k) tot++;
        ans=min(ans,tot);

    }
}
int main(){
    freopen("c.in","r",stdin);
    freopen("c.out","w",stdout);
    scanf("%d%d",&n,&k);
    fo(i,1,n){
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    if (k%2==0) work1(); else work2();
    printf("%d",ans);
    return 0;
}