目录
4.6 two pointers
问题 A: 二路归并排序(mergesort)递归法 [2*+]
题目描述
二路归并排序(mergesort)递归法
用递归法进行二路归并排序
输入:第一行一个数据n,表示有n个数要排序。接下来n行每行一个<=10^7的整数。
输出:n行,由小到大排序后的数据
数据规模:n<=10^5
思考:两个递归都会被执行吗?
思路
这题的OJ好像出错了,提交908,解决0,大多数人的错误都是运行错误80(我的代码也是),大家看看代码理解一下二路归并排序的思想就好。
另外,这里合并序列的函数不要命名为merge,因为merge是C++STL里自带的函数(也是用于序列合并)。
代码
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
using namespace std;
const int maxn = 100001;
void MyMerge(int a[], int L1, int R1, int L2, int R2){
int i = L1, j = L2;
int temp[maxn], index = 0;
while(i<=R1&&j<=R2){
if(a[i]<=a[j]) temp[index++] = a[i++];
else temp[index++] = a[j++];
}
while(i<=R1) temp[index++] = a[i++];
while(j<=R2) temp[index++] = a[j++];
for(int i=0;i<index;i++) a[L1+i] = temp[i];
}
void MergeSort(int a[], int left, int right){
if(left<right){
int mid = (left+right)/2;
MergeSort(a, left, mid);
MergeSort(a, mid+1, right);
MyMerge(a, left, mid, mid+1, right);
}
}
int a[maxn]={0};
int main(){
int n;
while(scanf("%d", &n) != EOF){
for(int i=0;i<n;i++) scanf("%d", &a[i]);
MergeSort(a, 0, n-1);
for(int i=0;i<n;i++) printf("%d\n", a[i]);
memset(a, 0, sizeof(a));
}
return 0;
}
问题 B: 基础排序III:归并排序
题目描述
归并排序是一个时间复杂度为O(nlogn)的算法,对于大量数据远远优于冒泡排序与插入排序。
这是一道排序练习题,数据量较大,请使用归并排序完成。
输入
第一行一个数字n,代表输入的组数
其后每组第一行输入一个数字m,代表待排序数字的个数
其后m行每行一个数据,大小在1~100000之间,互不相等,最多有10万个数据。
输出
升序输出排好序的数据,每行一个数字
样例输入
1
10
10
9
8
7
6
5
4
3
2
1
样例输出
1
2
3
4
5
6
7
8
9
10
思路
跟上一题一样,还是用二路归并排序来做,但是我建议把上一题的代码删了自己再打一遍,看着书打也好,跟着自己的记忆打也好,总之都能加深对归并排序的记忆和理解就行。
代码
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
using namespace std;
const int maxn = 100001;
void MyMerge(int a[], int L1, int R1, int L2, int R2){
int i = L1, j = L2;
int temp[maxn], index = 0;
while(i<=R1&&j<=R2){
if(a[i]<=a[j]) temp[index++] = a[i++];
else temp[index++] = a[j++];
}
while(i<=R1) temp[index++] = a[i++];
while(j<=R2) temp[index++] = a[j++];
for(int i=0;i<index;i++) a[L1+i] = temp[i];
}
void MergeSort(int a[], int left, int right){
if(left<right){
int mid = (left+right)/2;
MergeSort(a, left, mid);
MergeSort(a, mid+1, right);
MyMerge(a, left, mid, mid+1, right);
}
}
int a[maxn]={0};
int main(){
int n;
while(scanf("%d", &n) != EOF){
int m;
scanf("%d", &m);
for(int i=0;i<m;i++) scanf("%d", &a[i]);
MergeSort(a, 0, m-1);
for(int i=0;i<m;i++) printf("%d\n", a[i]);
memset(a, 0, sizeof(a));
}
return 0;
}
问题 C: 快速排序 qsort [2*]
题目描述
输入n个整数,用快速排序的方法进行排序
Input
第一行数字n 代表接下来有n个整数
接下来n行,每行一个整数
Output
升序输出排序结果
每行一个数据
Sample Input
5
12
18
14
13
16
Sample Output
12
13
14
16
18
Hint
n<=5000
每个数据<=5000
思路
这题是用快排,还是一样的,能自己打就自己打,不要做一个莫得感情的代码搬移机器。一样的,这里函数命名不要用partiton,因为也是STL里的东西。
代码
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
using namespace std;
const int maxn = 5001;
int MyPartition(int a[], int left, int right){
int temp = a[left];
while(left<right){
while(left<right&&a[right]>temp) right--;
a[left] = a[right];
while(left<right&&a[left]<=temp) left++;
a[right] = a[left];
}
a[left] = temp;
return left;
}
void QuickSort(int a[], int left, int right){
if(left<right){
int pos = MyPartition(a, left, right);
QuickSort(a, left, pos-1);
QuickSort(a, pos+1, right);
}
}
int a[maxn] = {0};
int main(){
int n;
while(scanf("%d", &n) != EOF){
for(int i=0;i<n;i++) scanf("%d", &a[i]);
QuickSort(a, 0, n-1);
for(int i=0;i<n;i++) printf("%d\n", a[i]);
memset(a, 0, sizeof(a));
}
return 0;
}
问题 D: 二分递归快排(Qsort) [2*]
题目描述
二分递归快排(Qsort)
用二分递归的方法实现快排
输入:第一行一个数据n,表示有n个数要排序。接下来n行每行一个<=10^7的整数。
输出:n行,由小到大排序后的数据
数据规模:n<=10^5
思考:两个递归都会被执行吗?有几种可能?
思路
这题和上题一样(好像上题是要求快排用非递归法实现??),尽量把快排的两个函数默写出来比较好。
代码
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
using namespace std;
const int maxn = 100000;
int MyPartition(int a[], int left, int right){
int temp = a[left];
while(left<right){
while(left<right&&a[right]>temp) right--;
a[left] = a[right];
while(left<right&&a[left]<=temp) left++;
a[right] = a[left];
}
a[left] = temp;
return left;
}
void QuickSort(int a[], int left, int right){
if(left<right){
int pos = MyPartition(a, left, right);
QuickSort(a, left, pos-1);
QuickSort(a, pos+1, right);
}
}
int a[maxn] = {0};
int main(){
int n;
while(scanf("%d", &n) != EOF){
for(int i=0;i<n;i++) scanf("%d", &a[i]);
QuickSort(a, 0, n-1);
for(int i=0;i<n;i++) printf("%d\n", a[i]);
memset(a, 0, sizeof(a));
}
return 0;
}
小结
虽然可能遇到排序题大多时候都用STL里的sort函数,但是掌握归并排序和快速排序的原理还是很重要的(即便很多时候不要求自己实现),可以和第四章最前面讲到的选择排序和插入排序以及最前面的冒泡排序作个比较,利用two pointers的优越性就在于可以免去许多无效的遍历。
4.7 其他高效技巧与算法
问题 A: 求第k大数
题目描述
给定一个长度为n(1≤n≤1,000,000)的无序正整数序列,以及另一个数k(1≤k≤1,000,000)(关于第k大的数:例如序列{1,2,3,4,5,6}中第3大的数是4。)
输入
第一行两个正整数m,n。
第二行为n个正整数。
输出
第k大的数。
样例输入
6 3
1 2 3 4 5 6
样例输出
4
思路
这题的第k大的数和书上的第k大的数的定义正好反过来(书上是从小往大数,这题是从大往小数),不过算法还是一样的,用随机选择算法。
代码
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<time.h>
#include<algorithm>
using namespace std;
const int maxn = 1000000;
int randpartition(int a[], int left, int right){
int p = (round(1.0*rand()/RAND_MAX*(right-left)+left));
swap(a[p],a[left]);
int temp = a[left];
while(left<right){
while(left<right&&a[right]>temp) right--;
a[left] = a[right];
while(left<right&&a[left]<=temp) left++;
a[right] = a[left];
}
a[left] = temp;
return left;
}
int randselect(int a[], int left, int right, int k){
if(left==right) return a[left];
int p = randpartition(a, left, right);
int m = right-p+1;//从后往前数
if(k==m) return a[p];
if(k<m) return randselect(a, p+1, right, k);//这里要倒过来
else return randselect(a, left, p-1, k-m);
}
int a[maxn] = {0};
int main(){
srand((unsigned)time(NULL));
int n, k;
while(scanf("%d%d", &n, &k) != EOF){
for(int i=0;i<n;i++) scanf("%d", &a[i]);
printf("%d\n", randselect(a, 0, n-1, k));
memset(a, 0, sizeof(a));
}
return 0;
}
小结
随机选择算法可以说是partition函数除了快排之后的又一个扩展算法,优点在于不用排序就能直接选择出用户所规定的第几大元素,时间复杂度是O(n)(虽然快排的时间复杂度O(nlogn)也不算太大)。