LeetCode 4. 寻找两个有序数组的中位数

最新推荐文章于 2021-03-04 16:18:24 发布

LIGHTINEYES

最新推荐文章于 2021-03-04 16:18:24 发布

阅读量104

点赞数

分类专栏： C# LeetCode 文章标签： LeetCode C#

本文链接：https://blog.csdn.net/shuai1210/article/details/103089913

版权

C# 同时被 2 个专栏收录

25 篇文章 0 订阅

订阅专栏

LeetCode

11 篇文章 0 订阅

订阅专栏

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数，

假设 nums1 和 nums2 不会同时为空。

        public static double FindMedianFromTwoArray(int[] nums1, int[] nums2) {
            int m = nums1.Length, n = nums2.Length;

            double ret = int.MinValue;

            if ((m + n) % 2 != 0) {//奇数
                int midIndex = (m + n) / 2 + 1;//中位数index
                int idm = 0, idn = 0;//两个数组的遍历标记位
                for (int i = 0; i < midIndex; i++) {
                    bool isOne = false;
                    if (idm < m) {//如果第一个数组可取值
                        ret = nums1[idm];
                        isOne = true;
                    }
                    if (idn < n) {//第二个数组可取值，需要和第一个数组取值比较大小
                        if (isOne) {
                            if (ret > nums2[idn]) {
                                ret = nums2[idn];
                                isOne = false;
                            }
                        }
                        else {
                            ret = nums2[idn];
                        }
                    }
                    if (isOne) {//修改相应的标记位
                        idm++;
                    }
                    else {
                        idn++;
                    }
                }
            }
            else {//偶数
                int mIndex1 = (m + n) / 2, mIndex2 = mIndex1 + 1;
                int idm = 0, idn = 0;
                for (int i = 0; i < mIndex2; i++) {
                    int num = int.MaxValue;
                    bool isOne = false;
                    if (idm < m) {
                        num = nums1[idm];
                        isOne = true;
                    }
                    if (idn < n) {
                        if (isOne) {
                            if (num > nums2[idn]) {
                                num = nums2[idn];
                                isOne = false;
                            }
                        }
                        else {
                            num = nums2[idn];
                        }
                    }
                    if (isOne) {
                        idm++;
                    }
                    else {
                        idn++;
                    }
                    //偶数个数的中位数是中间两个数和的一半
                    if (idm + idn == mIndex2) {
                        ret = (ret + num) / 2;
                    }
                    else {
                        ret = num;
                    }
                }
            }
            return ret;
        }