0

0

高效求解区间中位数:离线处理与快速选择优化教程

碧海醫心

碧海醫心

发布时间:2026-01-03 10:04:34

|

864人浏览过

|

来源于php中文网

原创

高效求解区间中位数:离线处理与快速选择优化教程

本文讲解如何在多组查询下高效计算任意子数组的中位数,针对 n, q ≤ 5×10⁴ 的约束,指出暴力排序的 o(q·n log n) 不可接受,并给出基于「整体二分」或「主席树」的正解思路,同时修正原始代码中的逻辑错误与边界问题。

在处理大量区间中位数查询(如本题中 Q 可达 5×10⁴)时,对每个查询都复制子数组并调用 Arrays.sort() 是严重低效的——单次排序最坏 O(len log len),最坏总复杂度可达 O(Q·N log N) ≈ 5×10⁴ × 6×10⁴ × log₂(6×10⁴) > 10⁹ 操作,在 Java 中必然超时。

更关键的是,原始代码存在多重逻辑错误

  • ❌ median(int N, int[] A) 中 Math.ceil(N / 2) 错误:N/2 是整数除法(如 5/2 = 2),Math.ceil(2) = 2,但题目明确要求 1-indexed 下第 ⌈len/2⌉ 个元素(即下标 ⌈len/2⌉ - 1)。正确写法应为 (N - 1) / 2(奇偶统一)或 (N + 1) / 2 - 1。
  • ❌ 示例输入 {2,4,5,3,1,6} 对应查询 (L,R) 未说明,但输出 3, 4, 5 暗示三组查询(如 [1,3]→[2,4,5]→中位数4? 实际应为 4;而 [1,5]→[1,2,3,4,5]→中位数3)。原始代码 getMedian() 并非按 (L,R) 查询,而是不断截取 A[1..mid+1],与题意完全不符。
  • ❌ getMedian() 递归式缩容无输入依据,属于误解题意。

正确解法需分场景选择

MedPeer
MedPeer

AI驱动的一站式科研服务平台

下载
场景 推荐方法 时间复杂度 适用性
Q 较小(≤10³)或 N 较小(≤10³) 暴力提取 + 快速选择(nth_element 思想) O(Q·len) 平均 简单直接,Java 可用 Collections.nthElement 模拟(需转 List)或手写快选
Q, N ≤ 5×10⁴,强制在线 主席树(可持久化线段树) O((N+Q) log C) 支持任意区间第 k 小,k = (R−L+1+1)/2
Q, N ≤ 5×10⁴,允许离线 整体二分 + 树状数组 O((N+Q) log C log N) 更易实现,常数小

? 推荐实践:使用快速选择优化单次查询(教学友好版)
虽最坏 O(Q·N),但在随机数据下表现良好,且代码简洁,适合理解核心逻辑:

import java.util.*;

public class MedianQueries {
    // 对子数组 A[L..R](1-indexed)求中位数:第 k 小,k = (length+1)/2
    public static int queryMedian(int[] A, int L, int R) {
        int len = R - L + 1;
        int k = (len + 1) / 2; // 1-indexed 第 k 小 → 0-indexed 第 k-1 个
        int[] sub = Arrays.copyOfRange(A, L - 1, R); // 转为0-indexed切片
        return quickSelect(sub, 0, sub.length - 1, k - 1);
    }

    private static int quickSelect(int[] arr, int left, int right, int k) {
        if (left == right) return arr[left];
        int pivotIndex = partition(arr, left, right);
        if (k == pivotIndex) {
            return arr[k];
        } else if (k < pivotIndex) {
            return quickSelect(arr, left, pivotIndex - 1, k);
        } else {
            return quickSelect(arr, pivotIndex + 1, right, k);
        }
    }

    private static int partition(int[] arr, int left, int right) {
        int pivot = arr[right];
        int i = left;
        for (int j = left; j < right; j++) {
            if (arr[j] <= pivot) {
                swap(arr, i++, j);
            }
        }
        swap(arr, i, right);
        return i;
    }

    private static void swap(int[] arr, int i, int j) {
        int t = arr[i]; arr[i] = arr[j]; arr[j] = t;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int[] A = new int[N];
        for (int i = 0; i < N; i++) A[i] = sc.nextInt();
        int Q = sc.nextInt();
        while (Q-- > 0) {
            int L = sc.nextInt(), R = sc.nextInt();
            System.out.println(queryMedian(A, L, R));
        }
    }
}

⚠️ 注意事项

  • 题目中“ceil(len/2)”在 1-indexed 下等价于 (len + 1) / 2(整数除法),例如 len=5 → 3,len=6 → 3(不是 3.5 向上取整为 4!因为中位数定义为第 ⌈len/2⌉ 小,6 个数的中位数是第 3 小,非平均);
  • Java 中无内置 nth_element,quickSelect 平均 O(n),最坏 O(n²),可通过随机化 pivot 优化;
  • 真正工业级解法应采用 主席树:预处理 O(N log C),单次查询 O(log C),C 为值域(需离散化),总复杂度 O(N log C + Q log C),稳定通过 5×10⁴ 数据。

总结:本题本质是「静态区间第 K 小」问题。初学者可从快选入手理解,进阶务必掌握主席树或整体二分——它们是解决大规模区间统计查询的基石工具

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

826

2023.06.15

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

727

2023.07.05

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

732

2023.07.31

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

396

2023.08.01

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

398

2023.08.02

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

445

2023.08.02

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

429

2023.08.02

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

16884

2023.08.03

php源码安装教程大全
php源码安装教程大全

本专题整合了php源码安装教程,阅读专题下面的文章了解更多详细内容。

177

2025.12.31

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Kotlin 教程
Kotlin 教程

共23课时 | 2.2万人学习

C# 教程
C# 教程

共94课时 | 5.8万人学习

Java 教程
Java 教程

共578课时 | 41万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号