wikioi1688 求逆序对

 

题目描述 Description

给定一个序列a1,a2,…,an，如果存在i<j并且ai>aj，那么我们称之为逆序对，求逆序对的数目

 

数据范围：N<=105。Ai<=105。时间限制为1s。

输入描述 Input Description

第一行为n,表示序列长度，接下来的n行，第i+1行表示序列中的第i个数。

输出描述 Output Description

所有逆序对总数.

样例输入 Sample Input

4

3

2

3

2

样例输出 Sample Output

3

 

感觉我又刷起水题了……

用归并排序线段树树状数组的都太low了

看我用treap搞之

每次询问比它大的有多少个，然后插入就好了

#include<cstdio>
#include<cstdlib>
#include<ctime>
#define LL long long
using namespace std;
struct SBT{
	int l,r,dat,rnd,rep,son;
}tree[200010];
int n,treesize,root;
LL ans;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
inline void update(int k)
{
	tree[k].son=tree[k].rep+tree[tree[k].l].son+tree[tree[k].r].son;
}
inline void right_rotate(int &k)
{
	int t=tree[k].l;
	tree[k].l=tree[t].r;
	tree[t].r=k;
	tree[t].son=tree[k].son;
	update(k);
	k=t;
}
inline void left_rotate(int &k)
{
	int t=tree[k].r;
	tree[k].r=tree[t].l;
	tree[t].l=k;
	tree[t].son=tree[k].son;
	update(k);
	k=t;
}
inline void insert(int &k,int x)
{
	if(!k)
	{
		k=++treesize;
		tree[k].dat=x;
		tree[k].rep=1;
		tree[k].son=1;
		return;
	}
	tree[k].son++;
	if (tree[k].dat==x)
	{
		tree[k].rep++;
		return;
	}
	if(x<tree[k].dat)
	{
		insert(tree[k].l,x);
		if(tree[tree[k].l].rnd>tree[k].rnd)right_rotate(k);
	}else
	{
		insert(tree[k].r,x);
		if(tree[tree[k].r].rnd>tree[k].rnd)left_rotate(k);
	}
}
inline int query(int k,int x)
{
	if (!k)return 0;
	if (tree[k].dat>x)return tree[k].rep+tree[tree[k].r].son+query(tree[k].l,x);
	if (tree[k].dat==x)return query(tree[k].r,x);
	if (tree[k].dat<x)return query(tree[k].r,x);
}
int main()
{
	n=read();
	for (int i=1;i<=n;i++)
	{
		int x=read();
		ans+=query(root,x);
		insert(root,x);
	}
	printf("%lld\n",ans);
}