[leetcode] find median of two sorted arrays

最新推荐文章于 2025-07-21 23:35:20 发布

iteye_15479

最新推荐文章于 2025-07-21 23:35:20 发布

阅读量130

点赞数

分类专栏： java algorithm 文章标签： java

java 同时被 2 个专栏收录

80 篇文章

订阅专栏

algorithm

34 篇文章

订阅专栏

本文提供了一种Java解决方案来寻找两个已排序数组中的中位数。通过比较两个数组元素并进行迭代，该方法能够在O(log(min(m, n)))的时间复杂度内找到中位数，其中m和n分别是两个数组的长度。

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

[url]http://leetcode.com/onlinejudge#question_4[/url]

Solution: a little ugly.


public class Solution {
    public double findMedianSortedArrays(int A[], int B[]) {
        // Start typing your Java solution below
        // DO NOT write main() function
        int len1 = A.length;
        int len2 = B.length;
        int i = 0; 
        int j = 0;
        int k = 0;
        double v = 0;
        double pre = 0;

        boolean odd = (len1 + len2) % 2 == 1;
        int mid = (len1 + len2) / 2 + 1;

        while(i<len1 && j<len2 && k < mid){
            if(A[i] < B[j]) { 
                v = A[i];
                if(k == mid-2){
                    pre = A[i];
                }                
                i++;
            }
            else {
                v = B[j];
                if(k == mid-2){
                    pre = B[j];
                }
                j++;                
            }            
            k++;            
        }
        if(k==mid){
            if(odd){
                return v;
            }else {
                return (double)(v + pre)/2;
            }
        }

        while(i<len1 && k < mid){
            v = A[i];
            if(k==mid-2) pre = A[i];
            i++;
            k++;            
        }
        if(k==mid){
            if(odd){
                return v;
            }else {
                return (double)(v + pre)/2;
            }
        }

        while(j<len2 && k<mid){

            v = B[j];
            if(k==mid-2) pre = B[j];
            j++;
            k++;
        }
        if(k==mid){
            if(odd){
                return v;
            }else {
                return (double)(v + pre)/2;
            }
        }
        return v;
    }
}