1029 Median (25分)（一个字短！）_cnt

1029 Median (25分)

分析

为了节省时间，先将第一个数组全部输入，而第二个数组一边输入一边找中位数，cnt记录在中位数之前一共有多少个数字，cnt_i用于计数在中位数之前，有多少个数字是属于第一个数组的，cnt_j用于计数在中位数之前有多少个数字是属于第二个数组的，当cnt_i+cnt_j=cnt时，表明中位数已经找到。
当cnt_i为n或者cnt_j为m，则表示第一个数组或者第二个数组中的元素全部位于中位数之前。
当cnt_i+cnt_j=cnt时，若cnt_i为n，则中位数为b，若cnt_j为m，则中位数为a[cnt_i]，若cnt_i<n且cnt_j<m，则中位数为b和a[cnt_i]中较小的那一个。

代码

#include<iostream>
using namespace std;
int n, m, cnt, cnt_i = 0, cnt_j = 0;
long int a[200010], b;
int main() {
 scanf("%d", &n);
 for (int i = 0; i < n; ++i)scanf("%ld", &a[i]);
 scanf("%d", &m);
 if (m)scanf("%ld", &b);
 cnt = (m + n) / 2 + (m + n) % 2 - 1;
 while (cnt_i + cnt_j < cnt) {
  if (cnt_i < n && cnt_j < m) {
   if (b >= a[cnt_i])++cnt_i;
   else {
    scanf("%ld", &b);
    ++cnt_j;
   }
  }
  if (cnt_i == n)++cnt_j;
  if (cnt_j == m)++cnt_i;
 }
 if (cnt_i < n && cnt_j < m)printf("%ld", a[cnt_i] < b ? a[cnt_i] : b);
 else if (cnt_i == n)printf("%ld", b);
 else printf("%ld", a[cnt_i]);
 return 0;
}