Solution:
这道题的意思是,给出两个递增的数组,它们的个数为n1、n2(n1、n2<200000),求出这两个数组的中位数,即将它们合并后(仍然递增)后的数组的中位数。
中位数 mid =(n1+n2+1)/2
一种方法是将这两个数组读入到一个数组中,再进行排序,最后直接取中位数即可,但这种方法不能通过全部测试点。
方法一:
#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
int n1,n2;
int num[400005];
int main(){
scanf("%d",&n1);
for(int i=1;i<=n1;i++){
scanf("%d",&num[i]);
}
scanf("%d",&n2);
for(int i=n1+1;i<=n1+n2;i++){
scanf("%d",&num[i]);
}
sort(num+1,num+n1+n2+1);
int temp=(n1+n2+1)/2;
printf("%d",num[temp]);
return 0;
}
另一种方法是:先读入数组1,然后在线处理数组2。cnt记录当前的个数,其初始值为0,当cnt=mid时,就得到了中位数。给数组1设置一个指针i,读入数组2时,每次读入一个数temp,在数组1中找出比temp小的数的个数,并且cnt++,若此时cnt==mid,就停止,当num[i]>=temp时,cnt++,并判断此时cnt是否等于mid,若相等,输出temp即可,若不相等,再从数组2读入一个temp,依次循环,若数组2读完后cnt仍小于mid,说明mid在数组1中,在数组1中找到mid即可。
#include<iostream>
#include<stdio.h>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
int n1,n2;
int num[200005];
int main(){
scanf("%d",&n1);
for(int i=1;i<=n1;i++){
scanf("%d",&num[i]);
}
scanf("%d",&n2);
num[n1+1]=INF;
int cnt=0;//计数
int temp;//读入第二个数组中的一个数
int mid=(n1+n2+1)/2;//中位数的位置
int i=1;
for(int j=1;j<=n2;j++){
scanf("%d",&temp);
while(num[i]<temp){
cnt++;
if(cnt==mid){
printf("%ld",num[i]);
}
i++;
}
cnt++;
if(cnt==mid){
printf("%ld",temp);
}
}
while(i<=n1){
cnt++;
if(cnt==mid){
printf("%ld",num[i]);
}
i++;
}
return 0;
}