AtCoderACGC001C Shorten Diameter

Posted 2021-01-03 artlover

Description:

给定一个(n)个点的树，要求删去最少的点使树的致直径不超过k

Solution:

如果(k)为偶数，对于最终状态一定是以每一个点为根任何点的深度不能超过(k/2)，那我们就以每个点为根求一下深度，然后再求一遍最小值，我们来看(k)为奇数时，我们先钦定一条边固定，分别以两端为根求深度，将问题转化为(k)为偶数即可

Code:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int MAXX=2010;
struct node
{
    int x,y;
}e[MAXX];
int head[MAXX],ver[MAXX<<1],nxt[MAXX<<1],deep[MAXX];
int n,k,tot;
inline void add(int x,int y){
    ver[++tot]=y;
    nxt[tot]=head[x];
    head[x]=tot;
}
inline void dfs(int x,int f,int dp){
    deep[x]=dp;
    for(int i=head[x];i;i=nxt[i]){
        int y=ver[i];
        if(y==f)continue;
        dfs(y,x,dp+1);
    }
} 
int main(){
   scanf("%d%d",&n,&k);
   for(int i=1;i<n;++i){
    scanf("%d%d",&e[i].x,&e[i].y);
    add(e[i].x,e[i].y);
    add(e[i].y,e[i].x);
   }
   int ans=0x3f3f3f3f;
   if(k&1){
      for(int i=1;i<n;++i){
        int cnt=0;
        dfs(e[i].x,e[i].y,0);
        dfs(e[i].y,e[i].x,0);
        for(int j=1;j<=n;++j)if(deep[j]>(k-1)/2)cnt++;
        ans=min(ans,cnt);
      } 
   }
   else {
      for(int i=1;i<=n;++i){
      int cnt=0;
      dfs(i,0,0);
      for(int j=1;j<=n;++j)if(deep[j]>k/2)cnt++;
      ans=min(ans,cnt);
      }
    }
    cout<<ans;
    return 0;
}