LeetCode OJ : Median of Two Sorted Arrays

来源：互联网发布：广州知用中学礼仪服编辑：程序博客网时间：2024/06/07 00:55

Median of Two Sorted Arrays

链接：https://leetcode.com/problems/median-of-two-sorted-arrays/

题解：

给两个有序数组A,B，求中位数，要求时间复杂度为O(log (m+n))。

先求中位数的位置idx=(A.length+B.length)/2，分别比较两个数组的最小值哪个最小，弹出那个数，一直弹出idx次。如果全部数字为偶数，则中位数是第idx-1和idx次弹出的数；如果全部数字为奇数，则中位数是第idx次弹出的数。

代码：

public class Solution {    public double findMedianSortedArrays(int A[], int B[]) {        int lena=A.length;        int lenb=B.length;        if(lena+lenb==1){            if(lena==0) return B[0];            else        return A[0];        }        int mid=(lena+lenb)/2;        int idxa=0;        int idxb=0;        int idx=0;        double midValue=0.0,ans=0.0;        for(;idx<=mid;++idx){            ans=midValue;            if(idxa==lena){                midValue=B[idxb];                ++idxb;            }            else if(idxb==lenb){                midValue=A[idxa];                ++idxa;            }            else if(A[idxa]<B[idxb]){                midValue=A[idxa];                ++idxa;            }            else{                midValue=B[idxb];                ++idxb;            }        }        if((lena+lenb)%2==1){            return midValue;        }        else{            return (ans+midValue)/2;        }    }}

来源：http://blog.csdn.net/acm_ted/article/details/44488325

0 0