D51 树的直径 [AGC001C] Shorten Diameter

视频链接：D51 树的直径 [AGC001C] Shorten Diameter_哔哩哔哩_bilibili

[AGC001C] Shorten Diameter - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

// 树的直径+逆向思维
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

#define N 2005
int head[N],to[N<<1],ne[N<<1],idx;
void add(int x,int y){
  to[++idx]=y;ne[idx]=head[x];head[x]=idx;
}
int n,k,tot,ans;

void dfs(int x,int fa,int step){
  tot++; //记录节点数
  if(step==0) return;
  for(int i=head[x];i;i=ne[i]){
    int y=to[i];
    if(y==fa)continue;
    dfs(y,x,step-1);
  }
}
int main(){
  scanf("%d%d",&n,&k);
  for(int i=1,u,v;i<n;i++){
    scanf("%d%d",&u,&v);
    add(u,v);add(v,u);
  }
  
  if(k%2==0){ //k为偶数
    for(int x=1;x<=n;x++){
      tot=0;
      dfs(x,0,k/2); //以x为中心扩展
      ans=max(ans,tot);
    }
  }
  else{ //k为奇数
    for(int x=1;x<=n;x++){
      for(int i=head[x];i;i=ne[i]){
        tot=0;
        int y=to[i];
        dfs(y,x,k/2);
        dfs(x,y,k/2); //以x,y为中心扩展
        ans=max(ans,tot);
      }
    }
  }
  printf("%d\n",n-ans);
}