2020.05.24日常总结——又一道特别好的 dp 题

$$\text{洛谷P1121 环状最大两段子段和}$$

$\text{【题意】：}$

• 给你一个首尾相接的数组 $a$（即环），选出其中连续 不重叠 且 非空 的 两段 使得这两段和最大。
• $2\le n\le 2\times 10^5,-1\times 10^4\le a_i\le 1\times 10^4(1\le i\le n)$。
• 注意两段子段可以相连（相当于一段）。

$\text{【思路】：}$

对于环上的 dp，我们一般考虑破环为链法，即把环转化为链。

破环为链法中，又有几种比较常见的处理方法，这里列举两种：

1. 把数组复制两遍，即把数组 $a_{1..n}$ 变成数组 $a_{1..2n}$，其中 $a_i=a_{i-n}(n+1⩽i⩽2n)$。这是最常用的一种破环为链法。
2. 不复制，直接用其它方法把环转化为链。

在本题中，我们发现虽然第一种很常用，但是我们仍然无法用它来解决问题。所以我们考虑第二种方法。

我们发现用第二种方法时，我们无法把所有的情况统一在一起讨论，这个时候，我们就需要用到另一种数学（$\text{OI}$ 也适用）的方法了，那就是：分类讨论。

• 第一种情况：两段子段都不跨越首尾，即类似于这样的：

  $$0001111100002220000$$

  $0$ 表示该位不选，$1$ 表示选且属于第一子段，$2$ 表示选且属于第二子段。

  我们记 $f_i$ 表示 $a_{1..i}$ 的最大子段和，$g_i$ 表示 $a_{i..n}$ 的最大子段和，则答案为：

  $$\underset{i=1}{\overset{n}{\max }}{f}_i+g_{i+1}$$

• 第二种情况：其中一个子段跨越了首尾，即类似于此：

  $$2220001111110000022$$

  命名方法同上。

  直接求很难，我们就考虑反着做，毕竟正难则反。

  我们发现 $0$ 所占据的正好就是两个子段，于是，我们即 $f_i$ 表示 $a_{1..i}$ 的最小子段和，$g_i$ 表示 $a_{i+1..n}$ 的最小子段和。则答案为：

  $$\sum _{i=1}^n{a}_i-\underset{i=1}{\overset{n}{\min }}{f}_i+g_{i+1}$$

  注意当 $f_i+g_{i+1}$ 正好就是整个区间的情况，需要特判，因为这样相当于选得数为空。

$\text{【代码】：}$

const int N=2e5+100;
#define ll long long
int n;ll ans,res,sum;
ll f[N],g[N],h[N],a[N];
const ll inf=0x3f3f3f3f3f;
inline ll calc_min_sum(){
	h[0]=h[n+1]=f[0]=g[n+1]=inf;
	for(register int i=1;i<=n;i++){
		h[i]=min(h[i-1]+a[i],a[i]);
		f[i]=min(f[i-1],h[i]);
	}
	for(register int i=n;i>=1;i--){
		h[i]=min(h[i+1]+a[i],a[i]);
		g[i]=min(g[i+1],h[i]);
	}
	register ll ret=inf;
	for(int i=1;i<=n;i++)
		ret=min(ret,f[i]+g[i+1]);
	return ret;
}
inline ll calc_max_sum(){
	h[0]=h[n+1]=f[0]=g[n+1]=-inf;
	for(register int i=1;i<=n;i++){
		h[i]=max(h[i-1]+a[i],a[i]);
		f[i]=max(f[i-1],h[i]);
	}
	for(register int i=n;i>=1;i--){
		h[i]=max(h[i+1]+a[i],a[i]);
		g[i]=max(g[i+1],h[i]);
	}
	register ll ret=-inf;
	for(int i=1;i<=n;i++)
		ret=max(ret,f[i]+g[i+1]);
	return ret;
}
int main(){
	freopen("t1.in","r",stdin);
	n=read();//我们的程序开始了 
	for(int i=1;i<=n;i++)
		sum+=(a[i]=read());
	ans=calc_max_sum();//求最大 
	res=calc_min_sum();//求最小 
	if (res==sum){//全部数子为负 
		res=ans=-inf;//最大;次大 
		for(int i=1;i<=n;i++){
			if (a[i]>res){
				ans=res;
				res=a[i];
			}
			else ans=max(ans,a[i]);
		}
		printf("%lld",ans+res);
	}
	else{//数字有正有负(普通情况) 
		printf("%lld",max(ans,sum-res));
	}
	return 0;
}