HDU-1394 线段树+逆序数

最新推荐文章于 2021-02-21 19:10:28 发布

一生何求L

最新推荐文章于 2021-02-21 19:10:28 发布

阅读量105

点赞数 1

分类专栏：线段树

本文链接：https://blog.csdn.net/weixin_45948204/article/details/105167900

版权

线段树专栏收录该内容

3 篇文章 0 订阅

订阅专栏

思路
查询该数到n已经插入的个数，再单点更新该数sum[root]=1，最后求出所有的逆序数的个数。
每次换到后面，说明小于a[i]有a[i]个数大于a[i]有n-a[i]-1个数说明少了有a[i]个逆序数多了n-a[i]-1个逆序数。

import java.util.Scanner;

public class Main_HDU1394 {
	static int n;
	static int[] a = new int[5005];
	static int[] sum = new int[5005<<2];
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		while(sc.hasNext()) {
			n = sc.nextInt();
			build(1,n,1);
			int res = 0;
			for(int i=1;i<=n;i++) {
				a[i] = sc.nextInt();
				res += query(a[i]+1,n,1,n,1);
				update(a[i]+1,1,n,1);
			}
			int ans = 99999999;
			for(int i=1;i<=n;i++) {
				res = res-a[i]+n-a[i]-1;  //将a[i]换到后面    小于a[i]有a[i]个数  大于a[i]有n-a[i]-1个数  说明少了有a[i]个逆序数  多了n-a[i]-1个逆序数  
				ans = Math.min(ans, res);
			}
			System.out.println(ans);
		}	
	}
	public static void build(int l,int r,int root) {
		sum[root] = 0;
		if(l==r) return;
		int m = (l+r)>>1;
		build(l,m,root<<1);
		build(m+1,r,root<<1|1);
	}
	public static void update(int x,int l,int r,int root) {
		if(l==r) {
			sum[root] = 1;    //插入该数
			return;
		}
		int m = (l+r)>>1;
		if(x<=m) 
			update(x,l,m,root<<1);
		else 
			update(x,m+1,r,root<<1|1);
		sum[root] = sum[root<<1] + sum[root<<1|1];
	}
	public static int query(int x,int y,int l,int r,int root) {
		if(x<=l&&r<=y) {
			return sum[root];   //查询[a[i]+1,n]已经插入的个数
		}
		int m = (l+r)>>1;
		int res = 0;
		if(x<=m)
			res += query(x,y,l,m,root<<1);
		if(y>m)
			res += query(x,y,m+1,r,root<<1|1);
		return res;
	}
}