Solution -「CF 1372E」Omkar and Last Floor

$\mathcal{Description}$

  Link.

  给定一个 $n \times m$ 的矩阵，每行被划分为若干段，你可以钦定每段中恰好一个位置为 $1$，其余位置为 $0$。设 $c_i$ 为第 $i$ 列 $1$ 的个数，最大化 $\sum_{i=1}^{m} c_i^2$。

  $n,m\le100$。

$\mathcal{Solution}$

  区间 DP，不过状态设计比较巧妙：令 $f(l,r)$ 表示确定在 $[l,r]$ 区间内的所有段的最大和。记 $c_{l,r,k}$ 表示在区间 $[l,r]$ 内，包含了第 $k$ 列的段的个数，转移：

$$f(l,r)=\max_{k\in[l,r]}\{f(l,k-1)+f(k+1,r)+c_{l,r,k}^2\}$$

  复杂度 $\mathcal O(n^4)$（求转移时在线求 $c_{l,r,k}$）。

$\mathcal{Code}$

/* Clearink */

#include <cstdio>

const int MAXN = 100;
int n, m, f[MAXN + 5][MAXN + 5], L[MAXN + 5][MAXN + 5], R[MAXN + 5][MAXN + 5];

inline void chkmax ( int& a, const int b ) { a < b ? a = b : 0; }

int main () {
	scanf ( "%d %d", &n, &m );
	for ( int i = 1, k; i <= n; ++ i ) {
		scanf ( "%d", &k );
		for ( int j = 1, l = 1, r; j <= k; ++ j, l = r + 1 ) {
			scanf ( "%*d %d", &r );
			for ( int h = l; h <= r; ++ h ) L[h][i] = l, R[h][i] = r;
		}
	}
	for ( int len = 1; len <= m; ++ len ) {
		for ( int l = 1, r; ( r = l + len - 1 ) <= m; ++ l ) {
			int& cur = f[l][r] = -1;
			for ( int k = l; k <= r; ++ k ) {
				int c = 0;
				for ( int h = 1; h <= n; ++ h ) c += l <= L[k][h] && R[k][h] <= r;
				chkmax ( cur, f[l][k - 1] + f[k + 1][r] + c * c );
			}
		}
	}
	printf ( "%d\n", f[1][m] );
	return 0;
}