分治法找两个序列中找第k小

最新推荐文章于 2023-05-04 21:30:05 发布
最新推荐文章于 2023-05-04 21:30:05 发布
阅读量1.1k 收藏 7
点赞数 2
分类专栏： 算法之路 算法设计与分析 文章标签： 算法 数据结构 分治算法
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_43800577/article/details/109609011
版权

两个有序数序列中找第k小


Abstrct

在计算机科学中，分治法是建基于多项分支递归的一种很重要的算法范型。字面上的解释是“分而治之”，就是把一个复杂的问题分成两个或更多的相同或相似的子问题，直到最后子问题可以简单的直接求解，原问题的解即子问题的解的合并。

这个技巧是很多高效算法的基础，如排序算法（归并排序、快速排序）、傅立叶变换（快速傅立叶变换）。

另一方面，理解及设计分治法算法的能力需要一定时间去掌握。正如以归纳法去证明一个理论，为了使递归能够推行，很多时候需要用一个较为概括或复杂的问题去取代原有问题。而且并没有一个系统性的方法去适当地概括问题。

以上这段话引自维基百科，下面我们根据分治算法的思想，构造求解“两个有序数序列中找第k小”问题的思路。


Problem Description

已知两个已经排好序（非减序）的序列X和Y，其中X的长度为m，Y长度为n，
现在请你用分治算法，找出X和Y的第k小的数，算法时间复杂度为O(max{logm, logn})。

此题请勿采用将序列X和Y合并找第k小的O(m+n)的一般方法，要充分利用X和Y已经排好序的这一特性。


Input

第一行有三个数，分别是长度m、长度n和k，中间空格相连（1<=m,n<=100000; 1<=k<=m+n）。
第二行m个数分别是非减序的序列X。第三行n个数分别是非减序的序列Y。


Output

序列X和Y的第k小的数。


Sample Input

5 6 7
1 8 12 12 21 
4 12 20 22 26 31

Sample Output

20

Hint

提示陆续写上来，不着急，先自行思考和讨论……


Solution

如果不用分治算法，我们可以直接将X序列和Y序列合并，然后for循环一遍，就能够找到第k小的元素，时间复杂度O(m+n)。

如果采用分治方法，我们应该怎么做呢？

在这里插入图片描述

首先我们假设：序列X为X [XBegin…XEnd]，而序列Y为Y [YBegin…YEnd]。

将序列X和Y都均分2段，即取X序列中间位置为 Xmid (Xmid =( xBegin+xEnd ) / 2 )，这样就将序列X分为X [XBegin…Xmid]和X [Xmid+1…XEnd] 两部分。同理，也取序列Y中间位置为Ymid，分为Y [YBegin…Ymid]和Y [Ymid+1…YEnd] 两部分。

记录序列X左段和序列Y左段元素个数合计为halfLen，即halfLen = Xmid-XBegin+1+Ymid-YBegin+1。

