leetcode 4. Median of Two Sorted Arrays

最新推荐文章于 2024-10-27 05:36:49 发布

Albert___

最新推荐文章于 2024-10-27 05:36:49 发布

阅读量247

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/Albert___/article/details/78324360

版权

leetcode 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

给了两个排好序的数组，如果再去把数组合并排序找中间数肯定超时。

这里分别从左端开始比较两个数组，当前个数k=0,i=j=0,如果array1[i]<=array2[j],i++,k++,当k为想要的中间数把它保存；

class Solution {
public:
        double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
            int k=0,i=0,j=0;
            int n1=nums1.size(),n2=nums2.size();
            vector<int>a;
            if(n1+n2==1){
               if(!n1)
                   return nums2[0];
                else return nums1[0];
            }
            while(i<n1&&j<n2)
            {
              double num;
              if(nums1[i]<=nums2[j]){
                 num=nums1[i];
                 i++;
              }
              else {
                   num=nums2[j];
                  j++;
                   }
                if(k==(n1+n2)/2||k==(n1+n2)/2-1)
                    a.push_back(num);
             k++;
            }
            while(i<n1)
            {
              if(k==(n1+n2)/2||k==(n1+n2)/2-1)
                    a.push_back(nums1[i]);
                i++;
                k++;
            }
            while(j<n2)
            {
              if(k==(n1+n2)/2||k==(n1+n2)/2-1)
                    a.push_back(nums2[j]);
                j++;
                k++;
            }
            if((n1+n2)&1)
                return a[1];
            else
                return (a[0]+a[1])/2.0;
        }
};