首页 > 其他 > 详细

【LeetCode OJ】Median of Two Sorted Arrays

时间:2015-11-04 20:58:45      阅读:266      评论:0      收藏:0      [点我收藏+]

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

题目:There are two sorted arrays nums1 and nums2 of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

解题思路:将两个有序数组合并为一个,设置一个 Map<Integer, Integer>,key值为序号,value值为数值,m+n个数的中位数要分两种情况讨论:

1、m+n为奇数:则(m+n)/2即为其中位数

2、m+n为偶数:则(((m+n)/2-1)+(m+n)/2)/2为中位数

示例代码如下:

public class Solution 
{
    public static double findMedianSortedArrays(int[] nums1, int[] nums2)
    {
        int m=nums1.length;
        int n=nums2.length;
        int index; //中位数的位置
        //m+n为奇数
        double result;
        if(m==1&&n==0)
            return nums1[0];
        if(m==0&&n==1)
            return nums2[0];
        Map<Integer, Integer> map=new TreeMap<Integer, Integer>();
        int i=0,j=0,k=0;
        while(i<m&&j<n)
        {
            if(nums1[i]<=nums2[j])
            {
                map.put(k,nums1[i]);
                i++;
                k++;
            }
            else
            {
                map.put(k,nums2[j]);
                j++;
                k++;
            }
        }
        while(i<m)
        {
            map.put(k,nums1[i]);
            i++;
            k++;
        }
        while(j<n)
        {
            map.put(k,nums2[j]);
            k++;
            j++;
        }
        //m+n为奇数
        if((m+n)%2==1)
        {
            result=map.get((m+n)/2);
            return result;
        }
        else
        {
            result=(double)(map.get((m+n)/2-1)+map.get((m+n)/2))/2;
            return result;
        }
    }
    
}

 

【LeetCode OJ】Median of Two Sorted Arrays

原文:http://www.cnblogs.com/xujian2014/p/4936995.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!