[树] hdu 6867 Tree

题目

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6867

题目思路： 给一个树，连接两个节点，求连接后的点对数量的最大值

思路

既然是树，那么就是有向图
如果要连接之后增加的点对数量最大 那就是连接叶子节点和根节点 形成一个环 在这个环上每一个结点增加的连接数量都是n-cnt[i] (没链接之前i的子树数量 )

那就先一个dfs 求每个节点子树
再来一个dfs求最大值就好了

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define int long long
using namespace std;
const int INF = 0x3f3f3f3f;
vector<int> v[500010];
int cnt[500010];
int sum=0,ans=0,n;
void dfs(int n1){
	cnt[n1]=1;
	for(int i=0;i<v[n1].size();i++){
		dfs(v[n1][i]);
		cnt[n1]+=cnt[v[n1][i]];
	}
	sum+=cnt[n1];
}
void dfs1(int n1,int s){
	ans=max(ans,s);
	for(int i=0;i<v[n1].size();i++){
		dfs1(v[n1][i],s+n-cnt[v[n1][i]]);
	}
}
signed main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int t;
	cin>>t;
	while(t--){
		cin>>n;
		for(int i=1;i<=n;i++) v[i].clear(),cnt[i]=0;
		for(int i=2;i<=n;i++){
			int a;
			cin>>a;
			v[a].push_back(i);
		}
		sum=0;
		dfs(1);
		ans=0;
		dfs1(1,sum);
		cout<<ans<<endl;
	}
    return 0;
}