蓝桥杯左孩zi右兄弟

题目：[P1560 - 蓝桥杯2021初赛] 左孩子右兄弟 - New Online Judge (ecustacm.cn)

思路：

• 题意很简单，对于每个结点做这样的操作：
• 选任意顺序将所有子节点中任选作为左结点，将该结点的兄弟结点依次在右侧连接，最后形成一个二叉树
• 要求我们求出一颗给定树改为二叉树后最大的高度为多少？
• 由于父节点不能改变，所以得到的二叉树根节点也还是1
• 假设以1为父节点的所有子节点中孩子结点最多的那个结点为x
• 显然我们需要将x结点放在他所有兄弟结点的最下面
• 设size[i]表示以i为父节点的结点数，length[x]表示以结点x为根节点的树的高度
• 此时高度就变为size[1]+length[x]
• 于是问题变为去寻找以x为根节点的二叉树的高度，而这显然与寻找1为根节点的树的高度相同，于是考虑递归dfs
• vector中存放的是以i为父节点的所有结点，p[k][i]是以k为父节点的第i个结点

#include<cmath>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
const ll N = 100005;
/*

* 题意很简单，对于每个结点做这样的操作：
* 选任意顺序将所有子节点中任选作为左结点，将该结点的兄弟结点依次在右侧连接，最后形成一个二叉树
* 要我们求出一颗给定树改为二叉树后最大的高度为多少？
* 由于父节点不能改变，所以得到的二叉树根节点也还是1
* 假设以1为父节点的所有子节点中孩子结点最多的那个结点为x
* 显然我们需要将x结点放在他所有兄弟结点的最下面
* 设size[i]表示以i为父节点的结点数，length[x]表示以结点x为根节点的树的高度
* 此时高度就变为size[1]+length[x]
* 于是问题变为去寻找以x为根节点的二叉树的高度，而这显然与寻找1为根节点的树的高度相同，于是考虑递归dfs
* vector中存放的是以i为父节点的所有结点，p[k][i]是以k为父节点的第i个结点
  */
  vector<int>p[N];
  int dfs(int k)
  {
      if (p[k].size() == 0)return 0;
      int maxx = 0;
      for (int i = 0; i < p[k].size(); ++i)
  	    maxx = max(maxx, dfs(p[k][i]));
      return maxx+p[k].size();
  }
  int main()
  {
      int n;
      cin >> n;
      for (int i = 2; i <= n; ++i)
      {
      	int a; cin >> a;
     	p[a].push_back(i);
      }
      cout<<dfs(1);
      return 0;
  }

/*
* 链式前向星存图，好处在于图的起点终点清晰，很容易获得思路。劣处是无法直接获取该节点有多少子节点，需要提前处理，这里用b[i]表示i的子节点数量。总体而言能用vector简单做还是选择vector更好。
* ps:树是有向边，只需要add(x,y)一次就可以
*/

#include<cmath>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
const ll N = 100005;
int head[N ], to[N ],nex[N ],cnt,b[N];
void add(int x, int y)
{
	to[++cnt] = y;
	nex[cnt] = head[x];
	head[x] = cnt;
}
int dfs(int x,int y)
{
	int maxx = 0;
	for (int i = head[x]; i != 0; i=nex[i])
	{
		if (to[i] == y)continue;
		maxx = max(maxx, dfs(to[i],x));
	}
	return maxx+b[x];
}
int main()
{
	int n;
	cin >> n;
	int a;
	for (int i = 2; i <= n; ++i)
	{
		cin >> a;
		add(a, i);
		b[a]++;
	}
	cout<<dfs(1, 0);
	return 0;
}