经典题find median of two sorted array

最新推荐文章于 2023-03-27 23:45:00 发布

caven_y

最新推荐文章于 2023-03-27 23:45:00 发布

阅读量505

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/caven_y/article/details/40646427

Leetcode 专栏收录该内容

40 篇文章

订阅专栏

本文详细解析了一个用于查找两个已排序数组中位数的算法，并指出并修正了三个关键错误：长度计算错误、下标计算错误和结果计算错误。通过实例演示了如何正确实现此算法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

10/30

大概写了15分钟， 3个错，两个typo，一个参数，不合格！

public class Solution {
    public double findMedianSortedArrays(int A[], int B[]) {
        int lenA = A.length, lenB = B.length;
        int len = (lenA+lenB);
        if(len%2 == 1){ //err1: typo lenA for len
            return findKthInArray(len/2+1,A,0,lenA,B,0,lenB);
        }else{
            return (findKthInArray(len/2,A,0,lenA,B,0,lenB) + findKthInArray(len/2+1,A,0,lenA,B,0,lenB))/2.0;
        }
    }
    
    public int findKthInArray(int k, int[] A, int startA, int lenA, int[] B, int startB, int lenB){
        // note that lenA <= lenB
        if(lenA > lenB){
            return findKthInArray(k,B,startB,lenB,A,startA,lenA);
        }
        
        // now lenA <= lenB
        if(lenA == 0)       return B[startB+k-1];
        else if(k==1)       return Math.min(A[startA],B[startB]);
        
        int pA = Math.min(k/2, lenA); //pA <= k/2 <= pB
        int pB = k-pA;
        
        int checkA = startA + pA - 1; // err2: lenA for startA
        int checkB = startB + pB - 1;
        
        if(A[checkA] == B[checkB])
            return A[checkA];
        else if(A[checkA] < B[checkB])
            return findKthInArray(k-pA,A, checkA+1, lenA-pA, B, startB, lenB);
        else
            return findKthInArray(k-pB, A, startA, lenA, B, checkB + 1, lenB-pB); // err3: forget about -pB and -pA
        
    }
}