AGC001 C - Shorten Diameter（枚举+dfs）

题意：

给定一棵n个顶点的树，
现在要删除一些点，使得数的直径<=k，
能删除点x的条件是去掉点x不会对树的连通性造成影响。
问最少删除多少个点可以满足要求。

数据范围：n<=2e3，k<=n-1

解法：

首先发现n<=2e3,应该是O(n^2)的算法

显然可以将问题变为最多可以保留多少个点.

考虑枚举直径的中点:
1.当k是偶数的时候,直径存在中点,那么枚举中点x,那么到点x的距离<=k/2的可以保留.
2.当k是奇数的时候,不存在中点,但是次数存在中间的边,那么枚举中间的那条边(x,v),
那么到点x和点v的距离<=k/2的可以保留.

从枚举的点开始dfs一下就行了

code：

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=1e6+5;
struct Node{
    int a,b;
}e[maxm];
vector<int>g[maxm];
int n,k;
int temp;
int limit;
void dfs(int x,int fa,int dep){
    if(dep>limit)return ;
    temp++;
    for(int v:g[x]){
        if(v==fa)continue;
        dfs(v,x,dep+1);
    }
}
int solve1(int x){
    temp=0;
    dfs(x,x,0);
    return n-temp;
}
int solve2(int p){
    temp=0;
    dfs(e[p].a,e[p].b,0);
    dfs(e[p].b,e[p].a,0);
    return n-temp;
}
signed main(){
    cin>>n>>k;
    for(int i=1;i<n;i++){
        int a,b;cin>>a>>b;
        e[i]={a,b};
        g[a].push_back(b);
        g[b].push_back(a);
    }
    int ans=1e9;
    limit=k/2;
    if(k%2==0){
        for(int i=1;i<=n;i++){
            int t=solve1(i);
            ans=min(ans,t);
        }
    }else{
        limit=k/2;
        for(int i=1;i<n;i++){
            int t=solve2(i);
            ans=min(ans,t);
        }
    }
    cout<<ans<<endl;
    return 0;
}

---