Codeforces Round #702 (Div. 3) E. Accidental Victory(二叉树的中序遍历)

https://codeforces.com/contest/1490/problem/D

从1到n,其中所有的数字恰好出现一次。

坡旅甲最近得到了一个长度为n的排列a[1…n]。坡旅甲喜欢树胜过排列,所以他想把排列a转换成一棵有根二叉树。他将不同整数的数组转换成一棵树,如下所示:

数组的最大元素成为树的根;
最大值左侧的所有元素—形成左子树(根据相同的规则构建,但应用于数组的左侧部分),但如果最大值左侧没有元素,则根没有左子树;
最大值右侧的所有元素—形成一个右子树(根据相同的规则构建,但应用于数组的右侧),但是如果最大值右侧没有元素,则根没有右子树。

求出每个数字的所在深度。
input 
2
4
1 2 4 3
5
1 1 1 1 1
output 
3
2 3 4 
5
1 2 3 4 5 
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const LL N=100200,M=2002;
int a[N];
int res[N];
void dfs(int l,int r,int depth)
{
    if(l>r) return ;//结束条件
    int idx=l;
    for(int i=l;i<=r;i++)
        if(a[i]>a[idx]) idx=i;//找到最大值的下标
    res[idx]=depth;//标记该点的深度
    dfs(l,idx-1,depth+1);//更改右边界
    dfs(idx+1,r,depth+1);//更改左边界
}
int main()
{
    cin.tie(0); cout.tie(0); ios::sync_with_stdio(false);
    LL T=1;
    cin>>T;
    while(T--)
    {
        LL n;
        cin>>n;
        for(int i=1;i<=n;i++)
            cin>>a[i];
        dfs(1,n,0);//从下标1到下标n中查找,开始先放深度为0的
        for(int i=1;i<=n;i++)
            cout<<res[i]<<" ";
        cout<<endl;
    }
    return 0;
}
posted @ 2022-09-05 01:03  Vijurria  阅读(24)  评论(0编辑  收藏  举报