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;
}