欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

通用算法之归并求逆序对

程序员文章站 2022-03-03 07:50:42
...

将数组分成小数组,对相邻的两个小数组统计逆序对的个数,统计完顺便排好序,避免重复计算。


import java.util.*;
public class Main {
 static int count=0;//全局变量,统计逆序对个数
 static void splitarray(int[]arr,int start,int end)
 {
  if(start>=end)
   return;
  int mid=(start+end)>>1;
  //将数组分成小数组
  splitarray(arr, start, mid);
  splitarray(arr, mid+1, end);
  merge(arr,start,mid,end);//对这相邻的两个小数组进行逆序对统计,排序
 }
 static void merge(int[]arr,int start,int mid,int end)
 {
  int []temp=new int[end-start+1];//用来暂时存放排好序的数,按顺序将小的数放入
  int i=start,j=mid+1,k=0;
  
  while(i<=mid&&j<=end)
  {
   if(arr[i]>arr[j])//存在逆序对
   {
    count+=mid-i+1;//i-mid区间内都与j形成逆序对
    temp[k++]=arr[j++];
   }
   else {
    temp[k++]=arr[i++];//无逆序对
   }
  }
  while(i<=mid)
  {
   temp[k++]=arr[i++];
  }
  while(j<=end)
   temp[k++]=arr[j++];
  for(int l=0;l<temp.length;l++)
   arr[start+l]=temp[l];//对原数组排序
 }
 public static void main(String[]args){
  Scanner in=new Scanner(System.in);
  int n=in.nextInt();
  int[]num=new int[n];
  for(int i=0;i<n;i++)
   num[i]=in.nextInt();
  if(n!=0)
   splitarray(num,0,num.length-1);
  System.out.print(count);
     }
}