算法思想
- 找分割点:找将数组分为左右两个区间的标记元素,一般取数组中间的元素。
- 递归:对左右两各区间的数组元素分别进行归并排序,使之有序。
- 将排好序的左右两个数组进行归并 使用temp数组来存储排好序的元素
- 将temp数组中的元素写回到原数组中
AcWing787
给定你一个长度为n的整数数列。
请你使用归并排序对这个数列按照从小到大进行排序。
并将排好序的数列按顺序输出。
输入格式
输入共两行,第一行包含整数 n。
第二行包含 n 个整数(所有整数均在1~10^9范围内),表示整个数列。
输出格式
输出共一行,包含 n 个整数,表示排好序的数列。
数据范围
1≤n≤100000
输入样例:
5
3 1 2 4 5
输出样例:
1 2 3 4 5
难度:简单
时/空限制:1s / 64MB
算法模板(787题解)
#include<iostream>
using namespace std;
const int N = 100100;
int v[N];
int temp[N];
int n;
void merge_sort(int v[], int left, int right)
{
if(left >= right)
return;
int mid = left + right >> 1;
merge_sort(v, left, mid); merge_sort(v, mid+1, right);
int k = 0; int i = left; int j = mid + 1;
while(i <= mid && j <= right){
if(v[i] < v[j])
temp[k++] = v[i++];
else
temp[k++] = v[j++];
}
while(i <= mid)
temp[k++] = v[i++];
while(j <= right)
temp[k++] = v[j++];
for(i = left, j = 0; i <= right; i++, j++)
v[i] = temp[j];
}
int main()
{
scanf("%d", &n);
for(int i = 0; i < n; i++)
scanf("%d", &v[i]);
merge_sort(v, 0, n-1);
for(int i = 0; i < n; i++)
printf("%d ",v[i]);
return 0;
}
AcWing788
给定一个长度为n的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i < j 且 a[i] > a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤100000
输入样例:
6
2 3 4 5 6 1
输出样例:
5
题目分析
- 利用归并排序来计算逆序对的数量,逆序对有三种情况,见图1.1.
- 第1,2种情况可由函数返回值来返回,重点是第三种情况,第三种情况返回到上一级函数中便是第一或第二种情况,归并排序函数是一种递归的函数。详情见下方逆序对题解中的merge_sort()函数。
- 如图1.1:第三种情况中,当v[i] > v[j]时,下标从i到j这段数组中的所有元素都与v[j]构成了逆序对,数量为mid - i + 1;
- 由此,我们可以来写我们的代码:
逆序对题解(788题解)
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int v[N],temp[N];
long long merge_sort(int v[], int left, int right)
{
if(left >= right)
return 0;
int mid = left + right >> 1;
long long res = merge_sort(v, left, mid) + merge_sort(v, mid+1, right);
//归并的过程
int i = left, j = mid+1, k = 0;
while(i <= mid && j <= right){
if(v[i] <= v[j])
temp[k++] = v[i++];
else{
temp[k++] = v[j++];
res += mid - i +
}
}
//扫尾
while(i <= mid)
temp[k++] = v[i++];
while(j <= right)
temp[k++] = v[j++];
//物归原主
for(int i = left, j = 0; i <= right; i++, j++)
v[i] = temp[j];
return res;
}
int main()
{
//读入
int n;
scanf("%d", &n);
for(int i = 0; i < n; i++)
scanf("%d", &v[i]);
long long L = merge_sort(v, 0, n-1);
printf("%lld",L);
return 0;
}