算法笔记：分治

分治的基本概念

把一个任务，分成形式和原任务相同，但规模更小的 几个部分任务（通常是两个部分），分别完成，或只 需要选一部完成。然后再处理完成后的这一个或几个 部分的结果，实现整个任务的完成。

分治的典型应用：归并排序

数组排序任务可以如下完成：

1）把前一半排序

2) 把后一半排序

3) 把两半归并到一个新的有序数组，然后再拷贝回原 数组，排序完成。

#include <iostream>
using namespace std;
/*
归并排序
*/
void Merge(int a[], int s, int m, int e, int tmp[])
{
	int p = 0;
	int p1 = s, p2 = m+1;
	while (p1 <= m&&p2 <= e)
	{
		if (a[p1] < a[p2])
			tmp[p++] = a[p1++];
		else
			tmp[p++] = a[p2++];
	}
	while (p1 <= m)
		tmp[p++] = a[p1++];
	while (p2 <= e)
		tmp[p++] = a[p2++];
	for (int i = 0; i < e - s + 1; ++i)
		a[s + i] = tmp[i];
}
void MergeSort(int a[], int s, int e, int tmp[])
{
	if (s < e)
	{
		int m = s + (e - s) / 2;
		MergeSort(a, s, m, tmp);
		MergeSort(a, m + 1, e, tmp);
		Merge(a, s, m, e, tmp);
	}
}
int a[10] = { 13,27,19,2,8,12,2,8,30,89 };
int b[10];
int main()
{
	int size = sizeof(a) / sizeof(int);
	MergeSort(a, 0, size - 1, b);
	for (int i = 0; i < size; i++)
		printf("%d ", a[i]);
	return 0;
}

 运行结果

 快速排序

数组排序任务可以如下完成：

1）设k=a[0], 将k挪到适当位置，使得比k小的元素都 在k左边,比k大的元素都在k右边，和k相等的，不关心 在k左右出现均可 （O(n)时间完成）

2) 把k左边的部分快速排序

3) 把k右边的部分快速排序

void QuickSort(int a[], int s, int e)
{
	if (s >= e)
		return;
	int k = a[s];
	int i = s, j = e,t;
	while (i != j)
	{
		while (a[j] >= k&&j>i)
			j--;
		while (a[i] <= k&&i<j)
			i++;
		t = a[j];
		a[j] = a[i];
		a[i] = t;
	}
	t = a[s];
	a[s] = a[i];
	a[i] = t;
	QuickSort(a, s, i-1);
	QuickSort(a, i + 1, e);
}
int a[10] = { 13,27,19,2,8,12,2,8,30,89 };
int main()
{
	int size = sizeof(a) / sizeof(int);
	QuickSort(a, 0, size - 1);
	for (int i = 0; i < size; i++)
		printf("%d ", a[i]);
	return 0;
}

运行结果

例题：输出前m大的数

描述 给定一个数组包含n个元素，统计前m大的数并且把这m个数从小到大输出。

输入 第一行包含一个整数n，表示数组的大小。n < 100000。 第二行包含n个整数，表示数组的元素，整数之间以一个空格分开 。每个整数的绝对值不超过100000000。 第三行包含一个整数m。m < n。

输出 从小到大输出前m大的数，每个数一行。

思路：把前m大的都弄到数组最右边，然后对这最右边m个元素排序， 再输出

void max_m(int a[], int s, int e, int m)
{
	int k = a[s];
	int i = s, j = e, t;
	while (i != j)
	{
		while (a[j] >= k && j > i)
			j--;
		while (a[i] <= k && i < j)
			i++;
		t = a[j];
		a[j] = a[i];
		a[i] = t;
	}
	t = a[s];
	a[s] = a[i];
	a[i] = t;
	if (e - i == m)
		QuickSort(a, i + 1, e);
	if (e - i + 1 == m)
		QuickSort(a, i, e);
	if (e - i + 1 < m)
		max_m(a, s, i - 1, m - e + i - 1);
	if (e - i > m)
		max_m(a, i + 1, e, m);
}
int main()
{
	int a[100000], n, m;
	cin >> n;
	for (int i=0; i<n; i++)
		scanf_s("%d", &a[i]);
	cin >> m;
	max_m(a, 0, n - 1, m);
	for (int i = n-m; i < n; i++)
		printf("%d ", a[i]);
	return 0;
}

运行结果

例题2：求排列的逆序数

考虑1,2,…,n (n ik， 那么就称(ij ,ik )是这个排列的一个逆序。 一个排列含有逆序的个数称为这个排列的逆序数。例如排列 263451 含有8个 逆序(2,1),(6,3),(6,4),(6,5),(6,1),(3,1),(4,1),(5,1)，因此该排列的逆序数就是8。 现给定1,2,…,n的一个排列，求它的逆序数。

思路：

1） 将数组分成两半，分别求出左半边的逆序数和右半边的逆序数

2） 再算有多少逆序是由左半边取一个数和右半边取一个数构成(要求O(n)实现）

2) 的关键：左半边和右半边都是排好序的。比如，都是从大到小排序的。这样，左右半边只需要从头到尾各扫一遍，就可以找出由两边各取一个数构成的逆序个数

int _ni_count(int a[], int s, int e, int m)
{
	int i = m, j = e, re = 0;
	while (i >= s)
	{
		for (; a[i] <= a[j] && j > m; j--);
		re += (j - m);
		i--;
	}
	return re;
}
int ni_count(int a[], int s, int e)
{
	int re = 0;
	if (s >= e)
		return re;
	int m = s + (e - s) / 2;
	re = ni_count(a, s, m) + ni_count(a, m + 1, e);
	QuickSort(a, s, m);
	QuickSort(a, m + 1, e);
	re += _ni_count(a, s, e, m);
	return re;
}
int main()
{
	int a[100000], n;
	cin >> n;
	for (int i = 0; i < n; i++)
		cin >> a[i];
	cout << ni_count(a, 0, n - 1);
	return 0;
}

运行结果