蓝桥杯2021省 A——左孩子右兄弟

蓝桥杯官网

洛谷

 [蓝桥杯 2021 省 A] 左孩子右兄弟

题目描述

对于一棵多叉树，我们可以通过“左孩子右兄弟”表示法，将其转化成一棵二叉树。

如果我们认为每个结点的子结点是无序的，那么得到的二叉树可能不唯一。换句话说，每个结点可以选任意子结点作为左孩子，并按任意顺序连接右兄弟。

给定一棵包含 N 个结点的多叉树，结点从 1 至 N 编号，其中 1号结点是根，每个结点的父结点的编号比自己的编号小。

请你计算其通过“左孩子右兄弟”表示法转化成的二叉树，高度最高是多少。注：只有根结点这一个结点的树高度为 0。

输入描述

输入的第一行包含一个整数 N 。以下 N-1 行, 每行包含一个整数, 依次表示 2 至 N 号结点的父结点编号。

输出描述

输出一个整数表示答案。

输入输出样例

输入
5
1
1
1
2

输出
4

#include <bits/stdc++.h>
using namespace std;
struct node
{
  int parent;
  stack<int> children;
  int height;
};
int main()
{
  int n = 0;
  cin >> n;
  node *tree = new node[n + 1];
  for(int i = 0; i < n + 1; i++)
    tree[i].height = 0;
  tree[0].parent = tree[1].parent = -1;
  for(int i = 2; i < n + 1; i++)
  {
    cin >> tree[i].parent;
    tree[tree[i].parent].children.push(i);
  }
  for(int i = n; i > 0; i--)
  {
    if(tree[i].children.empty())
      tree[i].height = 1;
    else
    {
      int num = tree[i].children.size();
      while(!tree[i].children.empty())
      {
        tree[i].height = max(tree[i].height, tree[tree[i].children.top()].height);
        tree[i].children.pop();
      }
      tree[i].height += num;
    }
  }
  cout << tree[1].height - 1 << endl;
  delete[] tree;
  return 0;
}