通用算法之归并求逆序对
程序员文章站
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);
}
}
上一篇: 知识补充