线段树 csu1555 Inversion Sequence

来源:互联网 发布:网络电子游戏官网 编辑:程序博客网 时间:2024/04/30 09:19

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


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

51 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