通用算法之归并求逆序对

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


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);
     }
}
发布了45 篇原创文章 · 获赞 4 · 访问量 1025

猜你喜欢

转载自blog.csdn.net/weixin_43838915/article/details/105736864