题解 | #在两个长度相等的排序数组中找到上中位数#

在两个长度相等的排序数组中找到上中位数

https://www.nowcoder.com/practice/6fbe70f3a51d44fa9395cfc49694404f

  1. 用哈希表存储原来的顺序,并且存储个数
  2. 遍历哈希表,感觉数据有点弱,数据是连在一起的,如果不是连在一起我感觉我这破烂代码wa了 Orz
  3. 判断条件看清楚就行,问题不大的

#include <cstdio>
#include <unordered_map>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * find median in two sorted array
     * @param arr1 int整型vector the array1
     * @param arr2 int整型vector the array2
     * @return int整型
     */
    int findMedianinTwoSortedAray(vector<int>& arr1, vector<int>& arr2) {
        // write code here
        unordered_map<int, int> mp;
        int maxn = 0;

        for (int i : arr1)
        {
            mp[i] ++;
            maxn = max(maxn, i);
        }
        for (int i : arr2)
        {
            mp[i] ++;
            maxn = max(maxn, i);
        }
        
        int m = arr1.size() + arr2.size(), k;
        if(m % 1)
        {
            k = m / 2 + 1;
        }
        else k = m / 2;

        for (int i = 0; i < maxn; i ++)
        {   
            //cout << i << ' ' << mp[i] << ' ' << k << endl; 1 2 3 3 4 4 5 6
            if(k - mp[i] == 0) return i;
            else if (k - mp[i] < 0) return i - 1;
            else k -= mp[i]; 
        }
        return 0;
    }
};

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务