CodeForces 501D Misha and Permutations Summation 【树状数组】

题目：点击打开链接

题意：给出长度n的两个数列，求字典序编号为这两个数列的字典序编号之和模n!的数列

思路：观察数列，可以发现一个数列的字典序编号等于从左往右每一位数的数字减去前面已经出现过的数的个数，然后再乘上各位右边剩余的数的个数的阶乘，我们很容易想到用树状数组保存前面出现的数并且快速计算出其个数，因为树状数组不能存零，所以每一个数先加一，再存进去，开一个ans数组储存每一位数的数字减去前面出现过的数的个数，然后将两次数列的ans对应相加，观察发现，ans数组的每一位的值，不能超过其位数（位数从右向左由一开始），所以我们从ans右端开始从右往左取模进位，得到答案的ans数组。然后我们考虑二分还原数列，方法类似上述过程的逆过程，从左往右根据该位ans的值二分，找到满足条件的最大的值，减一即为该位的答案值。

代码：

#include <stdio.h>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
#include <set>
#include <vector>
#include <map>
#define PR pair<int,int>
#define MP make_pair
#define fi first
#define se second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define sqr(x) ((x)*(x))
#define ll long long
const ll INF = 1e18;
const int inf=0x3f3f3f3f;
const int M=18;
const int N=200100;
const int MOD=100000;
const double eps=1e-10;
const double pi=acos(-1.0);
using namespace std;

int n;
int s[N],ans[N];

void add(int x,int v)
{
    for(int i=x;i<=n;i+=i&(-i))
        s[i]+=v;
}
int query(int x)
{
    int ans=0;
    for(int i=x;i;i-=i&(-i))
        ans+=s[i];
    return ans;
}

int main()
{
    int i,x;
    scanf("%d",&n);
    for(i=1;i<=n;i++) add(i,1);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&x);
        int tmp=query(++x)-1;
        ans[i]+=tmp;
        add(x,-1);
    }
    for(i=1;i<=n;i++) add(i,1);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&x);
        int tmp=query(++x)-1;
        ans[i]+=tmp;
        add(x,-1);
    }
    for(i=n;i>0;i--)
    {
        ans[i-1]+=ans[i]/(n-i+1);
        ans[i]=ans[i]%(n-i+1);
    }
    for(i=1;i<=n;i++) add(i,1);
    for(i=1;i<=n;i++)
    {
        int l=1,r=n,tmp;
        while(l<r)
        {
            int m=(l+r)/2;
            tmp=query(m);
            if(tmp<ans[i]+1) l=m+1;
            else r=m;
        }
        add(l,-1);
        printf("%d ",l-1);
    }
}