D. Med-imize (二分,dp,中位数)

D. Med-imize

在这里插入图片描述

思路:

题意是一个长度为n的数组,可以一次从中删掉k个连续的数,一直到剩下的数<=k为止,求剩下的数的中位数的最大值。

二分答案法,check函数:若x可以作为中位数,则剩余的数中大于等于x的数量要多于半数。
如何统计剩余的数中大于等于x的数量呢?可以使用动态规划。dp[i]保存最终数组中前i个数中>=x的数量。
首先发现,每次删除掉k个数,最终子序列的长度一定为 len =(n-1)%k +1
因为最终子序列中任两个数之间的间隔一定是k的倍数。那么将原数组按照%k分为k组,第i个数在最终数组中的位置就是(i - 1) % k + 1
记子序列的下标为 s t st st,原数组的下标为 i i i,st = (i - 1) % k + 1
从前往后,转移方程dp[st] = max(dp[st], dp[st - 1] + (a[i] >= mid))

代码:
#include <bits/stdc++.h>
#define endl '\n'
#define int long long
typedef long long ll;
using namespace std;

int a[500005];
int n, k;
int dp[500005];
int c[500005];
bool check(int mid) {
	int len = (n - 1) % k + 1;  //  最终序列长度
	int pd = len / 2 + 1; 	//  mid可行条件:最终序列中大于等于mid的数至少有pd个
	for (int i = 0; i <= k; i++) {
		dp[i] = 0;
	}
	for (int i = 1; i <= n; i++) {
		int st = (i - 1) % k + 1;
		dp[st] = max(dp[st], dp[st - 1] + (a[i] >= mid));
	}
	return dp[len] >= pd;
}

void solve() {
	int maxn = 0;
	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		maxn = max(maxn, a[i]);
	}
	if (n <= k) {  // 不足k个数直接排序输出答案
		sort(a + 1, a + n + 1);
		cout << a[(n + 1) / 2] << endl;
		return;
	}
	int l = 1, r = maxn;
	while (l <= r) {
		int mid = (l + r) / 2;
		if (check(mid))l = mid + 1;
		else r = mid - 1;
	}
	cout << r << endl;
}

signed main() {
	cin.tie(0)->ios::sync_with_stdio(0);
	int T = 1;
	cin >> T;
	while (T--) {
		solve();
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值