强行和lc联动
#include<vector>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e6 + 10;
int LEN;
vector<int> A, B;
int tmp[N];
void merge(vector<int>& A, int m, vector<int>& B, int n) {
int i = m - 1, j = n - 1, k = m + n - 1;
while (j >= 0)
{
if (i < 0 || B[j] >= A[i]) A[k--] = B[j--]; //反过来就不用再翻转了。直接把大得放后面。
else A[k --] = A[i--];
}
}
int main()
{
cin >> LEN;
A.resize(LEN/2);
for (int i = 0; i < LEN/2; i ++ )
cin >> A[i];
int idx = 0;
int m;
B.resize(LEN - LEN/2);
while (cin >> m)
{
B[idx ++] = m;
}
sort(A.begin(), A.end()); //这两行写出来感觉其他白写了。。
sort(B.begin(), B.end());
A.resize(LEN);
merge(A, LEN/2, B, idx);
for (int i = 0; i < LEN; i ++ ) cout << A[i] << ' ';
return 0;
}
使用了模板补全。
归并的思路不难。
这题就是二路归并,先把数组分成两块。
然后合并到tmp里(再幅回给原数组)
#include<iostream>
using namespace std;
const int N = 1e6 + 10;
int n;
int a[N];
int tmp[N];
void merge_sort(int q[], int l, int r) // 归并排序
{
if (l >= r) return;
int mid = (l + r) >> 1; //二分
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 ++ ]; //把当前比较出来小的放入。即q[i]
else tmp[k ++ ] = q[j ++ ]; //一样,这里q[j]小
//后面就是把另一个数组B已经空的时候,这个数组A的数放进去。
while (i <= mid) tmp[k ++ ] = q[i ++ ]; //(mid,r)空了,左边(0,mid]剩了一个
while (j <= r) tmp[k ++ ] = q[j ++ ]; //同理(0,mid)空了,右边(mid,r)剩了一个。(因为二分所以是一个)
for (i = l, j = 0; i <= r; i ++, j ++ ) q[i] = tmp[j];
}
int main()
{
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] << ' ';
return 0;
}