【leetcode阿里题库】4. Median of Two Sorted Arrays

最新推荐文章于 2023-06-12 16:30:30 发布

原创最新推荐文章于 2023-06-12 16:30:30 发布 · 167 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用双指针技术解决寻找两个已排序数组中位数问题的方法。通过定义两个指针分别在两个数组中移动，找到指定位置的元素，从而求出中位数。这种方法适用于处理大型数据集，能够有效地减少计算复杂度。

解题思路

双指针。

提交代码

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
    	int len1=nums1.length,len2=nums2.length;
    	int p1=(len1+len2-1)/2;
        int p2=(len1+len2)/2;
        
        int num1=findNth(nums1,nums2,p1);
        int num2=findNth(nums1,nums2,p2);
        return (double)(num1+num2)/2;
    }
    
    
    public int findNth(int[] nums1, int[] nums2, int pos){
    	int len1=nums1.length,len2=nums2.length;
    	int p1=0,p2=0;
        while((p1+p2)<pos) {
        	if(p1==len1)	p2++;
        	else if(p2==len2) p1++;
        	else {
        		if(nums1[p1]<nums2[p2])
            		p1++;
            	else
            		p2++;
        	}
        	
        }
        if(p1==len1)	return nums2[p2];
        else if(p2==len2)	return nums1[p1];
        else return nums1[p1]<nums2[p2]?nums1[p1]:nums2[p2];
    }
}