1029 有一个测试点内存超限

最新推荐文章于 2024-08-16 08:06:56 发布

Coding18

最新推荐文章于 2024-08-16 08:06:56 发布

阅读量582

点赞数

本文链接：https://blog.youkuaiyun.com/Coding18/article/details/86438469

版权

本文介绍了一种在两个已排序的整数数组中查找中位数的算法。通过比较两个数组的元素，算法逐步确定中位数的位置，最终输出中位数。此算法适用于数据科学和算法设计等领域。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 200004;
const int inf = 1000000000;
int a[maxn],b[maxn];
int main()
{
	int n1,n2;	
	scanf("%d",&n1);
	for(int i = 0; i < n1; i++)
	    scanf("%d",&a[i]);
	    a[n1] = inf;
	scanf("%d",&n2);
	for(int i = 0; i < n2; i++)
     	scanf("%d",&b[i]);
     	b[n2] = inf;
	int mid = (n1 + n2 -1) / 2;
	int temp = 0;
	int i = 0,j = 0;
	while(i < n1 && j < n2)
	{
		if(temp == mid) break;
		else if(a[i] <= b[j])
		{
			i++;
			temp++;
		} 
		else
		{
			j++;
			temp++;
		}
		
	}
	while(i < n1 && j < n2)
	{
		if(a[i] < b[j]) 
		{
			printf("%d\n",a[i]);
			return 0;
		}
		else 
		{
			printf("%d\n",b[j]);
			return 0;
		}
	}
    while(i < n1)
    {
    	if(temp == mid)
    	{
    		printf("%d\n",a[i]);
    		return 0;
    	}
    	
    	else 
		{
			i++;
    	    temp++;
		}
    }
    while(j < n2)
    {
        if(temp == mid)
       	{
    	   	printf("%d\n",b[j]);
    	   	return 0;
        }
     	else 
	   	{
	    	j++; 
		    temp++;
	   	}
    }
	return 0;
}