2020 Multi-University Training Contest 10

A.

题意：

跟节点是1.
第一行输入一个n，代表n个节点，第二行输入n-1个数，代表2~n节点的父亲。
如果节点x能走到节点y，点对数+1，只能走向自己的儿子，不能走向父亲。（即单向边）
你可以增加一条双向边，求增加一条边后可到达的最大点对数。

思路

加边后构成的环肯定可以两两到达，所以肯定是选一个叶子节点连接跟节点，在原有点对数的基础上加上增加的点对数。转化为求哪个叶子节点连跟节点后增加的点对数最多。

步骤

1、dfs记录每个节点的叶子节点
2、dfs2遍历树，记录当前路径增加的对点数==n-该节点儿子个数。
3、加上原来就有的点对数

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define mp make_pair
#define pb push_back
#define T int T; scanf("%d",&T);while(T--)
const ll mod=1e9+7;
const int maxn = 1e6+10;
struct edge{
	int to;
	int nxt;
}e[maxn];
int cnt;
int n;
int head[maxn];
int f[maxn];
int child[maxn];	//儿子个数
ll ans;
void add(int u, int v){
	e[cnt].to = v;
	e[cnt].nxt = head[u];
	head[u] = cnt ++;
}
void dfs(int u){	//记录每个节点儿子个数
	child[u] = 1;
	for(int i = head[u]; i != -1; i = e[i].nxt){
		int v = e[i].to;
		dfs(v);
		child[u] += child[v];
	}
}
void dfs2(int u, ll sum){		//sum记录当前路径可增加的点对数
	// cout << u << " " << sum << endl;
	if(child[u] == 1){
		ans = max(ans, sum);
		return;
	}
	for(int i = head[u]; i != -1; i = e[i].nxt){
		int v = e[i].to;
		dfs2(v,sum + n - child[v]);
	}
}
int main(){
	T{
		cnt = 0;
		ans = 0;
		scanf("%d", &n);
		for(int i = 0; i <= n; i ++){
			head[i] = -1;
		}
		for(int i = 2; i <= n; i ++){
			int v;
			scanf("%d", &v);
			add(v,i);
		}		
		dfs(1);
		dfs2(1,0);
		for(int i = 1; i <= n; i ++){
			ans += child[i];
		}
		printf("%lld\n", ans);
	}

	return 0;
}