【线段树I:点修改+区间查询】hdu 1394 Minimum Inversion Number

最新推荐文章于 2019-02-25 23:17:34 发布

Hacker_vision

最新推荐文章于 2019-02-25 23:17:34 发布

阅读量704

点赞数 1

分类专栏：数据结构文章标签：线段树

本文链接：https://blog.csdn.net/u012717411/article/details/47263535

版权

数据结构专栏收录该内容

28 篇文章 0 订阅

订阅专栏

【线段树I:点修改+区间查询】hdu 1394 Minimum Inversion Number

题目链接：hdu 1394 Minimum Inversion Number

题目大意

给你N个数的排列，求包括该序列在内的所有循环序列的最小逆序对数（N<5000）

这里写图片描述

暴力的话O(n^3)直接超时,线段树一开始还真不好想啊！最关键的是弄明白每个节点中的sum代表什么意思：【lc,rc】区间出现的数的个数，因为这道题n个数大小0~n-1,所以用线段树是可以的，否则数的范围太大爆空间！

这里写图片描述

说一下思路

①首先建立一个空树根节点代表区间[0,n-1]，每插入一个节点a[i]之后查询区间[a[i],n-1]出现的数个数，即逆序对数目，再把这个节点更新到线段树中（叶子节点sum=1,所有父节点回溯），边输入边查询边更新
②上一步可以得到初始序列的逆序对，怎样得到所有循环序列的呢？我们发现，当把a[0]移走，逆序对减少a[0]对，放到最后增加n-1-a[0]个，一次类推，遍历一遍查找最小值即可！！是不是规律呢？！

【In addition】本题最关键是sum表示的是区间[lc,rc]出现数的个数，理解这一点，update 和 query 就很好解决了！

参考代码

/*============================================*\
|*              线段树求逆序对个数               *|
|*          线段树表示的区间是【0，n-1】          *|
\*============================================*/
/*Author:Hacker_vision*/
#include<bits/stdc++.h>
using namespace  std;

const int _max = 5e3 + 10;
int n,a[_max];

struct segTree{
  int lc,rc;
  int sum;//区间[lc,rc]中出现数的个数
}segTree[_max<<2];

void build(int root,int L,int R){//构建线段树,O(logN)
  segTree[root].lc = L;
  segTree[root].rc = R;
  segTree[root].sum = 0;//初始逆序对数目为0
  if(L == R) return;//遍历到叶子节点不需要再赋值了，因为之前已经初始化了
  int mid = (L + R) >> 1;
  build(root<<1,L,mid);
  build(root<<1|1,mid+1,R);
}

void update(int root,int pos){
  if(segTree[root].lc == segTree[root].rc) {
    segTree[root].sum ++;
    return;
  }
  int mid = (segTree[root].lc + segTree[root].rc) >> 1;
  if(pos <= mid) update(root<<1,pos);
  else update(root<<1|1,pos);
  segTree[root].sum = segTree[root<<1].sum + segTree[root<<1|1].sum;//回溯父节点
}

int query(int root,int L,int R){//查询[L,R]有多少逆序数
  if(segTree[root].lc == L && segTree[root].rc == R) return segTree[root].sum;//不一定递归到根节点
  int mid = (segTree[root].lc + segTree[root].rc) >> 1;
  if(R <= mid) return query(root<<1,L,R);
  else if(mid < L) return query(root<<1|1,L,R);
  else return query(root<<1,L,mid)+query(root<<1|1,mid+1,R);
}

int main(){
  #ifndef ONLINE_JUDGE
  freopen("input.txt","r",stdin);
  #endif // ONLINE_JUDGE
  while(scanf("%d",&n)==1){
    build(1,0,n-1);//建一棵空的线段树
    int res = 0;
    for(int i = 1; i <= n; ++ i){//边输入边查询边更新
        scanf("%d",a+i);
        res += query(1,a[i],n-1);//查询[a[i],n]出现过几个数，即为逆序对数；
        update(1,a[i]);//更新节点 //注意a[i]+1会越界,因为如果a[i]是最大的数，区间左端点超过右端
    }
    int ans = res;
    for(int i = 1; i <= n; ++ i){//根据规律,把a[i]移至尾部：减少a[i]、增加n-1-a[i]个逆序对，遍历一遍查找最小值即可
       res = res-a[i]+(n-1-a[i]);
       ans = min(ans ,res);
    }
    printf("%d\n",ans);//输出最小值
  }
  return 0;
}