[AGC026D]Histogram Coloring

题目

传送门 to AtCoder

思路

考虑从上往下，每次涂好一行的颜色。为什么这样呢？我也不知道诶。

使用 $\mathrm{T}\mathrm{y}\mathrm{p}\mathrm{o}\mathrm{r}\mathrm{a}$ 前我的思路：

[海绵宝宝和派大星去抓水母了]，如果上一行和这一行是相同的，[外星人是紫色的]，需要分成两类，[水平线上绝海的瞳孔中绯色的第十一个业火的执行人的绀碧子弹沉默了十五分钟]，除了那些位置以外随便选……

用了 $\mathrm{T}\mathrm{y}\mathrm{p}\mathrm{o}\mathrm{r}\mathrm{a}$ 之后我的思路：

假如某一行中，有两个相邻的格子颜色相同。现在要涂它旁边的一行，不妨设是上面一行。显然这两个相邻的格子，上方是两个颜色相同的格子，且不与这两个格子颜色相同。那么从这里开始往两边拓展，不难发现：上一行的每个格子都必须是当前行对应的格子的颜色取反。

但如果当前行是红蓝相间的，那么上面一行也是红蓝相间，并且第一个是红或蓝均可。这也是容易发现的结论。

那么，考虑取出 $\min h$，两边的子区间互不干扰，可以独立递归计算。求出 $f(x,0)$ 和 $f(x,1)$，分别为 “总涂色方案” 和 “最后一行红蓝相间方案” 。考虑第 $\min h$ 行的状态：

• 如果 $\min h$ 行是红蓝相间，那么只需要子区间都是最后一行红蓝相间。方案数肯定是 $2\prod f(y,1)$，其中 $2$ 是枚举当前行的第一列是红还是蓝。
• 如果 $\min h$ 行不一定是红蓝相间，那么这一行在子区间覆盖到的范围都取相反颜色，即 $f(y,0)$，或者在它是红蓝相间时可以取相同颜色，即 $f(y,1)$，显然是 $2^w\prod [f(y,0)+f(y,1)]$，其中 $w$ 是没被子区间覆盖到的点数量，即 $\min h$ 的数量。

这是第 $\min h$ 行的状态。实际上我们要涂到第 $1$ 行。红蓝相间的，每一行都可以是颜色反转或不反转，乘 $2^{(\min h)-1}$，而不是红蓝相间的则不可。

不是红蓝相间的，需要用上面算出来的 “不一定红蓝相间” 减去 “红蓝相间” 的方案数。所以总结一下就是
$$\begin{gathered} f(x,1)=2^{(\min h)}\prod f(y,1) \\ f(x,0)=f(x,1)+2^w\prod [f(y,0)+f(y,1)]-2\prod f(y,1) \end{gathered}$$
子区间长度为 $1$ 应该视为红蓝相间，这样就让它可以颜色反转也可以不反转了。

显然是 $\mathcal{O}(n)$ 转移。只是需要快速幂，让它成为了 $\mathcal{O}(n\log h)$ 。当然，每次找区间 $\min h$，可以 $\mathcal{O}(n)$ 找，没必要那么麻烦，总复杂度 $\mathcal{O}(n^2)$ 也完全没问题嘛！

代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
# define drep(i,a,b) for(int i=(a); i>=(b); --i)
typedef long long int_;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
inline void writeint(int_ x){
	if(x > 9) writeint(x/10);
	putchar((x-x/10*10)^48);
}

const int Mod = 1e9+7;
inline int_ qkpow(int_ b,int q){
	int_ a = 1;
	for(; q; q>>=1,b=b*b%Mod)
		if(q&1) a = a*b%Mod;
	return a;
}

const int inv2 = (Mod+1)>>1;
const int MaxN = 105;
int dp[MaxN][2]; // 1:red and blue
int h[MaxN], cntNode;
int solve(int l,int r,int d=0){
	int o = ++ cntNode;
	int mn = *min_element(h+l,h+r+1);
	int lst = l, cnt = -1;
	dp[o][0] = dp[o][1] = 1;
	for(int i=l; i<=r+1; ++i){
		if(i != r+1 && h[i] != mn) continue;
		else ++ cnt; // exclude r+1
		if(i != lst){
			int x = solve(lst,i-1,mn);
			dp[o][1] = 1ll*dp[o][1]*dp[x][1]%Mod;
			dp[o][0] = 1ll*dp[o][0]*
				(dp[x][0]+dp[x][1])%Mod;
		}
		lst = i+1; // move on
	}
	dp[o][1] = (dp[o][1]<<1)%Mod; // two ways
	dp[o][0] = qkpow(2,cnt)*dp[o][0]%Mod;
	dp[o][0] = (dp[o][0]+Mod-dp[o][1])%Mod;
	dp[o][1] = qkpow(2,mn-d-1)*dp[o][1]%Mod;
	dp[o][0] = (dp[o][0]+dp[o][1])%Mod;
	return o;
}

int main(){
	int n = readint();
	rep(i,1,n) h[i] = readint();
	int rt = solve(1,n);
	printf("%d\n",dp[rt][0]);
	return 0;
}