线段树 csu1555 Inversion Sequence

题意:告诉你逆序对,还原逆序对


思路:首先我们来手动模拟一下样例

5
1 2 0 1 0
1的前面有1个比1大的,2的前面有2个比2大的,4的前面有一个比4大的

首先,1的位置可以直接确定,因为除了1以外所有的数都比1大,所以1肯定在第二个位置

得到 _ 1 _ _ _

然后考虑2,2的前面有两个数字比2大,因为我们已经确定的数字只有1,1已经比2小了,所以要无视1,其实就是说,2的前面要留下2个空位,这样那2个空位只能放3,4,5,就能保证2的前面一定会有2个数字比它大了,所以

得到 _ 1 _ 2 _

再考虑3,3的前面没有比3大的,1和2都比3小应该直接无视,那么3只能放在第一个位置

得到3 1 _ 2 _

再考虑4,4的前面有一个比它大,所以4的前面应该留一个空格

得到3 1 _ 2 4

最后

得到3 1 5 2 4


这下我们就能找到规律了,从1开始考虑到n,如果第i个数字的前面有A[i]个数比它大,那么第i个数字就应该放在上一个状态剩下的空位中的第A[i]+1的空位位置

然后问题就变成找空位位置了,可以用线段树去维护

如果左区间的空位数足够,就访问左树,否则,就在右树访问x-左区间的空位数

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<cstdio>
#include<string>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
 
using namespace std;
typedef long long LL;
typedef pair<LL, int> PLI;
 
const int MX = 2e4 + 5;
const int INF = 0x3f3f3f3f;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define root 1,n,1
#define RR rt<<1|1
#define LL rt<<1
 
int S[MX << 2], A[MX], B[MX];
 
void push_up(int rt) {
    S[rt] = S[LL] + S[RR];
}
 
void build(int l, int r, int rt) {
    if(l == r) {
        S[rt] = 1;
        return;
    }
 
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    push_up(rt);
}
 
int query(int x, int l, int r, int rt) {
    if(l == r) {
        S[rt] = 0;
        return l;
    }
 
    int m = (l + r) >> 1, ret;
    if(S[LL] >= x) ret = query(x, lson);
    else ret = query(x - S[LL], rson);
    push_up(rt);
    return ret;
}
 
int main() {
    int n;
    //freopen("input.txt","r",stdin);
    while(~scanf("%d", &n)) {
        build(root);
        for(int i = 1; i <= n; i++) {
            scanf("%d", &A[i]);
        }
 
        bool sign = true;
        for(int i = 1; i <= n; i++) {
            if(S[1] < A[i] + 1) {
                sign = false;
                break;
            }
            B[query(A[i] + 1, root)] = i;
        }
 
        if(!sign) printf("No solution\n");
        else {
            for(int i = 1; i <= n; i++) {
                printf("%d%c", B[i], i == n ? '\n' : ' ');
            }
        }
    }
    return 0;
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值