题目1004:Median(查找中位数)

问题来源

  http://ac.jobdu.com/problem.php?pid=1004

问题描述

  给你两个非降序序列,让你求中位数。中位数为第(n+1)/2个数(从0开始计算)。

问题分析

  这个问题有很多种解法,题目放的也很松。
  第一种解法:暴力。直接把两个子串读进一个数组,sort一下取中位数。真是暴力。
  第二种解法:读入两个子串A、B后,再开另外一个数组C,用两个索引依次按次序放入C,再求中位数。
  第三种解法:直接用两个索引p、q指向两个数组的开头,索引的移动可以理解为丢掉前面的数。通过分析我们知道,索引移动次数为(n+m-1)/2,我们要找的中位数就是min(A[p],B[q])。
  注意:解法二三里面有一个坑点,就是可能其中一个索引到头了,这时候只能移动另一个索引了, 最后我们的中位数也是A[p]或者B[q]了,数组越界取值可不行。

参考代码

//
// Created by AlvinZH on 2017/4/24.
// Copyright (c) AlvinZH. All rights reserved.
//

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;

int n,m;
long long a[1000005];
long long b[1000005];

int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=0;i<n;i++)
            scanf("%lld",&a[i]);
        scanf("%d",&m);
        for(int i=0;i<m;i++)
            scanf("%lld",&b[i]);

        long long ans;
        int p=0,q=0;
        for(int i=1;i<=(n+m-1)/2;i++)
        {
            if(p==n) q++;
            else if(q==m) p++;
            else if(a[p]<b[q]) p++;
            else q++;
        }
        if(p==n) ans=b[q];
        else if(q==m) ans=a[p];
        else ans=min(a[p],b[q]);
        printf("%lld\n",ans);
    }
}

 

作者: AlvinZH

出处: http://www.cnblogs.com/AlvinZH/

本人Github:https://github.com/Pacsiy/JobDu

本文版权归作者AlvinZH和博客园所有,欢迎转载和商用,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利.

转载于:https://www.cnblogs.com/AlvinZH/p/6759765.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目描述: 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2,请你找出这两个有序数组的中位数,并且要算法的时间复杂度为 O(log(m + n))。 示例 1: nums1 = [1, 3] nums2 = [2] 则中位数是 2.0 示例 2: nums1 = [1, 2] nums2 = [3, 4] 则中位数是 (2 + 3)/2 = 2.5 解题思路: 1. 将两个数组合并成一个数组,再对合并后的数组进行排序,找到中位数 时间复杂度为O(m+n)log(m+n),不符合题目 2. 利用二分法查找 首先,我们可以理解中位数的定义:如果某个有序数组的长度是奇数,那么其中位数就是最中间那个元素,如果是偶数,那么就是最中间两个元素的平均值。 在本题中,我们需要找到两个有序数组 A 和 B 的中位数,我们可以考虑从中位数的定义入手。如果我们能够将两个数组分别分成两部分,并且使得左半部分和右半部分分别满足以下两个条件: 1.左半部分的所有元素都小于右半部分的所有元素。 2.左半部分和右半部分分别包含了 A 和 B 数组的一半元素。 那么中位数就可以通过以下公式得到: median = max(left_part) (len(A) + len(B) 为偶数) median = (max(left_part) + min(right_part))/2 (len(A) + len(B) 为奇数) 为了使得划分满足上述两个条件,我们可以对较短的那个数组进行二分查找,然后根据查找到的位置将两个数组进行划分,最后再根据上述公式计算中位数。 具体实现见代码。 Java代码: class Solution { public double findMedianSortedArrays(int[] nums1, int[] nums2) { int m = nums1.length; int n = nums2.length; //确保nums1比nums2短 if (m > n) { int[] temp = nums1; nums1 = nums2; nums2 = temp; int tmp = m; m = n; n = tmp; } int iMin = 0, iMax = m; while (iMin <= iMax) { int i = (iMin + iMax) / 2; int j = (m + n + 1) / 2 - i; if (i < iMax && nums2[j - 1] > nums1[i]) { iMin = i + 1; } else if (i > iMin && nums1[i - 1] > nums2[j]) { iMax = i - 1; } else { int maxLeft = 0; if (i == 0) { maxLeft = nums2[j - 1]; } else if (j == 0) { maxLeft = nums1[i - 1]; } else { maxLeft = Math.max(nums1[i - 1], nums2[j - 1]); } if ((m + n) % 2 == 1) { return maxLeft; } int minRight = 0; if (i == m) { minRight = nums2[j]; } else if (j == n) { minRight = nums1[i]; } else { minRight = Math.min(nums1[i], nums2[j]); } return (maxLeft + minRight) / 2.0; } } return 0.0; } }

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值