1029. Median 解析

最新推荐文章于 2022-03-10 09:26:03 发布

原创最新推荐文章于 2022-03-10 09:26:03 发布 · 462 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#PAT

PAT 专栏收录该内容

138 篇文章

订阅专栏

本文介绍了一种通过合并两个已排序的长整型数组来找到中位数的算法。使用了C++语言，并利用了标准输入输出进行数据交互。算法首先读取两个数组的长度，然后分别读取数组元素，并通过迭代比较的方式确定中位数。

并归就好~~估计数据大直接用scanf了怕cin超时

#include <iostream>
#include <string>
#include <vector>


using namespace std;

vector <long long> l1;
vector <long long> l2;

int N1, N2;

int main() {
	long long temp = 0;
	
	scanf("%d", &N1);
	for (int i = 0; i < N1; i++) {
		scanf("%lld", &temp);
		l1.push_back(temp);
	}

	scanf("%d", &N2);
	for (int i = 0; i < N2; i++) {
		scanf("%lld", &temp);
		l2.push_back(temp);
	}

	int i = 0, j = 0;
	int len = N1 + N2 + 1;
	len /= 2;
	len;

	int mid;

	bool isFind = false;
	int count = 0;
	
	while (i < N1 && j < N2 && !isFind) {
		
		if (l1[i] < l2[j]) {
			count++;
			if (count == len) {
				mid = l1[i];
				isFind = true;
				break;
			}
			i++;
		}
		else {
			
			count++;
			if (count == len) {
				mid = l2[j];
				isFind = true;
				break;
			}
			j++;
		}
		
	}
	
	if (!isFind) {
		while (i < N1) {
		
			count++;
			if (count == len) {
				mid = l1[i];
				isFind = true;
				break;
			}
			i++;
		}
	}

	if (!isFind) {
		while (j < N2) {
		
			count++;
			if (count == len) {
				mid = l2[j];
				isFind = true;
				break;
			}
			j++;
		}
	}

	cout << mid << endl;

	system("pause");

	return 0;
}