PAT | T1010 Lehmer Code

和1009一样的，直接套我1009的模板就行，嘻嘻

#include <iostream>

using namespace std;

typedef long long ll;

// ================================= Segment Tree ================================= 

const int maxn = 1e6 + 5;

struct Node{
    int l,r,f,val; // val->区间和 f->懒惰标记，区间加几f就等于几 l,r->对应的区间为[l,r]
}sgt[maxn * 4];
inline int ls(int k){return k << 1;} // 左孩子
inline int rs(int k){return k << 1 | 1;} // 右孩子
inline void pushup(int k){sgt[k].val = (sgt[ls(k)].val + sgt[rs(k)].val);} // 该节点的val = 左孩子的val + 右孩子的val
inline void pushdown(int k){
    sgt[ls(k)].f += sgt[k].f;
    sgt[rs(k)].f += sgt[k].f;
    sgt[ls(k)].val += (sgt[ls(k)].r - sgt[ls(k)].l + 1) * sgt[k].f; // 整个区间的和要+=区间元素的个数*f
    sgt[rs(k)].val += (sgt[rs(k)].r - sgt[rs(k)].l + 1) * sgt[k].f;
    sgt[k].f = 0;
}
void build(int l,int r,int k = 1){ // 原数组中[l,r]以sgt[k]为根build线段树
    sgt[k].l = l,sgt[k].r = r;
    if(l == r){
        sgt[k].val = 0;
        return;
    }
    int m = (l + r) >> 1;
    build(l,m,ls(k)); // 递归建树
    build(m + 1,r,rs(k));
    pushup(k);
}
void modify(int x,int y,int z,int k = 1){ // 令[x,y]区间内数值全部加z
    int l = sgt[k].l,r = sgt[k].r;
    if(x <= l && y >= r){
        sgt[k].f += z; // 懒惰标记，表示sgt[k]的子树内的元素全部加z
        sgt[k].val += (r - l + 1) * z; // 更新sgt[k]的部分和
        return;
    }
    if(sgt[k].f) pushdown(k); // 若这个区间不完全包括要改的区间，先pushdown
    int m = (l + r) >> 1;
    if(x <= m) modify(x,y,z,ls(k));
    if(y > m) modify(x,y,z,rs(k));
    pushup(k);
}
int query(int x,int y,int k = 1){ // 查询[x,y]区间内数值之和
    int l = sgt[k].l,r = sgt[k].r;
    if(x <= l && y >= r) return sgt[k].val;
    if(sgt[k].f) pushdown(k);
    int sum = 0,m = (l + r) >> 1;
    if(x <= m) sum += query(x,y,ls(k));
    if(y > m) sum += query(x,y,rs(k));
    return sum;
}

// ================================= Segment Tree ================================= 

int smaller[maxn]; // 这个数前面比它大的数的个数 & 这个数后面比它小的数的个数
int a[maxn]; // 原始数组
int seq[maxn]; // 离散化后数组


// ================================== Discretize ================================== 
struct N{
    int val,order;
}temp[maxn];

#include <algorithm>

bool cmp(N a,N b){
	return a.val < b.val;
}
void discretize(int n){ // 离散化
    for(int i = 1;i <= n;i++){
        temp[i].val = a[i];
        temp[i].order = i;
    }
    sort(temp + 1,temp + n + 1,cmp);
    for(int i = 1;i <= n;i++){
        seq[temp[i].order] = i;
    }
}

// ================================== Discretize ================================== 

// 逆序对数
void inverseNum(int n){
    build(1,n);
    for(int i = n;i >= 1;i--){
		modify(seq[i],seq[i],1); // [seq[i],seq[i]]区间加一
		smaller[i] = query(1,seq[i] - 1); // 区间内数值之和
	}
}

int main(){
	int n;
	scanf("%d",&n);
	for(int i = 1;i <= n;i++){
		scanf("%d",&a[i]);
	}
	discretize(n);
	inverseNum(n);
	for(int i = 1;i <= n;i++){
		printf("%d",smaller[i]);
		if(i != n) printf(" ");
	}
	system("pause");
	return 0;
}