解题思路:
递归求解,对于每个结点,我们只需要找到经过"左孩子右兄弟"变换后高度最高的子树即可,将其作为最右侧的孩子结点,取其他任意一个孩子结点作为左孩子结点后进行"左孩子右兄弟"的变换即可使得得到的树高度最高。
注意事项:
注意到结点数N范围是10^5,因此对于每个树节点,存储其孩子用动态数组vector,否则开不了数组。
同时因为用到了递归,为树节点设置一个height参量,存储每次递归返回前得到的变换后最高树高,这样可以减少递归次数。
参考代码:
#include<iostream> #include<vector> using namespace std; struct Node { vector<int> children;//存储子结点 int height=-1;//选取左孩子后的最高树高度 }; Node node[100001]; int max_height(int n) { if (node[n].children.size() == 0) return 0; int max = 0; int leftchild; for (int i = 0; i < node[n].children.size(); ++i) { if (max < max_height(node[n].children[i])) { max = node[node[n].children[i]].height; leftchild = node[n].children[i]; } } node[n].height = node[n].children.size() + max; return node[n].children.size() + max; } int main() { int N, i, f; cin >> N; for (i = 2; i <= N; ++i) { cin >> f; node[f].children.push_back(i); } cout << max_height(1); return 0; }
0.0分
3 人评分
Plurato 2022-04-07 19:44:01 |
是我打错了,对不起