此时需要比较X[Xmid]和Y[Ymid]：

  1. 当X[Xmid] < Y[Ymid]时，在合并的数组中，原X[xBegin…Xmid]所有元素一定在Y[Ymid]的左侧，
    （1） 若k < halfLen，则此时第k大的元素一定不会大于Y[Ymid]这个元素，
    故以后没有必要搜索 Y[Ymid…YEnd]这些元素，可弃Y后半段数据。
    此时只需递归的对X序列+Y序列的前半段，去搜索第k小的数。

    （2） 若k >= halfLen，则此时第k大的元素一定不会小于X[Xmid]这个元素，
    故以后没有必要搜索 X[XBegin…Xmid]这些元素，可弃X前半段数据。
    此时只需递归的对X序列的后半段+Y序列，去搜索第 k-(Xmid-XBegin+1）小的数。

  2. 当X[Xmid] >= Y[Ymid]时，在合并的数组中，原Y[YBegin…Ymid]的所有元素一定在X[Xmid]的左侧，
    （1） 若k < halfLen，则此时第k大的元素一定不会大于X[Xmid]这个元素，
    故以后没有必要搜索 X[Xmid…xEnd]这些元素，可弃X后半段数据。
    此时只需递归的对X序列的前半段+Y序列，去搜索第k小的数。

    （2） 若k >= halfLen，则此时第k大的元素一定不会小于Y[Ymid]这个元素，
    故以后没有必要搜索 Y[YBegin…Ymid]这些元素，可弃Y前半段数据。
    此时只需递归的对X序列+Y序列的后半段，去搜索第 k-(Ymid-YBegin+1）小的数。

至于递归的边界如下：

if (XBegin > XEnd) return Y[YBegin + k - 1];  //X序列为空时，直接返回Y序列的第k小元素。
if (YBegin > YEnd) return X[XBegin + k - 1];  //Y序列为空时，直接返回X序列的第k小元素。

在这里插入图片描述

我们再回顾一下以上思路~


首先我们假设：序列X为X [XBegin...XEnd]，而序列Y为Y [YBegin...YEnd]。

将序列X和Y都均分2段，即取X序列中间位置为 Xmid (Xmid =( xBegin+XEnd ) / 2 )，这样就将序列X分为X [XBegin...Xmid]和X [Xmid+1...XEnd] 两部分。同理，也取序列Y中间位置为Ymid，分为Y [YBegin...Ymid]和Y [Ymid+1...YEnd] 两部分。

记录序列X左段和序列Y左段元素个数合计为halfLen，即halfLen = Xmid-XBegin+1+Ymid-YBegin+1。

此时需要比较X[Xmid]和Y[Ymid]：

1. 当X[Xmid] < Y[Ymid]时，在合并的数组中，原X[xBegin...Xmid]所有元素一定在Y[Ymid]的左侧，
   （1） 若k < halfLen，则此时第k大的元素一定不会大于Y[Ymid]这个元素，
         故以后没有必要搜索 Y[Ymid...YEnd]这些元素，可弃Y后半段数据。
         此时只需递归的对X序列+Y序列的前半段，去搜索第k小的数。
 
   （2） 若k >= halfLen，则此时第k大的元素一定不会小于X[Xmid]这个元素，
         故以后没有必要搜索 X[XBegin...Xmid]这些元素，可弃X前半段数据。
         此时只需递归的对X序列的后半段+Y序列，去搜索第 k-(Xmid-XBegin+1）小的数。
 
2. 当X[Xmid] >= Y[Ymid]时，在合并的数组中，原Y[YBegin...Ymid]的所有元素一定在X[Xmid]的左侧，
   （1） 若k < halfLen，则此时第k大的元素一定不会大于X[Xmid]这个元素，
         故以后没有必要搜索 X[Xmid...xEnd]这些元素，可弃X后半段数据。
         此时只需递归的对X序列的前半段+Y序列，去搜索第k小的数。
 
   （2） 若k >= halfLen，则此时第k大的元素一定不会小于Y[Ymid]这个元素，
         故以后没有必要搜索 Y[YBegin...Ymid]这些元素，可弃Y前半段数据。
         此时只需递归的对X序列+Y序列的后半段，去搜索第 k-(Ymid-YBegin+1）小的数。


至于递归的边界如下：

if (XBegin > XEnd) return Y[YBegin + k - 1];  //X序列为空时，直接返回Y序列的第k小元素。
if (YBegin > YEnd) return X[XBegin + k - 1];  //Y序列为空时，直接返回X序列的第k小元素。

时间复杂度为 O(max{log n, long m})

在这里插入图片描述


(1)我们来看一下X[Xmid]<Y[Ymid]时，k<halfLen的例子:


在这里插入图片描述


(2)我们来看一下X[Xmid]<Y[Ymid]时，k>=halfLen的例子:

在这里插入图片描述

(3)我们来看一下X[Xmid]>=Y[Ymid]时，k<halfLen的例子:

在这里插入图片描述


(4)我们来看一下X[Xmid]>=Y[Ymid]时，k>=halfLen的例子:

在这里插入图片描述


分治核心伪代码如下：



if X[Xmid]<Y[Ymid]:
	if k<halfLen:
		将Y右区间丢弃，在序列X+序列Y左区间中分治，寻找第k小的元素
	else:
		在序列X右区间+序列Y中分治,寻找第k-(Xmid-XBegin+1)小的元素
else:
	if k<halfLen:
		将X右区间丢弃，在序列X左区间+序列Y中分治，寻找第k小的元素
	else:
		在序列X+序列Y右区间中分治,寻找第k-(Xmid-XBegin+1)小的元素

Code

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;
typedef long long ll;

const int maxn=1e5+100;
const int N=1e6+100;

int n,m,T;
int X[N],Y[N];
int k;

int split(int XBegin,int XEnd,int YBegin,int YEnd,int k) //分治算法
{
    if(XBegin>XEnd)return Y[YBegin+k-1]; // X序列为空时，直接返回Y序列的第k小元素
    if(YBegin>YEnd)return X[XBegin+k-1]; // Y序列为空时，直接返回X序列的第k小元素

    int Xmid=(XBegin+XEnd)>>1; // 求X序列中间位置
    int Ymid=(YBegin+YEnd)>>1; // 求Y序列中间位置

    int halfLen=(Xmid-XBegin+1)+(Ymid-YBegin+1); // 求X[XBegin...Xmid]与Y[YBegin...Ymid]共有多少元素

    if(X[Xmid]<Y[Ymid]) // 此时X[XBegin...Xmid]必然在Y[mid]的左边
    {
        if(k<halfLen) // 第k小的元素必然不大于Y[mid]，因此可以丢弃Y[Ymid+1...YEnd]区间
            return split(XBegin,XEnd,YBegin,Ymid-1,k);
        else // 第k小的元素必然不小于X[mid]。此时可以丢弃X[XBegin...Xmid]区间,等价于寻找第k-(Xmid-XBegin+1)小元素
            return split(Xmid+1,XEnd,YBegin,YEnd,k-(Xmid-XBegin+1));
    }
    else
    {
        if(k<halfLen) // 第k小的元素必然不大于X[mid]，因此可以丢弃X[Xmid+1...XEnd]区间
            return split(XBegin,Xmid-1,YBegin,YEnd,k);
        else // 第k小的元素必然不小于Y[mid]。此时可以丢弃Y[YBegin...Ymid]区间,等价于寻找第k-(Ymid-YBegin+1)小元素
            return split(XBegin,XEnd,Ymid+1,YEnd,k-(Ymid-YBegin+1));
    }

}



void solve()
{
    scanf("%d%d%d",&n,&m,&k);

    for(int i=0; i<n; i++)
        scanf("%d",X+i);

    for(int i=0; i<m; i++)
        scanf("%d",Y+i);


    int ans=split(0,n-1,0,m-1,k);

    printf("%d\n",ans);

}


int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
#endif
    solve();

    /*
    *
    *　　┏┓　　　┏┓+ +
    *　┏┛┻━━━┛┻┓ + +
    *　┃　　　　　　　┃
    *　┃　　　━　　　┃ ++ + + +
    *  ████━████+
    *  ◥██◤　◥██◤ +
    *　┃　　　┻　　　┃
    *　┃　　　　　　　┃ + +
    *　┗━┓　　　┏━┛
    *　　　┃　　　┃ + + + +Code is far away from 　
    *　　　┃　　　┃ + bug with the animal protecting
    *　　　┃　 　 ┗━━━┓ 神兽保佑,代码无bug　
    *　　　┃ 　　　　　　 ┣┓
    *　　  ┃ 　　　　　 　┏┛
    *　    ┗┓┓┏━┳┓┏┛ + + + +
    *　　　　┃┫┫　┃┫┫
    *　　　　┗┻┛　┗┻┛+ + + +
    */
}


最后感谢小伙伴们的学习噢~

__Wedream__
关注
  • 2
    点赞
  • 踩
  • 7
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
专栏目录
作业6——分治算法求数组第k小元素
Chussuchieh的博客
05-22 1838
1.问题 分治算法求数组第k小元素 2.解析 将输入的数组分成n/5组元素，剩余的元素舍弃，取出每组的中项（即第三个）放在M[]中，取出M的中项mid（排序后取出中项），将原数组中所有数与mid比较，根据比较结果将原数组分成三组，再根据k值与三个数组大小的比较去相应的数组找要求的元素。 3.设计 void sort// 排序小规模的数组范围 void Select( input, size , k...
在两排序数组中寻找第K小的数
weixin_30242907的博客
05-27 303
题目：在两个排序数组中寻找第K小的数 举例： arr1=[1,2,3,4,5],arr2=[3,4,5],k=1 1是所有数中第一小的数，所以返回1 arr1=[1,2,3],arr2=[3,4,5,6],k=4 3是所有数中第4小的数，所以返回3 要求：如果arr1的长度为N，arr2的长度为M,时间复杂度请达到O(log(min{M,N}),额外空间复杂度为O(1) 思路：...
【分治】求第k小值
UtralK的专栏
10-24 1213
根据算导写的。 用快速排序的划分过程，表示出一个元素在数组里是第几大。 然后分情况讨论。 注意细节！！ 如，j=i-l+1;       exit(find(...));的exit；  function Find(l,r,k:longint):longint; var x,i,j,t:longint; begin if
找出第k小的元素
kt1776133839的博客
02-16 774
完整程序： #include<iostream> #include<algorithm> using namespace std; int a[1024]; //采用分治策略找出第k小元素的算法 //在a[left:right]中选择第k小的元素 int select(int left,int right,int k){ //找到了第k小的元素 ...
算法篇：分治法求线性表中第k小的数
s_m_c的博客
05-11 776
//第k小的数 /*算法思想：先进行一次快速排序，根据快速排序作为基准的那个数字排序后的位置，来确定我们要找的第k小的那个数 在当前位置的左边还是右边，如果在左边就往左递归，在右边同理。直到作为基准的那个数的位置的下标刚好是（k-1） （默认下标从零开始），证明已经找到了第k小的数，返回它就ok。*/ #include "tou.h" using namespace std; int n = 0, k = 0; int sort1(int a, int b,std::vector<int> &a
找第K小问题C语言-分治法
09-16
在计算机科学领域，找第K小问题是一种常见的算法问题，其目标是从一组整数中找到第K个最小的元素。这种问题在数据结构和算法分析中占有重要地位，因为它经常出现在排序算法的优化和在线查询系统的设计中。本文将详细...
实验4 分治法1
08-03
1. **元素选择**：给定一个序列和一个整数k，找到序列中第k小的元素及其位置。不需对序列进行排序，需要设计一个有效的算法来找到第k小的元素。 2. **最大子段和问题**：对于一个整数序列，找到和最大的连续子序列...
分治算法求最大值与最小值，找最小元素
11-17
在本题中，我们关注的是如何运用分治算法来找到一组数字中的最大值和最小值，以及如何找出第k个最小元素。这两个问题都可以通过分治思想进行有效解决。 首先，寻找最大值和最小值。传统的做法是遍历整个数组，但...
C语言分治法实现归并排序
08-27
归并排序的基本思想是将待排序序列通过分治法分为若干个有序子序列，然后每两个子序列合并为一个子序列，经过多次合并后整合为一张有序表。这种方法可以将复杂的排序问题分解成多个小的排序问题，从而提高排序效率。...
用分治法实现元素选择
03-24
在这个特定的问题中，我们要利用分治法来实现“元素选择”，即找出线性序列集中第k小的元素及其位置。下面我们将深入探讨这个过程。 首先，理解问题的关键在于如何有效地比较并排序序列中的元素。分治法的基本步骤...
拜托，面试别再问我TopK了！！！
weixin_33912638的博客
09-21 232
前言：本文将介绍随机选择，分治法，减治法的思想，以及TopK问题优化的来龙去脉，原理与细节，保证有收获。 面试中，TopK，是问得比较多的几个问题之一，到底有几种方法，这些方案里蕴含的优化思路究竟是怎么样的，今天和大家聊一聊。 画外音：除非校招，我在面试过程中从不问TopK这个问题，默认大家都知道。 问题描述： 从arr[1, n]这n个数中，找出最...
C++：分治算法之选择问题的选择第k小元素问题
m0_63064861的博客
05-03 5057
具体来说，对于选择第 k 小元素的问题，可以利用分治算法的思想将数组按照主元划分为两个部分，并根据主元在数组中的位置关系，决定继续处理数组的哪一部分。可以使用递归的方式实现分治算法解决选择问题中的选择第 k 小元素问题。快速选择算法的基本思路是选取数组中的一个主元，将数组划分为两个部分，并根据主元在数组中的位置关系，决定继续处理数组的哪一部分。分治算法是一种将问题分成若干个子问题处理的思想，通过将原问题分解为多个规模较小的子问题并逐个解决，最终将子问题的解组合起来得到原问题的解。
实验6 分治法选第K小元素
LLoadin的博客
04-07 896
问题 选出所给数组中第K小的元素 解析 分治法选第K小元素 1、首先将给定数组分为n/5各组，每组最多5个元素。将每小组进行排序，找出每组的中位数 2、用select递归找到所有中位数的中位数t，然后划分数组，t是第K小的元素，划分地区中元素数目比K小1，同时有n-k个元素在另一片区域 3、如果 ln==k，返回t，若小于，递归找到前区域第ln个，否则在后区域递归第ln-k个元素 设计 int p...
【洛谷】P1923 求第 k 小的数（分治+快速排序）
IAMLSL的博客
03-26 2289
题目描述 输入n(n<5000000且n为奇数)个数字ai(0<ai<109)n(n<5000000 且 n 为奇数) 个数字 a_i(0<a_{i}<10^9)n(n<5000000且n为奇数)个数字ai​(0<ai​<109)，输出这些数字的第kkk小的数。最小的数是第000小。 请尽量不要使用 nth_element 来写本题，因为本题的重点在于练习分治算法。 输入输出样例 输入 5 1 4 3 2 1 5 输出 2 题目链接：P1923
选第 k 小元素:特定分治策略
zkpj12的博客
04-19 209
选第 k 小元素:特定分治策略实验报告1.问题2.解析3.设计4.分析5.源码 实验报告 课程名称 《算法分析与设计》 实验日期 2021年 3月29日 实验名称 选第k小元素：特定分治策略 1.问题 [描述算法问题，首选形式化方式（数学语言），其次才是非形式化方式（日常语言）] 在n个无序整数中，找出第k（k≤n）小的数。第k小的数意味着从小到大排在第k位置的数。 2.解析 [问题的理解和推导，可用电子版直接在此编
算法分析与设计 实验6 选第k小元素：特定分治策略
Tryna
04-19 185
1、问题： 设L是n个元素的集合，从L中选取第k小的元素，其中1<=k<=n。 这里第k小的元素是指，当L按从小到大排好序之后，排在第k个位置的元素。 利用特定分治策略选出第k小的元素。 2、解析： 3、设计(核心伪代码)： void Closest_Pair(int left, int right, int flag) { //最近点对距离 先对x排序，flag表示禁用的点 if(left == right) return ; //只剩1个点 if(le
分治算法 求第k小元素 O(n) & O(nlog2^n)
热门推荐
机器学习
11-22 1万+
BFPRT算法：时间复杂度O(n)求第k小的数字（分治算法+快排） 各位小伙伴，由于本篇文章代码太过杂乱。我于 2018年12月25日 对文中介绍的算法进行了重写。点击上面的蓝色字体，可以阅读重写后的文章，修复了一些存在的错误。 最容易想到的算法是采用一种排序算法先将数组按不降的次序排好，然后从排好序的数组中捡出第k个元素。这样的算法在最坏情况下时间复杂度是O(nlog...
分治算法求第k小的元素
最新发布
qq_62185081的博客
05-04 598
将整个问题分解成若干小问题后再分而治之。如果分解得到的子问题相对来说还是太大，则可反复使用分治策略将这些子问题分成更小的同类型子问题，直至产生方便求解的子问题，必要时逐步合并这些子问题的解，从而得到问题的解。通过快速排序，我们可以得到本次的数在整个数组的第几位，再与要求的k进行比较，直到本次快排的中轴数的位置与k相等，就可以得到结果。可以由递归过程来表示，因为分治法就是一种找大规模问题与小规模问题关系的方法，是递归设计的一种具体策略。用分治算法求第k小的数，其本质就是快速排序，不过我们只需要排一半。
从2个有序数组中找第k小那个数
weixin_46838716的博客
04-05 1430
两个升序数组，寻找融合之后升序数组中第k小的那个数。
用分治法求n个元素序列中第2大的元素
03-23
4. 将原序列按照中位数m进行划分，得到两个子序列L和R。 5. 如果L的长度大于等于k，则在L中递归调用快速选择算法，寻找第k大的元素。 6. 如果L的长度小于k，那么在R中递归调用快速选择算法，寻找第k-L.length-1大...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

__Wedream__

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值