LeetCode4-寻找两个正序数组的中位数

题意&思路

题意：两个正数单调不减的整数数组，找到这两个数组中的中位数
思路：一开始我考虑直接在两个数组中找到中位数对应的元素，这样节省空间，但是试了一下情况实在是太多了，因为无法知道连续两个数组处于哪个数组因此无法解决总数是偶数的情况，而且还要考虑数组越界，感觉很麻烦，所以我选择把他们先合并到一个数组里，然后直接算出中位数取出来就完事了，不过这个Leetcode不能用ceil函数还是挺坑的

AC代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
            int sum=nums1.size()+nums2.size();
            int mid=0;
            double ans=0;
            vector<int> tmp;
            int i,j,k;
            i=j=0;k=1;
            while(k<=sum)
            {
                while(i<nums1.size()&&j<nums2.size())
                {
                    if(nums1[i]<=nums2[j]){tmp.push_back(nums1[i]);i++;k++;}
                    else {tmp.push_back(nums2[j]);j++;k++;}
                }
                while(i<nums1.size())
                {
                   tmp.push_back(nums1[i]);                    
                    k++;i++;                    
                }
                while(j<nums2.size())
                {
                    tmp.push_back(nums2[j]);                    
                    k++;j++;                 
                }
            }
            if(sum&1)
            {
                int mid=(sum+1)/2;
                ans=(double)tmp[mid-1];
            }
            else 
            {
                int mid=sum/2;
                ans=(double)(tmp[mid]+tmp[mid-1])/2.0;
            }
            return ans;
    }
};