1029 Median(25 分)

最后一位设为INF,这样比较的时候就不用判断i和j是不是到达了k1和k2
(没有AC,最后内存超限,为什么要卡这个,,,感觉好没意思)

#include<iostream>
#include<algorithm>
using namespace std;
const int INF = 1e9 + 10;
const int maxn = 2e5 + 10;
int a[maxn], b[maxn], k1, k2;
int main()
{
    scanf("%d", &k1);
    for (int i = 0; i < k1; i++)scanf("%d", &a[i]);
    a[k1] = INF;
    scanf("%d", &k2);
    for (int i = 0; i < k2; i++)scanf("%d", &b[i]);
    b[k2] = INF;
    int mid = (k1 + k2 - 1) / 2;//找的是答案的前一个
    int cnt = 0, i = 0, j = 0;
    while (cnt < mid)
    {
        if (a[i] < b[j])i++;
        else j++;
        cnt++;
    }
    if (a[i] < b[j])printf("%d", a[i]);//这里找最后一个
    else printf("%d", b[j]);
    return 0;
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。