链接:https://ac.nowcoder.com/acm/contest/5758/I
来源:牛客网
题目描述
小明遇到了一个问题希望你能帮他解决
现在有n个数字排成一列构成数组A,数组A中存在n个数a[i], 其中1<=i<=n。
数组sj为删除数组A中的第j个数后,剩余n-1个数构成的数组,其中1<=j<=n。
小明希望你把s1~sn的数组按照字典序大小排列起来,
若两个数组相等,则认为删除元素编号小的数组字典序更小
输入描述:
输入数据第一行是t,表示数据的组数,接下来每组数据输入n,接下来一行
一共n个数,a[i]表示数组的第i个数
(t<=10,n <= 1e5,a[i] <= 1e9)
输出描述:
输出一行n个整数,b1,b2…bn,表示Sb1 < Sb2 < … < Sbn
示例1
输入
复制
1
7
1 1 2 1 1 1 2
输出
复制
3 7 4 5 6 1 2
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 1e5 + 7;
int a[maxn], f[maxn], ans[maxn];
bool cmp(int x, int y){
if(x > y)return !cmp(y, x);
if(f[x + 1] <= y && a[f[x + 1]] > a[f[x + 1] - 1])
return false;
else
return true;
}
int main(){
int t;
scanf("%d", &t);
while(t--){
int n;
scanf("%d", &n);
for(int i = 1; i <= n; ++i){
scanf("%d", &a[i]);
ans[i] = i;
}
int tmp = n + 1;
for(int i = n; i >= 1; --i){
if(a[i] != a[i - 1]){
tmp = i;
}
f[i] = tmp;
}
sort(ans + 1, ans + 1 + n, cmp);
for(int i = 1; i <= n; ++i){
printf("%d%c", ans[i], " \n"[i == n]);
}
}
}