PAT | T1027 Larry and Inversions

这里主要就是要注意：
对于某一个排列[i … j]，假设已知其逆序对数为X，将其reverse后的逆序对数为(i - j) * (i - j + 1) / 2
而对于某个排列[1,…,i,…,j,…,n]，假设已知其逆序对数为wholeSeq,将[i,…,j]reverse后，整个排列的逆序对数为wholeSeq - 2 * X + (i - j) * (i - j + 1) / 2;
附用线段树求某个区间逆序对数的模板

#include <iostream>

using namespace std;

const int maxn = 1010;

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;
}

int seq[maxn];
int inverseNum[maxn * maxn]; // [i,j]区间内的逆序对数为inverseNum[i * n + j]

int main(){
	int n;
	scanf("%d",&n);
	build(1,n);
	int wholeSeq = 0;
	for(int i = 0;i < n;i++){
		scanf("%d",&seq[i]);
		modify(seq[i],seq[i],1);
		inverseNum[i] = query(seq[i] + 1,n);
		wholeSeq += inverseNum[i];
	}
	for(int i = 0;i < n;i++){
		build(1,n);
		int total = 0;
		for(int j = i;j < n;j++){
			modify(seq[j],seq[j],1);
			int inverse = query(seq[j] + 1,n);
			total += inverse;
			int res = wholeSeq - 2 * total + (j - i) * (j - i + 1) / 2;
			printf("%d",res);
			if(i != n - 1 || j != n - 1) printf(" ");
		}
	}
	system("pause");
	return 0;
}