CF111C Petya and Spiders

Fire_Raku發表於2024-06-07

CF111C Petya and Spiders

狀壓 dp

觀察到資料範圍 \(n\cdot m\le40\),所以最短邊 \(\le 6\)。然後題目的要求就是將網格用四連通塊全覆蓋的最少數量。

然後考慮一行一行放蜘蛛,那麼假設當前考慮完了前 \(i\) 行(前 \(i-1\) 行全部鋪滿),那麼第 \(i\) 行每個位置有三種情況:

  1. 建立避難所
  2. 左或右或上有避難所可以蹭
  3. 沒有找到避難所

那麼用三進位制狀態 \(s\) 記錄當前行狀態,設 \(f_{i,s}\) 表示考慮完前 \(i\) 行,第 \(i\) 行的狀態為 \(s\) 的方案數。

轉移列舉下一行的狀態(這裡只列舉放不放避難所),根據第 \(i\) 行填充第 \(i+1\)\(2\) 情況的位置。需要判斷合法,因為轉移到 \(f_{i+1,t}\) 要保證第 \(i\) 行最終被填滿。

注意初始化 \(f_{0,lim}=0\),複雜度 \(O(m6^n)\)

#include <bits/stdc++.h>
#define pii std::pair<int, int>
#define fi first
#define se second
#define pb push_back

using i64 = long long;
using ull = unsigned long long;
const i64 iinf = 0x3f3f3f3f, linf = 0x3f3f3f3f3f3f3f3f;
const int N = 3 * 3 * 3 * 3 * 3 * 3 * 3, M = 41;
int n, m, lim, ans;
int f[M][N];
int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
	std::cin >> n >> m;
	if(n > m) std::swap(n, m);

	lim = 1;
	for(int i = 0; i < n; i++) lim *= 3;
	for(int i = 0; i <= m; i++) {
		for(int s = 0; s < lim; s++) f[i][s] = -iinf;
	}
	f[0][lim - 1] = 0; //注意初始化

	for(int i = 1; i <= m; i++) {
		for(int s = 0; s < lim; s++) { //上一行
			for(int t = 0; t < (1 << n); t++) { //當前行
				int T = 0, cnt = 0;
				for(int j = 0, base = 1; j < n; j++, base *= 3) {
					if(!((t >> j) & 1)) continue;
					if(j + 1 < n && !((t >> 1 >> j) & 1)) T += base * 2;
					else if(j && !((t << 1 >> j) & 1)) T += base * 2;
					else if(!((s / base) % 3)) T += base * 2;
					else T += base;
					if((s / base) % 3 == 1) cnt = -iinf;
					else cnt++; 
				}
				f[i][T] = std::max(f[i][T], f[i - 1][s] + cnt);
			} 
		}
	}

	for(int s = 0; s < lim; s++) {
		bool flg = 1;
		for(int i = 0, base = 1; i < n; i++, base *= 3) {
			if((s / base) % 3 == 1) flg = 0;
		}
		if(flg) ans = std::max(ans, f[m][s]);
	}
	std::cout << ans << "\n";

	return 0;
}