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() 递归式缩容无输入依据,属于误解题意。

正确解法需分场景选择

CreateWise AI
CreateWise AI

为播客创作者设计的AI创作工具,AI自动去口癖、提交亮点和生成Show notes、标题等

下载
场景 推荐方法 时间复杂度 适用性
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 小」问题。初学者可从快选入手理解,进阶务必掌握主席树或整体二分——它们是解决大规模区间统计查询的基石工具

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
sort排序函数用法
sort排序函数用法

sort排序函数的用法:1、对列表进行排序,默认情况下,sort函数按升序排序,因此最终输出的结果是按从小到大的顺序排列的;2、对元组进行排序,默认情况下,sort函数按元素的大小进行排序,因此最终输出的结果是按从小到大的顺序排列的;3、对字典进行排序,由于字典是无序的,因此排序后的结果仍然是原来的字典,使用一个lambda表达式作为key参数的值,用于指定排序的依据。

409

2023.09.04

string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1051

2023.08.02

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

614

2024.08.29

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

335

2025.08.29

C++中int的含义
C++中int的含义

本专题整合了C++中int相关内容,阅读专题下面的文章了解更多详细内容。

235

2025.08.29

string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1051

2023.08.02

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

614

2024.08.29

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

335

2025.08.29

TypeScript类型系统进阶与大型前端项目实践
TypeScript类型系统进阶与大型前端项目实践

本专题围绕 TypeScript 在大型前端项目中的应用展开,深入讲解类型系统设计与工程化开发方法。内容包括泛型与高级类型、类型推断机制、声明文件编写、模块化结构设计以及代码规范管理。通过真实项目案例分析,帮助开发者构建类型安全、结构清晰、易维护的前端工程体系,提高团队协作效率与代码质量。

26

2026.03.13

热门下载

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

精品课程

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

共23课时 | 4.4万人学习

C# 教程
C# 教程

共94课时 | 11.3万人学习

Java 教程
Java 教程

共578课时 | 81.5万人学习

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

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