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;
}