http://acm.hdu.edu.cn/showproblem.php?pid=1394
线段树求解逆序数
const int MAX = 5000;
int seg[MAX + 10];
struct node{
int l, r, sum;
int mid(){
return (l + r) >> 1;
}
}tree[MAX << 2];
void buildtree(int l, int r, int rt){
tree[rt].l = l;
tree[rt].r = r;
tree[rt].sum = 0;
if(l == r) return;
int mid = tree[rt].mid();
buildtree(l, mid, rt << 1);
buildtree(mid + 1, r, rt << 1 | 1);
}
void update(int l, int r, int rt, int id){
if(l == r){
tree[rt].sum++;
return;
}
int mid = tree[rt].mid();
if(id <= mid)
update(l, mid, rt << 1, id);
else
update(mid + 1, r, rt << 1 | 1, id);
tree[rt].sum = tree[rt << 1].sum + tree[rt << 1 | 1].sum;
}
int query(int l, int r, int rt){
if(tree[rt].l >= l && tree[rt].r <= r)
return tree[rt].sum;
int mid = tree[rt].mid(),ans = 0;
if(l >= mid)
ans += query(l, r, rt << 1 | 1);
else if(r < mid)
ans += query(l, r, rt << 1);
else{
ans += query(l, mid, rt << 1);
ans += query(mid+1, r, rt << 1 | 1);
}
return ans;
}
query部分求解【seg[i],n-1】中有多少点已经进入线段树中,因为这个区间中的数都比seg[i]先插入而且都大于seg[i],所以点数就是seg[i]的逆序数。
把这些点的逆序数全部相加就是整个序列的逆序数。
主函数部分求解逆序数之后,递推出所有情况的逆序数,选择一个最小的:
int main(){
// freopen("in.txt", "r", stdin);
int N;
while(scanf("%d",&N) != EOF){
buildtree(0, N-1, 1);
int sum = 0;
for(int i=0; i<N; i++){
scanf("%d",&seg[i]);
sum += query(seg[i], N-1, 1);
update(0, N-1, 1, seg[i]);
}
int ans = sum;
for(int i=0; i<N; i++){
sum = sum + N - 2 * seg[i] - 1;
if(sum < ans) ans = sum;
}
printf("%d\n",ans);
}
return 0;
}
因为序列为[0, n-1],若最前面一个数为x,序列中比x 小的数为[0, x-1], 共x个,比x大的数为[x+1, n-1], 共n-x-1个,将x移到最后,比x小的数的逆序数均减1, x的前面比x大的数有n-x-1个,x的逆序数增加n-x-1。 所以新序列的逆序数为原序列的逆序数加上n-2*x-1。