Codeforces Round 922 (Div. 2) D. Blocking Elements 二分答案、动态规划、单调队列

题目链接

题目大意

        给一个长度为 n （1<= n <=10^5) 的数组 a （1<=ai<=10^9) ，需要删除一些数，使数组被分为若干段，这些段的和的最大值为 x，并且设删除的数的和为 y，则价值为max(x,y)。求能得到的最小价值。

思路

      总体上二分答案。用 dp[ i ]表示考虑前 i 个数，并且删除第 i 个数，最大区间和不大于x（二分的值）的最小价值。则有，dp[ i ]=a[ i ]+dp[ j ]，j 为上一个删除的数，此时要保证 j+1 ~i-1区间内的数之和要小于x，且dp[j] 是满足条件下能转移的最小的dp值，这可用单调队列快速获得。

        注意，判断时，判断的是dp[ n+1]是否小与等于x（二分的值），这样可以使得第n个元素是有可能是在区间内的而不是一定被删掉，故要用到a[n+1]，要将其初始为0。

code

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e5 + 10;
int a[N], s[N];
int n;

bool cheak(int x) {
	vector<int>dp(n + 10);
	deque<int>q;
	q.push_back(0);//队列放dp数组的下标
	for (int i = 1; i <= n+1; ++i) {
		while (q.size() && (s[i - 1] - s[q.front()]) > x) q.pop_front();
		//当区间和大于x时，不合法，pop

		dp[i] = a[i] + dp[q.front()];
		//转移，当前的dp[i]由最小的合法的上一个dp值转移而来

		while (q.size() && dp[q.back()] >= dp[i]) q.pop_back();
		//实现单调队列，当前要加入的dp[i]下标比队列里的都大，
		// 要是队列尾部的dp值大于或等于dp[i]的话，dp[i]比他们更优，
		// 所以pop掉，使得每次拿出来的都是最优的dp值

		q.push_back(i);
	}
	return dp[n + 1] <= x;
}
void solve() {
	cin >> n;
	for (int i = 1; i <= n; ++i) {
		cin >> a[i];
		s[i] = s[i - 1] + a[i];
	}
	a[n + 1] = 0;//有多组数据，故要初始化

	int l = 1, r = s[n];
	while (l < r) {
		int mid = l + r >> 1;
		if (cheak(mid))r = mid;
		else l = mid + 1;
	}
	cout << l << '\n';
	return;
}
signed main() {
	ios::sync_with_stdio(0);
	cin.tie(0), cout.tie(0);
	int t = 1;
	cin >> t;
	while (t--) solve();
	return 0;
}