题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1394
题意:求某种规定序列中的最小逆序数。
递推比较最小那部分比较简单,就不说了。
主要想说:求逆序数可以用构建线段树的方法来做。其实思想和计数排序的思想差不多。每次处理a[i]时,先统计一下已经被计数的前几个数的计数和。(比较的是值。)然后再更新这个计数和。这道题的数据范围和下标范围是一样的,所以可以直接做。
详见代码:
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <map>
#include <queue>
#include <algorithm>
using namespace std;
#define Maxn 5005
#define lx x<<1
#define rx ((x<<1) + 1)
#define MID ((l + r)>>1)
int A[Maxn];
int S[Maxn<<2];
void pushUp(int x)
{
S[x] = S[lx] + S[rx];
}
void build(int l,int r,int x)
{
if(l == r)
{
S[x] = 0;
return;
}
build(l,MID,lx);
build(MID+1,r,rx);
pushUp(x);
}
int query(int L,int R,int l,int r,int x)
{
int ans = 0;
if(L<=l && r<=R) return S[x];
if(L<=MID) ans += query(L,R,l,MID,lx);
if(R>=MID+1) ans += query(L,R,MID+1,r,rx);
return ans;
}
void update(int p,int d,int l,int r,int x)
{
if(l == r)
{
S[x] +=d;
return;
}
if(p<=MID) update(p,d,l,MID,lx);
else update(p,d,MID+1,r,rx);
pushUp(x);
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.txt","r",stdin);
#endif
int n;
int sum,ans,temp;
while(scanf(" %d",&n)!=EOF)
{
sum = ans = 0;
build(0,n-1,1);
for(int i=0;i<n;i++)
{
scanf(" %d",&A[i]);
temp = query(A[i],n-1,0,n-1,1);
sum += temp;
update(A[i],1,0,n-1,1);
}
ans = sum;
//递推求解,每次A[i]都在首位,则分为大于它的部分和小于它的部分。
for(int i=0;i<n;i++)
{
sum += n-1-A[i] - A[i];
ans = min(ans,sum);
}
printf("%d\n",ans);
}
return 0;
}
但是如果数据范围和下标范围不是一个范围的话,我们就不能直接用这个方法 做了。。试想,如果一个数有最大可以有十亿,难道我们要开一个十亿大小的数组来标记么。。这显然不现实。所以我们要对数据进行离散话,毕竟数据的数量达不到那么大。比如说这道题:http://poj.org/problem?id=2299。也是求逆序数。数据的范围特别大。但是数量比较少。非常适合离散化。
关于什么是离散话,参考:http://www.matrix67.com/blog/archives/108 和 http://baike.baidu.com/view/3392254.htm
那么离散化+线段树我们就可以做这道题了(不要忘了用long long,int存储不下最终结果):
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <map>
#include <queue>
#include <algorithm>
using namespace std;
#define Maxn 500005
#define lx x<<1
#define rx ((x<<1) + 1)
#define MID ((l + r)>>1)
#define LL long long
int rank[Maxn];
LL S[Maxn<<2];
struct Node
{
int val;
int id;
bool operator <(const Node &a) const
{
return val < a.val;
}
} A[Maxn];
void pushUp(int x)
{
S[x] = S[lx] + S[rx];
}
void build(int l,int r,int x)
{
if(l == r)
{
S[x] = 0;
return;
}
build(l,MID,lx);
build(MID+1,r,rx);
pushUp(x);
}
LL query(int L,int R,int l,int r,int x)
{
LL ans = 0;
if(L<=l && r<=R) return S[x];
if(L<=MID) ans += query(L,R,l,MID,lx);
if(R>=MID+1) ans += query(L,R,MID+1,r,rx);
return ans;
}
void update(int p,int d,int l,int r,int x)
{
if(l == r)
{
S[x] +=d;
return;
}
if(p<=MID) update(p,d,l,MID,lx);
else update(p,d,MID+1,r,rx);
pushUp(x);
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.txt","r",stdin);
#endif
int n;
LL ans = 0;
int temp;
while(scanf(" %d",&n)!=EOF)
{
ans = 0;
if(n == 0) break;
build(0,n-1,1);
for(int i=0;i<n;i++)
{
scanf(" %d",&A[i].val);
A[i].id = i;
}
sort(A,A+n);
for(int i=0;i<n;i++) rank[A[i].id] = i;
for(int i=0;i<n;i++)
{
temp = rank[i];
ans += query(temp+1,n-1,0,n-1,1);
update(temp,1,0,n-1,1);
}
printf("%lld\n",ans);
}
return 0;
}
另外,求逆序数也可以用树状数组和归并排序来做。
用线段树属于平衡树做法的一种。