Educational Codeforces Round 88 D. Yet Another Yet Another Task（枚举）

题目链接

传送门

题目大意

选一个区间$[l,r]$，使得$\sum _{l\le i\le r}{a}_i-\underset{l\le i\le r}{\max }{a}_i$最大。

题目分析

自己的分析：
自然的想法就是枚举最大值$a_i$，然后看看最大值能延伸的最左边和最右边（单调栈解决），最后再用某数据结构得到区间内的最大子段和。可惜这个数据结构我不知道怎么维护（网上好像用线段树解决的，点这里），然后瞄了一下$a_i$的范围，可能要用桶排序，但是用了后维护区间很麻烦，然后就不知道怎么做了。。。

正解：
首先解决：给出数组$a$，计算最大区间和$\sum _{l\le i\le r}{a}_i$，允许的时间复杂度为$O(n)$
方法：$dp$，设$d{p}_i$为：以$i$结尾的最大区间和，那么，转移方程为：
$$d{p}_i=max(d{p}_{i-1}+a_i,a_i)=max(d{p}_{i-1},0)+a_i$$
接下来，因为$a_i$的范围很小，所以考虑枚举最大值$mx(-30\le mx\le 30)$：
对于数组$a$，我们可以把大于$mx$的$a_i$，赋值为无穷小（作用：保证最大值为$mx$），然后对数组$a$，计算一次最大区间和，得到的结果为$sum$。最后，用$sum-mx$去更新答案。注意$sum=0$时要特殊处理。

代码

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
const int inf = 0x3f3f3f3f;

int n;
int a[maxn];
int t[maxn];
int dp[maxn];

int max_sum() {
	int ans = 0;
	for(int i = 1; i <= n; i++) {
		dp[i] = max(dp[i-1], 0) + t[i];
		ans = max(ans, dp[i]);
	}
	return ans;
}

int main() {
	scanf("%d", &n);
	for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
	
	int ans = 0;
	for(int mx = -30; mx <= 30; mx++) {
		for(int i = 1; i <= n; i++) t[i] = a[i];
		for(int i = 1; i <= n; i++) if(t[i] > mx) t[i] = -inf;
		int sum = max_sum();
		if(sum) ans = max(ans, sum-mx);
	}
	
	printf("%d\n", ans);
	return 0;
}