PAT | T1009 Triple Inversions

线段树求逆序对，线段树模板可以直接用，是我在AgOH大佬那里摘下来的，加了一些注释啥的，只有少数地方需要改动。

#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 ================================= 

ll larger[maxn],smaller[maxn]; // 这个数前面比它大的数的个数 & 这个数后面比它小的数的个数
int seq[maxn]; // 原始数组

int main(){
	// 以某个数为中间的逆序对个数为(这个数前面比它大的数的个数) * (这个数后面比它小的数的个数)
	// 这个题因为数就是从1~n的，所以不用离散化
	int n;
	scanf("%d",&n);
	long long res = 0;
	build(1,n); // 先建树
	for(int i = 1;i <= n;i++){
		scanf("%d",&seq[i]);
		modify(seq[i],seq[i],1); // [seq[i],seq[i]]区间加一
		larger[i] = query(seq[i] + 1,n); // 区间内数值之和
	}
	build(1,n); // 再重新建树
	for(int i = n;i >= 1;i--){ // 从后往前插入
		modify(seq[i],seq[i],1);
		smaller[i] = query(1,seq[i] - 1); // 区间内数值之和
	}
	for(int i = 2;i < n;i++){
		res += larger[i] * smaller[i];
	}
	printf("%lld",res);
	system("pause");
	return 0;
}