题目大意:
给定一个0到n-1的数字组成的序列,它的逆序数,然后把第一个数字放到末尾,得到一个新的序列,再求逆序数,再把新序列的第一个数字放到末尾,一直这样做,求所有这些序列的逆序数的最小值。
思路:
首先用权值线段树(或树状数组)求出当前序列的逆序对数。然后把第一个数放到最后一个,计算他的贡献:即他当前能组成的逆序数对为后面比他小的数(a[i]个),把他放入最后时,他贡献的逆序对数为:(n-a[i]+1)即他前面比他小的元素个数。所以他的贡献为:n-a[i]+1-a[i]。
#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
const int N=5000+10;
struct node
{
int l,r,sum;
}tree[N*4];
void build(int dep,int l,int r)
{
tree[dep].l=l;
tree[dep].r=r;
tree[dep].sum=0;
if(l==r)
{
return ;
}
int mid=(l+r)>>1;
build(dep<<1,l,mid);
build(dep<<1|1,mid+1,r);
}
void update(int dep,int x)
{
if(tree[dep].l==x&&tree[dep].r==x)
{
tree[dep].sum++;
return ;
}
int mid=(tree[dep].l+tree[dep].r)>>1;
if(mid>=x)
update(dep<<1,x);
else
update(dep<<1|1,x);
tree[dep].sum=tree[dep<<1].sum+tree[dep<<1|1].sum;
return ;
}
int query(int dep,int l,int r)
{
if(tree[dep].l==l&&tree[dep].r==r)
{
return tree[dep].sum;
}
int mid=(tree[dep].l+tree[dep].r)>>1;
if(mid>=r)
return query(dep<<1,l,r);
else if(mid<l)
return query(dep<<1|1,l,r);
else
return query(dep<<1,l,mid)+query(dep<<1|1,mid+1,r);
}
int a[N];
int main()
{
int n;
while(~scanf("%d",&n))
{
memset(a,0,sizeof(a));
int ans=0;
build(1,1,n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
a[i]++;
ans+=query(1,a[i],n);
update(1,a[i]);
}
// printf("%d\n",ans);
int sum=ans;
for(int i=1;i<=n;i++)
{
ans+=(n-a[i]+1-a[i]);
sum=min(ans,sum);
}
printf("%d\n",sum);
}
return 0;
}