题目描述
有一组数,对于其中任意两个数组,若前面一个大于后面一个数字,则这两个数字组成一个逆序对。请设计一个高效的算法,计算给定数组中的逆序对个数。
给定一个int数组A和它的大小n,请返回A中的逆序对个数。保证n小于等于5000。
测试样例:
[1,2,3,4,5,6,7,0],8
返回:7
思路:
数组的逆序对就是求————(当前数组长度 减去 秩)————,与上题(维护x的秩)一样,维护一组二叉查找树,每次添加新值后得到该数组对应的秩,用当前数组长度-秩
对所有值的添加过程都进行逆序对计算,并加和即可
import java.util.*;
public class AntiOrder {
public int count(int[] A, int n) {
// write code here
int result[]=new int[n];
if(n==0){result[n-1]=0;return 0;}
buildTree(A, n,result);
int count=0;
for (int i = 0; i < result.length; i++) {
count += i-result[i];
}
return count;
}
public void buildTree(int[]A,int n,int[] result){
Node head = new Node(A[0]);
int count = 1;
while(count<n){//1-n-1个子节点
Node newnode = new Node(A[count]);
insertTree(head,newnode);
result[count]=getLeftNum(head,newnode.val);
count++;
}
}
public void insertTree(Node head,Node newnode){
if(newnode.val<=head.val){//判定左侧
if(head.left!=null){
insertTree(head.left, newnode);
}else{
head.left=newnode;
}
head.leftsize++;
}//if
else {//右侧
if(head.right!=null){
insertTree(head.right, newnode);
}else {
head.right = newnode;
}
}//else
}
public int getLeftNum(Node head,int val){
if(val==head.val)return head.leftsize;
else if(val<head.val){
return getLeftNum(head.left, val);
}else{
return head.leftsize+1+getLeftNum(head.right, val);
}
}
}
class Node{
public Node(int val){
this.val = val;
}
int leftsize =0;
int val;
Node left,right;
}