PAT 1029. Median (25) 求2个排序数组的中位数——剔除法

leetcode类似题链接

/*************************
题意：
求2个排序数组的中位数
************************/

/***********************
剔除法，以求第几大元素的角度去看
例如7个元素，则第4大的元素就是中位数。
那么只要数组1和数组2 总共剔除了3个元素后
便可知道第4大是哪一个。
这题leetcode写过题解了，详见链接：
*************************/
/***********************
笔记：

*********************/
#include<iostream>
#include<stdio.h>
#include<string>
#include<vector>
#include<queue>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<stack>
#include<map>
#include<set>
#include<unordered_map>
using namespace std;
#define M 110000
#define INF 0x7fffffff
vector<long long> v1,v2;

int GetKth(int L1,int L2,int K){
	int k1,k2;

	//v1被剔除完了，则直接输出结果
	if(L1>=v1.size()){
		return v2[L2+K-1];
	}
	//v2被剔除完了，则直接输出结果
	else if(L2>=v2.size()){
		return v1[L1+K-1];
	}//求第1大的元素，则可以直接得到
	else if(K==1){ 
		return v1[L1]<v2[L2]?v1[L1]:v2[L2];
	}


	k1 = k2 = K/2; //第5大元素，则判断当前数组1的第2大元素，和数组2的第2大元素
	k1--;  k2--;			  //k1和k2仅仅是对应于下标，故下标要-1
	
	if(k1 >= v1.size())
		k1 = v1.size()-1;
	if(k2 >= v2.size())
		k2 = v2.size()-1;
	
	if(v1[L1+k1] < v2[L2+k2]){
		return GetKth(L1+k1+1,L2,K-k1-1);
	}
	else if(v1[L1+k1] > v2[L2+k2]){
		return GetKth(L1,L2+k2+1,K-k2-1);
	}
	else{ //2者相等，判断一下是否已经可得到结果的情况
		if(k1+1+k2+1==K){ //当二者相加为K时，则是其中的1个（并非K为偶数，因为k1可能因为长度问题被压缩）
			return v1[L1+k1];
		}
		else{   //K为奇数，则k1+k2+2还达不到K，都剔除，向后。
			return GetKth(L1+k1+1, L2+k2+1,K-k1-k2-2);
		}
	}
}

int main(){
	long long n1,n2,num;
	int i;
	cin>>n1;
	for(i=0;i<n1;i++){
		scanf("%lld",&num);
		v1.push_back(num);
	}
	cin>>n2;
	for(i=0;i<n2;i++){
		scanf("%lld",&num);
		v2.push_back(num);
	}
	int K,K1,K2;
	if((n1+n2) % 2 ==1){
		K = (n1+n2)/2 + 1; //要找第K大的元素。
		printf("%lld\n",GetKth(0,0,K));
	}  
	//总数为偶数，则要找2个中位数取平均值
	else {
		K=(n1+n2)/2;  
		printf("%lld\n",GetKth(0,0,K));
	}

	return 0;
}