已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A0,A1,⋯,AN−1的中位数指A(N−1)/2的值,即第⌊(N+1)/2⌋个数(A0为第1个数)。
输入格式:
输入分三行。第一行给出序列的公共长度N(0<N≤100000),随后每行输入一个序列的信息,即N个非降序排列的整数。数字用空格间隔。
输出格式:
在一行中输出两个输入序列的并集序列的中位数。
输入样例1:
5
1 3 5 7 9
2 3 4 5 6
输出样例1:
4
输入样例2:
6
-100 -10 1 1 1 1
-50 0 2 3 4 5
输出样例2:
1
代码1:
#include <stdio.h>
#include <stdlib.h>
#define N 100000
int main()
{
int a[N], b[N],c[2*N];
int length, flag1 = 0, flag2 = 0;
scanf("%d", &length);
for (int i = 0; i < length; i++) {
scanf("%d", &a[i]);
}
for (int i = 0; i < length; i++) {
scanf("%d", &b[i]);
}
int j = 0; //j用来记录c数组的下标
while (flag1 != length && flag2 != length) {
if (a[flag1] < b[flag2]) {
c[j] = a[flag1];
j++;
flag1++;
}
else if (a[flag1] > b[flag2]) {
c[j] = b[flag2];
j++;
flag2++;
}
else if (a[flag1] == b[flag2]) {
c[j] = a[flag1];
j++;
flag1++;
c[j] = b[flag2];
j++;
flag2++;
}
}
if (flag1 != length) {
while (flag1 != length) {
c[j] = a[flag1];
j++;
flag1++;
}
}
if (flag2 != length) {
while (flag2 != length) {
c[j] = b[flag2];
j++;
flag2++;
}
}
/*
测试用
for (int i = 0; i < 2 * length; i++) {
printf("%d ", c[i]);
}
*/
printf("%d", c[(2 * length + 1) / 2 - 1]);
return 0;
}
很简单的一道题,和2-3 链表拼接一样的思路。
最后在输出中位数时,比如我们输出的是第五个数,第五个数的下标是4,所以要减一
9.23 二刷
新的思路:
#include<iostream>
#include<string>
#include<algorithm>
#include<vector>
using namespace std;
int main() {
int N,number;
vector<int> v1, v2,v3;
cin >> N;
for (int i = 0; i < N; i++) {
cin >> number;
v1.push_back(number);
}
for (int i = 0; i < N; i++) {
cin >> number;
v2.push_back(number);
}
int temp1 = 0;
int temp2 = 0;
int num = 0; //记录数量,达到目标数量就输出
while (!v1.empty() || !v2.empty()) {
if (v1[temp1] <= v2[temp2]) {
v3.push_back(v1[temp1]);
temp1++;
num++;
}
else {
v3.push_back(v2[temp2]);
temp2++;
num++;
}
if (num == N) {
printf("%d", v3[v3.size() - 1]);
return 0;
}
}
system("pause");
}
由题干:
“已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A0,A1,⋯,AN−1的中位数指A(N−1)/2的值,即第⌊(N+1)/2⌋个数(A0为第1个数)。”可知,我们最后要输出的是第⌊(N+1)/2⌋个数,(注意这里是向下取整符号)。
假设N是偶数,比如是6,那么(6+1)/2等于3.5,向下取整得3。
因为这里的N是我们输入的非降序序列S1, S2的长度的二倍,所以不可能是奇数。
所以我们不用把他们的并集全求出来,用一个变量记录并集的元素的个数,达到相应数量直接输出即可