题目链接:
HDU 1394 Minimum Inversion Number
题意:
给出一组数,可以将每个数组首的数依次移到数组尾,求整个过程中(包括初始状态)的最少逆序对数。
分析:
用线段树求初始数组的逆序对数。对每个读入的数在线段树中查询之前有多少比它大的数字出现,那么这就是该数的逆序对数,累加就是整个初始数列的逆序对数。然后将读入的数插入到线段树叶子结点中,(指该数字已经出现了),线段树中需要记录区间已经出现数字个数。对于每个从数列首移到数列尾的移动,可以用 tmp+=((n−1−a[i])−a[i]) ;表示。对于每一个后移的 a[i] 会减少a[i]个逆序对(数据从0开始),同时增加 n−a[i]−1 个逆序对,因为比 a[i] 大的数字个数是 n−a[i]−1 个,过程取最小值即可。
也可以用归并排序做。
时间复杂度: O(nlogn) 。
//1484K 62MS
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <climits>
#define lson(x) (x<<1)
#define rson(x) ((x<<1)|1)
using namespace std;
const int maxn=5050;
int n,ans;
int data[maxn];
struct SegTree{
int left,right,num;//num是区间含有数字个数
}segtree[maxn<<2];
inline void build(int left,int right,int cur)
{
segtree[cur].left=left;
segtree[cur].right=right;
segtree[cur].num=0;
if(left==right) return ;
int mid=(left+right)>>1;
build(left,mid,lson(cur));
build(mid+1,right,rson(cur));
}
inline void query(int a,int b,int cur)
{
int left=segtree[cur].left;
int right=segtree[cur].right;
if(left==a&&right==b){
ans+=segtree[cur].num;
return;
}
int mid=(left+right)>>1;
if(b<=mid) query(a,b,lson(cur));
else if(a>mid) query(a,b,rson(cur));
else {
query(a,mid,lson(cur));
query(mid+1,b,rson(cur));
}
}
inline void update(int cur,int pos)//将叶子结点pos的num赋为1,表示pos已经出现了
{
int left=segtree[cur].left;
int right=segtree[cur].right;
if(left==right){//即:left=right=pos
segtree[cur].num=1;
return ;
}
int mid=(left+right)>>1;
if(pos<=mid) update(lson(cur),pos);
else update(rson(cur),pos);
segtree[cur].num=segtree[lson(cur)].num+segtree[rson(cur)].num;
}
int main()
{
//freopen("hdu1394in.txt","r",stdin);
while(~scanf("%d",&n)){
ans=0;
build(0,n-1,1);
for(int i=0;i<n;i++){
scanf("%d",&data[i]);
query(data[i],n-1,1);
update(1,data[i]);
}
int tmp=ans;
for(int i=0;i<n;i++){
tmp+=((n-data[i]-1)-data[i]);//对于每一个后移的a[i]会减少a[i]个逆序对(数据从0开始)
//同时增加n-a[i]-1个逆序对,因为比a[i]大的数字个数是n-a[i]-1个
ans=min(ans,tmp);
}
printf("%d\n",ans);
}
return 0;
}
//1440K 62MS
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=5050;
int n,ans;
int data[maxn],tmp[maxn],T[maxn];
void Merge_sort(int* tmp,int low,int high,int* T)
{
if(low+1<high){
int mid=(low+high)>>1;
Merge_sort(tmp,low,mid,T);
Merge_sort(tmp,mid,high,T);
int p=low,q=mid,i=low;
while(p<mid||q<high){
if(q>=high||(p<mid&&tmp[p]<tmp[q])){
T[i++]=tmp[p++];
}else {
T[i++]=tmp[q++];
ans+=(mid-p);
}
}
for(int i=low;i<high;i++){
tmp[i]=T[i];
}
}
}
int main()
{
freopen("hdu1394in.txt","r",stdin);
while(~scanf("%d",&n)){
ans=0;
for(int i=0;i<n;i++){
scanf("%d",&data[i]);
tmp[i]=data[i];
}
Merge_sort(tmp,0,n,T);
int res=ans;
for(int i=0;i<n;i++){
res+=(n-1-data[i]*2);
ans=min(ans,res);
}
printf("%d\n",ans);
}
return 0;
}