树的重心(树形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;
}

浙公网安备 33010602011771号