树的重心(树形dp)


我最开始将n减了1,应为边长有n-1条,但是这会导致计算时出现错误,因为其他地方会用到n

#include<iostream>
#include<vector>
using namespace std;
const int N=1e5+5;
vector<int>v[N];
int f[N];
int vis[N];
int ans=1e5;
int n;
int dfs(int x){
    vis[x]=1;
    int sum=1;
    int size=0;
    for(int i=0;i<v[x].size();i++){
        if(!vis[v[x][i]]){
            int son=v[x][i];
            int s=dfs(son);
            size=max(size,s);
            sum+=s;
        }
    }
    ans=min(ans,max(n-sum,size));
    return sum;
}
int main(){
    cin>>n;
    int m=n-1;
    while(m--){
        int a,b;
        cin>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    dfs(1);
    cout<<ans;
    
    return 0;
}
posted @ 2025-02-21 16:29  郭轩均  阅读(17)  评论(0)    收藏  举报