牛客寒假训练营4--F

题目传送门
这个题其实是一道送分题…可是我被卡在神(gai)奇(si)的E题上了,没有时间写这个水题了…
E题是如此的神(e)奇(xin)以至于我实在是不想给它单独写一篇总结… (不就是一个贪心+高精)

读一遍题,稍稍画图看一下就知道是要你求树上有几对点之间的距离为偶数在稍微观察一下就能发现,深度为奇数的节点到深度为奇的节点之间的距离一定为偶,同理深度为偶的也是(当个小知识点记着吧,说不定什么时候能救命?)然后这题就可以水水过去了

AC代码:

#include <iostream>
#include <queue>
#include <cstring>
#define MAX 1000010 
#define ll long long 
using namespace std;
ll num;
int c[MAX];
struct node{
	int to;
	int nxt;
}edge[MAX];

int head[MAX];
int cnt=0;

void addedge(int from,int to)
{
	edge[cnt]={to,head[from]};
	head[from]=cnt++;
}

void bfs(int s)
{
	queue<int> que;
	que.push(s);
	c[s]=1;
	while(!que.empty())
	{
		int t=que.front();
		que.pop();
		for(int i=head[t];i!=-1;i=edge[i].nxt)
		{
			int to=edge[i].to;
			c[to]=c[t]+1;
			que.push(to);
		}
	}
}

void init()
{
	
	memset(head,-1,sizeof(head));
}

int main()
{
	init();
	ll ans=0;
	int n;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int a;
		cin>>a;
		addedge(a,i+1);
	}
	bfs(1);
	for(int i=1;i<=n;i++)
	{
		if(c[i]%2==1)
		num++;
	}
	ans+=num*(num-1);
	num=n-num;
	ans+=(num-1)*num;
	cout<<ans;
	return 0;
	
}

和神(e)奇(xin)的E题比,这题简直是天使啊 (可惜没抓住)