给出a b两个有序数组,已按从小到大排序排好,长度均为2n,求两个数组合并后的中位数。
采用二分的方法,同时二分两个数组来寻找中位数,复杂度为log(2n)。
规定A数组下标为0到2n-1,B数组也是。
定义查询范围la=0,ra=2n-1;lb=0,rb=2n-1;
反复判断a[amid]和b[bmid]的大小。
amid=(la+lb)/2;bmid=(ra+rb)/2;
1.如果ra==la,中位数为(a[amid]+b[bmid])/2.0;//查询数组长度为1时,直接取两个数字的平均数
2.如果a[amid]==b[bmid],则if((ra-la+1)&1)中位数为a[amid](查询数组长度为奇数时,直接取当前数),else中位数为(a[amid]+min(a[amid+1],b[bmid+1]))/2.0(查询数组长度为偶数时,中位数应该是当前数和其后一位的平均数);
3.如果a[amid]<b[bmid],则中位数位于a[amid+1]-a[ra]和b[lb]-b[bmid],将查询范围缩短一半,la=amid+1,rb=bmid;
4.如果a[amid]>b[bmid],则中位数位于a[la]-a[amid]和b[bmid+1]-b[rb],将查询范围缩短一半,ra=amid,lb=bmid+1;