力扣HOT100 - 4. 寻找两个正序数组的中位数

解题思路:

两个数组合并,然后根据奇偶返回中位数。

java 复制代码
class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.length;
        int n = nums2.length;
        int[] nums = new int[m + n];
        if (m == 0) {
            if (n % 2 == 0) return (nums2[n / 2 - 1] + nums2[n / 2]) / 2.0;
            else return nums2[n / 2];
        }
        if (n == 0) {
            if (m % 2 == 0) return (nums1[m / 2 - 1] + nums1[m / 2]) / 2.0;
            else return nums1[m / 2];
        }

        int cnt = 0;
        int i = 0, j = 0;
        while (cnt != (m + n)) {
            if (nums1[i] < nums2[j]) nums[cnt++] = nums1[i++];
            else nums[cnt++] = nums2[j++];

            if (i == m) {
                while (j != n)
                    nums[cnt++] = nums2[j++];
            }
            if (j == n) {
                while (i != m)
                    nums[cnt++] = nums1[i++];
            }
        }
        if (cnt % 2 == 0) return (nums[cnt / 2 - 1] + nums[cnt / 2]) / 2.0;
        else return nums[cnt / 2];
    }
}
相关推荐
Felven7 分钟前
A. Eating Game
算法
竹叶青_七弦26 分钟前
从复制粘贴到Harness:Java后端AI编程的工程化之路
java
wuminyu29 分钟前
Linux内核线程调度与虚拟线程调度机制系统级深度剖析
java·linux·c语言·jvm·c++
小溪学编程1 小时前
Java InputStream 详解:从基础到实战
java·python·php
血小板要健康1 小时前
队列 + 宽搜(BFS):二叉树层序遍历 算法总结
java·数据结构·笔记·算法·leetcode·宽度优先
Felven1 小时前
A. Riptide
算法·c 算法
m0_739312871 小时前
四元数、李群SO(3)/李代数so(3)的作用及应用场景
算法·机器人·自动驾驶
A黄俊辉A1 小时前
【无标题】
java
BestHeaker1 小时前
跨企业接口对接:IQDS / CPK 数据解析与协作避坑指南(五)
java·服务器·前端
artificiali2 小时前
880 第4章多元
人工智能·算法