题意分析:
(1)给出两个有序序列,求这两个有序序列的中位数。
(2)由于合并两个有序序列,再排序取中位数,复杂度为O(nlogn),会有几个案例超时,考虑到中位数的特殊性质,中位数一定是一个有序序列的中间位置的那个数,如果我们呢合并两个序列,那样之前给的两个有序序列这个条件我们就没用上。我们可以通过两个指针从队首依次顺序比较这两个序列的值并计数,数到第(M+N)/2的那个元素就是中位数了
可能坑点:
#include <iostream>
#include <stdio.h>
#include <vector>
using namespace std;
long s1[1000001];
long s2[1000001];
long getMedian(long s1[],long s2[],int N,int M)
{
int i=0,j=0;
vector<long >vec;
while(vec.size()!=(M+N+1)/2)
{
if(i==N)//这里是个坑
{
vec.push_back(s2[j]);
j++;
}
else if(j==M)//这里是个坑
{
vec.push_back(s1[i]);
i++;
}
else if(s1[i]<=s2[j])
{
vec.push_back(s1[i]);
i++;
}
else if(s1[i]>s2[j])
{
vec.push_back(s2[j]);
j++;
}
}
return vec.back();
}
int main()
{
int N,M;
scanf("%d",&N);
int i=0,j=0;
while(i<N)scanf("%ld",&s1[i++]);
scanf("%d",&M);
while(j<M)scanf("%ld",&s2[j++]);
if(N==0)
{
cout<<s2[(M-1)/2]<<endl;
return 0;
}
if(M==0)
{
cout<<s1[(N-1)/2]<<endl;
return 0;
}
cout<<getMedian(s1,s2,N,M)<<endl;
return 0;
}