Atcoder 1981 Shorten Diameter 树,枚举,思维

题意

给你一棵树,问至少去掉多少个点,能够使得树的直径不大于k.去完点之后的树必须连通.

题解

这题思路高妙.
考虑要求的直径的奇偶性.
如果直径长度$k$是偶数,必然能够保证找到一个点,任何一个点离它的距离都不超过$\frac{k}{2}$.
同理如果直径长度$k$是奇数,必然能够找到一条边,任何一个点离这条边连接的两点的距离的较小值都不超过$\frac{k-1}{2}$.
满足条件的点就是重心.
那么我们暴力枚举去完点以后的重心,然后对于每一个重心来说都把与它距离大于$\frac{k}{2}$的都去掉,算出最小值输出即可.
注意奇数和偶数处理稍有不同,比较坑.

#include<bits/stdc++.h> //Ithea Myse Valgulious
namespace chtholly{
typedef long long ll;
#define re0 register int
#define rec register char
#define rel register ll
#define gc getchar
#define pc putchar
#define p32 pc(' ')
#define pl puts("")
/*By Citrus*/
inline int read(){
  int x=0,f=1;char c=gc();
  for (;!isdigit(c);c=gc()) f^=c=='-';
  for (;isdigit(c);c=gc()) x=(x<<3)+(x<<1)+(c^'0');
  return f?x:-x;
  }
template <typename mitsuha>
inline bool read(mitsuha &x){
  x=0;int f=1;char c=gc();
  for (;!isdigit(c)&&~c;c=gc()) f^=c=='-';
  if (!~c) return 0;
  for (;isdigit(c);c=gc()) x=(x<<3)+(x<<1)+(c^'0');
  return x=f?x:-x,1;
  }
template <typename mitsuha>
inline int write(mitsuha x){
  if (!x) return 0&pc(48);
  if (x<0) x=-x,pc('-');
  int bit[20],i,p=0;
  for (;x;x/=10) bit[++p]=x%10;
  for (i=p;i;--i) pc(bit[i]+48);
  return 0;
  }
inline char fuhao(){
  char c=gc();
  for (;isspace(c);c=gc());
  return c;
  }
}using namespace chtholly;
using namespace std;
const int sanae=2e3;
typedef int fuko[sanae|15];
vector<int> lj[sanae|15];
fuko d[sanae|15];
struct edge{int u,v;}e[sanae|13];
void dfs(int p,int u,int fa){
d[p][u]=d[p][fa]+1;
for (int i:lj[u]) if (i^fa) dfs(p,i,u);
}

int main(){
int i,j,n=read(),k=read();
for (i=1;i<n;++i){
  int u=read(),v=read();
  lj[u].push_back(v);
  lj[v].push_back(u);
  e[i]=edge{u,v};
  }
for (i=1;i<=n;++i) dfs(i,i,0); //暴力跑每两个点之间的距离.
int llx=9999;
if (k&1){//奇数要枚举边
  for (i=1;i<n;++i){
    int tmp=0;
    for (j=1;j<=n;++j){
      if (min(d[e[i].u][j],d[e[i].v][j])>k/2+1) tmp++;
      }
    llx=min(llx,tmp);
    }return write(llx);
  }
for (i=1;i<=n;++i){//偶数枚举点
  int tmp=0;
  for (j=1;j<=n;++j) if (d[i][j]>k/2+1) tmp++;
  llx=min(tmp,llx); 
  }write(llx);
}