LeetCode 4. Median of Two Sorted Arrays

来源:互联网 发布:linux cpu核心使用率 编辑:程序博客网 时间:2024/05/16 11:19

这道题很经典,它归类在分治算法,感觉和求一堆数的第k大差不多,不过这里是有两个已经排好序的数组,所以比较麻烦一些。整体的思路如下:
1、要求出两个数列的中间数,可以转换成求第k小(大)的数;
2、要求第k小的数,可以分为将第一个数列的k/2个数和第二个数列的k/2个数进行对比,小的那一个数列的左边k/2个数就可以丢弃了,如此递归下去,递归的终止条件为:(1)如果有一个数列的长度为0,则取出另一个数列的第k大;(2)如果k=1,则直接比较两个数列的第一个数然后取出较小的数即可;(3)这里可以假设第一个数列总是比第二个数列要短,方便写代码。
代码如下:

int min(int a, int b){    return a<b?a:b;}double FindKthSmallest(int* num1, int m, int* num2, int n, int k){    if(m > n) //保证num1总是比num2短        return FindKthSmallest(num2, n, num1, m, k);    if(m == 0)    {        return num2[k-1];    }     if(k == 1)//这里防止了下面的pa-1小于0    {        return min(num1[0],num2[0]);    }    int pa = min(m, k/2), pb = k-pa;//pb不能取min(n,k/2),否则过不了[1,2]和[1,2]这种样例    if(num1[pa-1] < num2[pb-1])        return FindKthSmallest(num1+pa, m-pa, num2, n, k-pa);    else if(num1[pa-1] > num2[pb-1])        return FindKthSmallest(num1, m, num2+pb, n-pb, k-pb);    else        return num1[pa-1];}double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){    int k = (nums1Size+nums2Size)/2;    if((nums1Size+nums2Size)%2 == 0)    {           int a = FindKthSmallest(nums1, nums1Size, nums2, nums2Size, k);        int b = FindKthSmallest(nums1, nums1Size, nums2, nums2Size, k+1);        return (a+b)/2.0;    }    else         return FindKthSmallest(nums1, nums1Size, nums2, nums2Size, k+1);}
0 0
原创粉丝点击