JAVA程序设计：翻转对（LeetCode：493）

给定一个数组 nums ，如果 i < j 且 nums[i] > 2*nums[j] 我们就将 (i, j) 称作一个重要翻转对。

你需要返回给定数组中的重要翻转对的数量。

示例 1:

输入: [1,3,2,3,1]
输出: 2
示例 2:

输入: [2,4,3,5,1]
输出: 3
注意:

给定数组的长度不会超过50000。
输入数组中的所有数字都在32位整数的表示范围内。

思路：其实和求数组的逆序对数目是一个道理，所以我只放两种非常常见的方法的代码：

方法一：归并排序求解

class Solution {
    public int reversePairs(int[] nums) {
        return MergeSort(nums,0,nums.length-1);
    }
    
    private int MergeSort(int[] nums,int l,int r) {

    	if(l>=r) return	0;
    	
    	int mid=(l+r)/2,count=0;
    	count+=MergeSort(nums,l,mid);
    	count+=MergeSort(nums,mid+1,r);
    	int j=mid+1;
    	for(int i=l;i<=mid;i++) {
    		while(j<=r && nums[i]>(long)nums[j]*2)
    			j++;
    		count+=j-(mid+1);
    	}
    	MySort(nums,l,r);
    	return count;
    }
    
    private void MySort(int[] nums,int l,int r) {
    	//System.out.println(l + " "+ r);
    	if(l==r) return;
    	int[] a=new int[r-l+1];
    	int mid=(l+r)/2,p1=l,p2=mid+1,len=0;
    	while(p1<=mid && p2<=r) {
    		if(nums[p1]<=nums[p2]) {
    			a[len++]=nums[p1++];
    		}
    		else
    			a[len++]=nums[p2++];
    	}
    	while(p1<=mid) {
    		a[len++]=nums[p1++];
    	}
    	while(p2<=r) {
    		a[len++]=nums[p2++];
    	}
    	for(int i=l;i<=r;i++)
    		nums[i]=a[i-l];
    }
}

方法二：树状数组

class Solution {
	
	private int[] c;
	
    public int reversePairs(int[] nums) {
    	
    	int n=nums.length;
    	c=new int[n+1];
    	List<Integer> list=new ArrayList<>();
    	
    	for(int i=0;i<nums.length;i++)
    		list.add(nums[i]);
    	
    	Collections.sort(list);
    	
    	int ans=0;
    	for(int i=0;i<n;i++) {
    		ans+=query(n,find(list,(long)2*nums[i]+1)+1);
    		update(find(list,(long)nums[i])+1,1);
    	}
    	
    	return ans;
    }
    
    private void update(int index,int val) {
    	while(index>0) {
    		c[index]+=val;
    		index-=index&(-index);
    	}
    }
    
    private int query(int n,int index) {
    	int res=0;
    	while(index<=n) {
    		res+=c[index];
    		index+=index&(-index);
    	}
    	return res;
    }
    
    private int find(List<Integer> list,long x) {
    	int l=0,r=list.size()-1,p=r+1;
    	while(l<=r) {
    		int mid=(l+r)/2;
    		if(list.get(mid)>=x) {
    			p=mid;
    			r=mid-1;
    		}
    		else
    			l=mid+1;
    	}
    	return p;
    }
}