CF1611D Weights Assignment For Tree Edges

题意

T组测试用例
n个节点
第一行n个数表示b[i] 是 i 的父节点
第二行n个数表示p[i] 到根节点的距离是 第 i 小的

要求按给的排名给每条边赋值，根节点距离是 0 ，如果没有符合要求的 输出-1，如果有多种答案，输出任意其一。

思路

p[i] 到根节点的距离是 第 i 小的，那么就把它当成p[i] 到根节点的距离是i，再开一个d[i]数组，使d[p[i]] = i，意思是i 点到根节点的距离是d[i].
要想符合要求，那么肯定一个子节点到根节点的距离大于它的父节点到根节点的距离，否则就是-1。
最后按顺序输出d[i] - d[b[i]]，就可以按要求输出每条边的边权。
d[i] 是i到根节点的距离
i是子节点，b[i] 是父节点

#include <iostream>
#include <cstring>
#include <algorithm>using namespace std;
const int N = 2e5 + 10;
int b[N], p[N], d[N];
int main()
{
    int t; cin >> t;while (t -- ){
    int n; cin >> n;for (int i = 1; i <= n; i ++ ){
    cin >> b[i];}for (int i = 1; i <= n; i ++ ){
    cin >> p[i];d[p[i]] = i;}bool tt = 0;for (int i = 1; i <= n; i ++ ){
    if (d[i] < d[b[i]]) tt = 1;}if (tt == 1) cout << "-1" << endl;else{
    for (int i = 1; i <= n; i ++ ){
    cout << d[i] - d[b[i]] << " ";}cout << endl;}}return 0;
}