CF1611D题解

lijinghui · 2024-01-22 14:24:22 · 题解

题解

题目传送门

分析

我们可以按照到根距离长短（ p 数组 ）构造。每次使该点（ p_i ）到根（ 整个树的 ）的距离为上一个（ p_{i-1} ）到根的距离加一个正整数（ 一般可以为 1 ，但不一定是 1 ，比如该节点的根到根的距离本身就很长 ) 。

期间，如果有一点比他的任意代根到根（ 整个树的 ）的距离小，则无法继续构造。

根节点到根的距离不是最小，也无法继续构造。（ 划重点，本人就在这栽过 ）

任意节点节点到其根的距离为该节点到根（ 整个树的 ）的距离减其根到根（ 整个树的 ）的距离。

小代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+10;
bool flag;
struct node
{
    int p,w,g,f,num,jl;
}a[N];
void run()
{
    flag=1;
    int n,sg;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i].g;
        if(a[i].g==i)sg=i;
        a[i].jl=-1;
    }
    for(int i=1;i<=n;i++)cin>>a[i].p;
    if(a[1].p!=sg)flag=0;
    a[sg].jl=0;
    a[sg].w=0;
    for(int i=2;i<=n;i++)
    {
        if(a[a[a[i].p].g].jl<0)flag=0;
        else
        {
            a[a[i].p].w=i-a[a[a[i].p].g].jl;
            a[a[i].p].jl=i;
        }
    }
    if(!flag)cout<<-1;
    else for(int i=1;i<=n;i++)cout<<a[i].w<<" ";
    cout<<endl;
    return;
}
int main()
{
    int t;
    cin>>t;
    while(t--)run(); 
    return 0;
 }