CF1370D.Odd-Even Subsequence 题解 动态规划+二分

题目链接:https://codeforces.com/problemset/problem/1370/D

解题思路:

一开始的想法是定义状态 \(dp_{i,j}\) 表示“前 \(i\) 个数,选了第 \(i\) 个数,共选了 \(j\) 个数的最小值”。则状态转移方程为(没有验证过):

\[dp_{i,j} = \min_{k \in [0,i-2]} \{ dp_{k,j-2}, a_i \} \]

但是就算状态压缩,遍历的状态就达到了 \(O(n^2)\),肯定是超时的。

然后考虑二分答案+DP,假设最大值较小的那个序列的最大值取 \(D\),然后 \(dp[i]\) 表示 \(a_i\) 对应的那个奇偶性的子序列的最小值不超过 \(D\) 并且选了 \(a_i\) 的情况下最多能选多少数。

状态转移方程为:

\(dp[i] = \max \{ dp[k] \} + 2\)

其中 \(k\) 满足 \(k \le i-2\) 且 \(dp[k] \le D\)。

状态转移可以用 单调队列/单调栈 (抱歉,只需要一个到 \(i-2\) 的数表示满足条件的最大值即可)优化。

则最终只要存在一个 \(dp[i] \ge k\) 或者存在一个 \(i \lt n\) 满足 \(dp[i] \ge k-1\),则 \(D\) 可作为一个正确的答案,对此进行二分。

示例代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn = 200020;
int n, k, a[maxn], dp[maxn];
bool check(int D) {
    int tmp = 0;
    for (int i = 1; i <= n; i ++) {
        if (i == 1) {
            dp[i] = (a[i] <= D) ? 1 : 0;
        }
        else {
            tmp = max(tmp, dp[i-2]);
            if (a[i] > D) dp[i] = 0;
            else dp[i] = tmp + 2;
        }
        if (i < n && dp[i] >= k-1 || dp[i] >= k) return true;
    }
    return false;
}
int main() {
    scanf("%d%d", &n, &k);
    int L = 1, R = 0, res;
    for (int i = 1; i <= n; i ++) {
        scanf("%d", a+i);
        R = max(R, a[i]);
    }
    while (L <= R) {
        int mid = (L + R) / 2;
        if (check(mid)) {
            res = mid;
            R = mid - 1;
        }
        else L = mid + 1;
    }
    printf("%d\n", res);
    return 0;
}
上一篇:fastDFS环境的搭建


下一篇:D. Maximum Sum on Even Positions(翻转1次,求最大偶数位和)