D. Weights Assignment For Tree Edges(11.27)

Problem - D - Codeforces

题意：给你数组a表示i的父结点是a[i]，给你数组b表示结点b[i]到根结点的距离在所有结点里排第i（每个结点到根结点的距离是不同的），让你找符合条件的数组w(表示结点i到i的父结点的距离为w[i])，且w[i]>w[a[i]]。

分析：不妨让b[i]的值为结点i到根结点的距离，（b[i]表示距离排名，且b数组每个值都不一样，变相的告诉你到结点到根结点的距离，不用白不用），所以每个结点到父结点的距离就是该结点到根的距离-父结点到根的距离。

唬人题，题目说的好复杂，看不懂题(´-ι_-｀)，思路不算复杂。

附一张测试样例1图解：

#include<iostream>
using namespace std;
int a[200005],b[200005],w[200005];
int main()
{
    int t,n,root,x;
    cin>>t;
    while(t--)
    {
        int f=1;
        cin>>n;
        for(int i=1; i<=n; i++)
        {
            cin>>a[i];
            if(a[i]==i)
                root=i;
        }
        for(int i=1;i<=n;i++)
        {
            cin>>x;
            b[x]=i;
        }
        for(int i=1;i<=n;i++)
        {
            if(b[a[i]]>b[i])//如果i的父亲的边的权值更大，不符合题意。
            {
                cout<<-1<<endl;
                f=0;
                break;
            }
        }
        if(f==1)
        {
            for(int i=1;i<=n;i++)
                cout<<b[i]-b[a[i]]<<" ";//自己到祖先的距离减父亲到祖先的距离
            cout<<endl;
        }
    }
    return 0;
}