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