先说一下线段树求逆序数的原理:
从1到数列中最大的数n+1建树
然后从a[1]开始 对于每个i ,我先查询ai+1到n+1 (注意 这是数,不是下标)已经有多少个数存在于树中了,有多少个,那么与ai构成的逆序对数就是多少。cnt+=query(1,ai+1,n+1);然后插入。
最后cnt就是逆序对数
模板
#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;
const int maxn=1e5+5;
struct node{
int l,r,value;
}tree[maxn<<2];
void build(int m,int l,int r)
{
tree[m].l=l;
tree[m].r=r;
tree[m].value=0;
if(l==r)
return;
int mid=(l+r)>>1;
build(m<<1,l,mid);
build(m<<1|1,mid+1,r);
}
void updata(int m,int x)
{
if(tree[m].l==tree[m].r)
{
tree[m].value++;
return;
}
int mid=(tree[m].l+tree[m].r)>>1;
if(x<=mid) updata(m<<1,x);
else updata(m<<1|1,x);
tree[m].value++;
}
int query(int m,int l,int r)
{
if(l>r)return 0;
if(tree[m].l==l&&tree[m].r==r)
return tree[m].value;
int mid=(tree[m].l+tree[m].r)>>1;
int temp;
if(r<=mid)temp=query(m<<1,l,r);
else if(l>mid) temp=query(m<<1|1,l,r);
else temp=query(m<<1,l,mid)+query(m<<1|1,mid+1,r);
return temp;
}
int main()
{
int n,a[5005];
while(cin>>n&&n)
{
int cnt=0;
build(1,0,n+1);
for(int i=1;i<=n;i++)
{
cin>>a[i];
cnt+=query(1,a[i]+1,n+1);
updata(1,a[i]);
}
printf("%d\n",cnt);
}
}
对于POJ1394这道题
他是把数列头放到数列尾
然后求最小的逆序对数
我们只需要求完原序列的逆序对
遍历每个a[i] 如果a[i]作为序列首的时候 减小的逆序对数是a[i] ( 因为他是序列首,那么后面有a[i]个比他小的数),同理 增加的逆序对数是(n-1-a[i])
for(int i=1;i<=n;i++)
{
cnt=cnt-a[i]+(n-1-a[i]);
minn=min(minn,cnt);
}
离散化:
#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
#include<vector>
#define ll long long
using namespace std;
const int maxn=5e5+5;
struct node{
int l,r,value;
}tree[maxn<<2];
int n,m,a[500005],b[500005],save[500005],hashh[500005];
int getid(int x) {return lower_bound(save+1,save+m+1,x)-save;}
void build(int m,int l,int r)
{
tree[m].l=l;
tree[m].r=r;
tree[m].value=0;
if(l==r)
return;
int mid=(l+r)>>1;
build(m<<1,l,mid);
build(m<<1|1,mid+1,r);
}
void updata(int m,int x)
{
if(tree[m].l==tree[m].r)
{
tree[m].value++;
return;
}
int mid=(tree[m].l+tree[m].r)>>1;
if(x<=mid) updata(m<<1,x);
else updata(m<<1|1,x);
tree[m].value++;
}
int query(int m,int l,int r)
{
if(l>r)return 0;
if(tree[m].l==l&&tree[m].r==r)
return tree[m].value;
int mid=(tree[m].l+tree[m].r)>>1;
int temp;
if(r<=mid)temp=query(m<<1,l,r);
else if(l>mid) temp=query(m<<1|1,l,r);
else temp=query(m<<1,l,mid)+query(m<<1|1,mid+1,r);
return temp;
}
int main()
{
while(scanf("%d",&n)&&n)
{
ll cnt=0;
build(1,0,n+1);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
hashh[i]=a[i];
}
sort(hashh+1,hashh+n+1);
m=1;
save[1]=hashh[1];
for(int i=2;i<=n;i++) if(hashh[i]!=hashh[i-1]) save[++m]=hashh[i];
int size=m;
for(int i=1;i<=n;i++)
b[i]=getid(a[i]);
for(int i=1;i<=n;i++)
{
cnt+=query(1,b[i]+1,size+1);
updata(1,b[i]);
}
cout<<cnt<<endl;
}
}