pat1029Median (25)

来源:互联网 发布:linux 查看服务器类型 编辑:程序博客网 时间:2024/05/17 07:33

题意分析:

(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;}


0 0
原创粉丝点击