萌新学习算法——归并排序进阶(二)
我们在前面学习了归并排序在有序的情况下比快排优,接下来我们讲的的是一道有归并思想引出的变种做法。
我们还是用洛谷的题目来进行讲解:P1908 逆序对
题目描述
猫猫TOM和小老鼠JERRY最近又较量上了,但是毕竟都是成年人,他们已经不喜欢再玩那种你追我赶的游戏,现在他们喜欢玩统计。最近,TOM老猫查阅到一个人类称之为“逆序对”的东西,这东西是这样定义的:对于给定的一段正整数序列,逆序对就是序列中ai>aj且i<j的有序对。知道这概念后,他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。
Update:数据已加强。
输入格式
第一行,一个数n,表示序列中有n个数。
第二行n个数,表示给定的序列。序列中每个数字不超过10^9
输出格式
给定序列中逆序对的数目。
我们还是通过比较来看出有归并排序的好在哪里,针对这道题我们可能会用暴力法解决,接下来我们用暴力法来看这道题。
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader reader=new BufferedReader(new InputStreamReader(System.in));
String str[]=reader.readLine().split(" ");
int n=Integer.parseInt(str[0]);
int data[]=new int[n];
str=reader.readLine().split(" ");
for(int i=0;i<n;i++)
data[i]=Integer.parseInt(str[i]);
int count=0;
for(int i=0;i<n-1;i++)
for(int j=i+1;j<n;j++)
if(data[i]>data[j]) //满足逆序对条件
count++;
System.out.println(count);
}
}
上面的暴力破解肯定会爆,所以接下来我们利用归并排序来实现:
Merge方法:
public static void Meger(int start ,int end)
{
if(start>=end) return ; //退出条件
int mid=(start+end)/2;
Meger(start, mid);
Meger(mid+1, end);
int i=start,j=mid+1,k=start;
while(i<=mid&&j<=end)
{
if(data[i]>data[j])
{
temp[k++]=data[i++];
count+=end-j+1; //比之前的归并排序加了计算前面的数大于后面的数
}
else
temp[k++]=data[j++];
}
while(i<=mid)
temp[k++]=data[i++];
while(j<=end)
temp[k++]=data[j++];
for(i=start;i<=end;i++)
data[i]=temp[i];
}
完整代码:
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main {
public static int n,data[],temp[];
public static long count=0;
public static void Meger(int start ,int end)
{
if(start>=end) return ; //退出条件
int mid=(start+end)/2;
Meger(start, mid);
Meger(mid+1, end);
int i=start,j=mid+1,k=start;
while(i<=mid&&j<=end)
{
if(data[i]>data[j])
{
temp[k++]=data[i++];
count+=end-j+1; //比之前的归并排序加了计算前面的数大于后面的数
}
else
temp[k++]=data[j++];
}
while(i<=mid)
temp[k++]=data[i++];
while(j<=end)
temp[k++]=data[j++];
for(i=start;i<=end;i++)
data[i]=temp[i];
}
public static void main(String[] args) throws IOException {
BufferedReader reader=new BufferedReader(new InputStreamReader(System.in));
String str[]=reader.readLine().split(" ");
n=Integer.parseInt(str[0]);
data=new int[n];
str=reader.readLine().split(" ");
for(int i=0;i<n;i++)
data[i]=Integer.parseInt(str[i]);
temp=new int[n];
Meger(0,n-1);
System.out.println(count);
}
}