寻找两个有序序列的中位数___O(logn) —— 二分

题目链接:点我啊O(∩_∩)O~~
    已知有两个等长的非降序序列 S 1 S1 S1 S 2 S2 S2,求 S 1 S1 S1 S 2 S2 S2并集的中位数

输入样例1:

5
1 3 5 7 9
2 3 4 5 6

输出样例1:

4

输入样例2:

6
-100 -10 1 1 1 1
-50 0 2 3 4 5

输出样例2:

1

    思想这里要求时间复杂度为 O ( l o g n ) O(logn) O(logn),那么就只能用二分来解决,如何二分呢?
    假设 S 1 S1 S1 S 2 S2 S2的中位数分别为 a a a b b b
① ① :若 a = b a=b a=b,则答案即为该值
② ② :若 a < b a<b a<b,则答案一定在 S 1 S1 S1的右半部分 + S 1 S1 S1的左半部分(即把 S 1 S1 S1放到 S 2 S2 S2之前),即舍弃 S 1 S1 S1 a a a 之前的部分,舍弃 S 2 S2 S2 b b b 之后的部分
③ ③ :若 a > b a>b a>b,同 ② ② 答案一定在 S 1 S1 S1的左半部分 + S 2 S2 S2的右半部分(即把 S 2 S2 S2放到S1之前),即舍弃 S 1 S1 S1 a a a 之后的部分,舍弃 S 2 S2 S2 b b b 之前的部分

同时注意一下 l e f t > = r i g h t left>=right left>=right的时候直接返回就行了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

void f(int a[], int b[], int l1, int r1, int l2, int r2){
	if(l1>=r1 || l2>=r2) {
		printf("%d\n", a[l1]>b[l2]?b[l2]:a[l1]);
		return;
	}
	int mid1 = (l1 + r1) / 2;
	int mid2 = (l2 + r2) / 2;
//	printf("%d %d\n", mid1, mid2);
	if(a[mid1] == b[mid2]){
		printf("%d\n", a[mid1]);
		return;
	}
	else if(a[mid1] < b[mid2]){
		if((l1+r1) & 1) f(a, b, mid1+1, r1, l2, mid2);	//偶数个元素 
		else f(a, b, mid1, r1, l2, mid2);
		
	}
	else {
		if((l2+r2) & 1) f(a, b, l1, mid1, mid2+1, r2);
		else f(a, b, l1, mid1, mid2, r2);
	}
}

int main(){
	int n, a[100010], b[100010];
	scanf("%d", &n);
	for(int i=1; i<=n; i++) scanf("%d", a+i);
	for(int i=1; i<=n; i++) scanf("%d", b+i);
	f(a, b, 1, n, 1, n);
}
题目描述: 给定两个长度为n的有序序列A和B,本题要时间复杂度为O(logn)的算法,找出两个序列中位数。 输入格式: 第一行输入一个整数n,表示序列的长度。 第二行输入n个整数,表示序列A。 第三行输入n个整数,表示序列B。 输出格式: 输出一个整数,表示两个序列中位数。 输入样例: 5 1 3 5 7 9 2 4 6 8 10 输出样例: 5 解题思路: 中位数的定义是:将一个集合分成两个长度相等的子集,其中一个子集中的元素总是大于另一个子集中的元素。根据这个定义,我们可以将两个有序序列分别划分成两个长度相等的子集,使得一个子集中的元素总是大于另一个子集中的元素。假设序列A和序列B的长度都为n,我们可以将序列A和序列B分别划分成两个长度为n/2的子序列,分别为A1、A2和B1、B2。如果A1的中位数小于B1的中位数,那么A1中的元素肯定都小于两个序列中位数,因此可以将A1中的元素全部舍去,将A2和B1、B2组成新的序列,继续进行划分。如果A1的中位数大于B1的中位数,那么B1中的元素肯定都小于两个序列中位数,因此可以将B1中的元素全部舍去,将A1、A2和B2组成新的序列,继续进行划分。如果A1的中位数等于B1的中位数,那么A1和B1中的元素都小于两个序列中位数,因此可以将A1、B1和A2、B2组成新的序列,继续进行划分。每次划分后,序列的长度都会减半,因此时间复杂度为O(logn)。 代码实现:
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值