核心思路:
- 定分界点:mid=l+r/2
- 左右递归
- 归并(双指针归并)
代码模板:
int tmp[];
void merge_sort(int q[], int l, int r)
{
if (l >= r) return;
int mid =(l+r)/2;
merge_sort(q, l, mid);
merge_sort(q, mid + 1, r);
int k = 0, i = l, j = mid + 1;
while (i <= mid && j <= r)
if (q[i] <= q[j]) tmp[k ++ ] = q[i ++ ];
else tmp[k ++ ] = q[j ++ ];
while (i <= mid) tmp[k ++ ] = q[i ++ ];
while (j <= r) tmp[k ++ ] = q[j ++ ];
for (i = l, j = 0; i <= r; i ++, j ++ ) q[i] = tmp[j];
}
求逆序对模板:
const int N = 1e6 + 10;
int tmp[N];
long long merge_sort(int q[], int l, int r)
{
long long res = 0;
if (l >= r) return res;
int mid = (l + r) / 2;
res=merge_sort(q,l,mid)+merge_sort(q,mid+1,r);
int k = 0, i = l, j = mid + 1;
while (i <= mid && j <= r)
if (q[i] <= q[j]) tmp[k++] = q[i++];
else
{
tmp[k++] = q[j++];
res+=(mid-i+1);
}
while (i <= mid) tmp[k++] = q[i++];
while (j <= r) tmp[k++] = q[j++];
for (i = l, j = 0; i <= r; i++, j++) q[i] = tmp[j];
return res;
}
题目:
给定你一个长度为 nn 的整数数列。
请你使用归并排序对这个数列按照从小到大进行排序。
并将排好序的数列按顺序输出。
输入格式
输入共两行,第一行包含整数 nn。
第二行包含 nn 个整数(所有整数均在 1∼1091∼109 范围内),表示整个数列。
输出格式
输出共一行,包含 nn 个整数,表示排好序的数列。
数据范围
1≤n≤1000001≤n≤100000
输入样例:
5
3 1 2 4 5
输出样例:
1 2 3 4 5
答案
#include<iostream>
using namespace std;
const int N=1e6+10;
int tmp[N];
void merge_sort(int q[], int l, int r)
{
if (l >= r) return;
int mid = (l + r) / 2;
merge_sort(q, l, mid);
merge_sort(q, mid + 1, r);
int k = 0, i = l, j = mid + 1;
while (i <= mid && j <= r)
if (q[i] <= q[j]) tmp[k++] = q[i++];
else tmp[k++] = q[j++];
while (i <= mid) tmp[k++] = q[i++];
while (j <= r) tmp[k++] = q[j++];
for (i = l, j = 0; i <= r; i++, j++) q[i] = tmp[j];
}
int a[100010];
int main()
{
int n;
cin >> n;
for (int i = 0;i < n;i++)
{
cin >> a[i];
}
merge_sort(a, 0, n - 1);
for (int i = 0;i < n;i++)
{
cout << a[i] << " ";
}
}
给定一个长度为 nn 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 ii 个和第 jj 个元素,如果满足 i<ji<j 且 a[i]>a[j]a[i]>a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 nn,表示数列的长度。
第二行包含 nn 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤1000001≤n≤100000,
数列中的元素的取值范围 [1,109][1,109]。
输入样例:
6
2 3 4 5 6 1
输出样例:
5
代码:
暴力:
但是超时:
#include<iostream>
using namespace std;
typedef long long ll;
const int N = 100000;
int main()
{
int n;
ll count=0;
int a[N];
cin >> n;
for (int i = 0;i < n;i++)
{
cin >> a[i];
}
int i,j;
for ( i = 0;i < n;i++)
{
for (j = i+1;j < n;j++)
{
if (a[i] > a[j])
{
count++;
}
}
}
cout << count;
}
用归并排序:
#include<iostream>
using namespace std;
const int N = 1e6 + 10;
int tmp[N];
long long merge_sort(int q[], int l, int r)
{
long long res = 0;
if (l >= r) return res;
int mid = (l + r) / 2;
res=merge_sort(q,l,mid)+merge_sort(q,mid+1,r);
int k = 0, i = l, j = mid + 1;
while (i <= mid && j <= r)
if (q[i] <= q[j]) tmp[k++] = q[i++];
else
{
tmp[k++] = q[j++];
res+=(mid-i+1);
}
while (i <= mid) tmp[k++] = q[i++];
while (j <= r) tmp[k++] = q[j++];
for (i = l, j = 0; i <= r; i++, j++) q[i] = tmp[j];
return res;
}
int a[N];
int main()
{
int n;
cin >> n;
for (int i = 0;i < n;i++)
{
cin >> a[i];
}
long long res = merge_sort(a,0,n-1);
cout << res;
}