算法1.2 归并排序

归并排序

思想

1. 分治法： 将 $a[l,r]$ 分解成 $a[l,mid],a[mid+1,r]$ 两部分
2. 调用 $MergeSort(l,mid)$ 和 $MergeSort(mid+1,r)$，将这两个函数各自排好序
3. 二路归并：将两个有序数组合并成一个大的有序数组，覆盖 $a[l,r]$
   1. 申请空间，使其大小为两个已经排序序列之和，该空间用来存放合并后的序列
   2. 设定两个指针，最初位置分别为两个已经排序序列的起始位置
   3. 比较两个指针所指向的元素，选择相对小的元素放入到合并空间，并移动指针到下一位置
   4. 重复步骤3直到某一指针超出序列尾
   5. 将另一序列剩下的所有元素直接复制到合并序列尾

图解

代码

void MergeSort(int l, int r)
{
	if(l == r)
		return ;
	MergeSort(l, (l + r) / 2);
	MergeSort((l + r) / 2 + 1, r);
	int i = l, j = (l + r) / 2 + 1, cnt = l;
	int b[n];
	while(i <= (l + r) / 2 && j <= r)
	{
		if(a[i] < a[j])
			b[cnt++] = a[i++];
		else
			b[cnt++] = a[j++];
	}
	while(i <= (l + r) / 2)
		b[cnt++] = a[i++];
	while(j <= r)
		b[cnt++] = a[j++];
	for(; l <= r; l++)
		a[l] = b[l];
}

分析

• 归并排序是稳定的排序;
• 时间复杂度为$O(nlogn)$，空间复杂度为$O(n)$。

逆序对

定义

对于给定的一段正整数序列，逆序对就是序列中 $a_i>a_j$ 且 $i<j$的有序对。

代码

//求逆序对的个数
void mergesort(int l, int r)
{
	if(l == r)	
		return;
	mergesort(l, (l + r) / 2);
	mergesort((l + r) / 2 + 1, r);
	int i = l, j = (l + r) / 2 + 1, cnt = l;
	int b[n];
	while(i <= (l + r) / 2 && j <= r)
	{
		if(a[j] < a[i])
		{
			s += (l + r) / 2 - i + 1;
			b[cnt++] = a[j++];
		}
		else	
			b[cnt++] = a[i++];
	}
	while(i <= (l + r) / 2)
		b[cnt++] = a[i++];
	while(j <= r)
		b[cnt++] = a[j++];
	for(int i = l; i <= r; i++)
		a[i] = b[i];
}

树状数组解逆序对

注意数各不相同！

int main()
{
	scanf("%d", &n);
	for(int i = 1; i <= n; i++)
	{
		scanf("%d",&a);
		node[i].id = i;
		node[i].data = a;
	}
	sort(node + 1, node + 1 + n);
	for(int i = 1;i <= n; i++)
	{
		add(node[i].id, 1);  //次数+1
		ans += i - sum(node[i].id); //得到之前有多少个比你大的数
	}
	printf("%d", ans);
	return 0;
}