思维题 D. Weights Assignment For Tree Edges

题意

这题我是真的犯病,读了一个小时的假题目
就是有两行数
fa[i]表示第i个点的父节点(根节点的父节点是自己)
rankl[i]表示离根节点第i个近的点
如果能够成立
要求输出 第i个点和其父节点的距离
else
输出-1

思路

创建一个数组d[i]表示 第i个点到根节点的距离
d[rankl[i]]=i-1
然后 如果 d[i]-d[fa[i]]<0那么就不可能成立

AC代码

#include<bits/stdc++.h>
#define endl "\n"
#define INF 0x3f3f3f3f3f3f3f
typedef long long ll;
const ll mod = 1e9 + 7;
const double PI = acos(-1.0);
const double EI = exp(1.0);
const int N = 1e6 + 50;
const int maxn = 2 * N;
const double eps = 1e-8;
using namespace std;

int fa[N];
vector<int>son[N];
int rankl[N];
int d[N];
void solve()
{
	int n;
	int root = 0,flag=1;
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> fa[i];
	}
	for (int i = 1; i <= n; i++)
	{
		cin >> rankl[i];
		d[rankl[i]] = i;
	}
	for (int i = 1; i <= n; i++)
	{
		if (d[i] - d[fa[i]] < 0)
			flag = 0;
	}
	if (flag)
	{
		for (int i = 1; i <= n; i++)
		{
			cout << d[i] - d[fa[i]] << " ";
		}
	}
	else
	{
		cout << -1;
	}
	cout << endl;
}
int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(); cout.tie(0);
	int t; cin >> t; while (t--)
		solve();
	return 0; 
}