Educational Codeforces Round 107 (Rated for Div. 2) E. Colorings and Dominoes(思维)

LINK

考虑一个确定的染色方案中,每个$o$非红即蓝

那么就有一个非常好的性质,就是一个格子如果为红色不可能去和上下的格子结合,一个格子如果是蓝色那么不可能和左右结合

所以可以把行和列拆开来,分别计算每一行的最大贡献和每一列的最大贡献

定义$f[i]$表示连续$i$个$o$的所有方案造成的总贡献

Ⅰ.当第$i$个位置为蓝色时,贡献是$f[i-1]$

Ⅱ.当第$i$个位置是红色时

①.当第$i-1$个位置也是红色时,让$i,i-1$结合结果不会变差

此时$[1,i-2]$的总贡献为$f[i-2]$,而$i,i-1$形成的贡献取决于$[1,i-2]$的方案数也就是$2^{i-2}$

②.当第$i-1$个位置为蓝色时,答案为$f[i-2]$

所以总的转移方程为

$f[i]=f[i-1]+f[i-2]+(f[i-2]+2^{i-2})$

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 3e5+10;
const int mod = 998244353;
int n,m,sum,f[maxn],base[maxn];
vector<string>a;
int quick(int x,int n)
{
	int ans = 1;
	for( ; n ; n>>1,x=x*x%mod )
		if( n&1 )	ans = ans*x%mod;
	return ans;
}
signed main()
{
	cin >> n >> m;
	a.push_back( "" );//push空串只是为了偏移一下,因为下面使用的下标都是从1起的
	for(int i=1;i<=n;i++)
	{
		string s; cin >> s; s = " "+s; a.push_back( s );
		for(int j=1;j<=m;j++)
			sum += ( a[i][j]=='o' );
	}
	base[0] = 1; base[1] = 2;
	for(int i=2;i<=sum;i++)
	{
		base[i] = base[i-1]*2%mod;
		f[i] = ( f[i-1]+f[i-2]+f[i-2]+base[i-2] )%mod;
	}
	int ans = 0;
	for(int i=1;i<=n;i++)
	{
		int cnt = 0;
		for(int j=1;j<=m;j++)
		{
			if( a[i][j]=='o' )	cnt++;
			else	ans = ( ans+f[cnt]*base[sum-cnt]%mod )%mod, cnt = 0;
		}
		ans = ( ans+f[cnt]*base[sum-cnt]%mod )%mod, cnt = 0;
	}
	for(int j=1;j<=m;j++)
	{
		int cnt = 0;
		for(int i=1;i<=n;i++)
		{
			if( a[i][j]=='o' )	cnt++;
			else	ans = ( ans+f[cnt]*base[sum-cnt]%mod )%mod, cnt = 0;
		}
		ans = ( ans+f[cnt]*base[sum-cnt]%mod )%mod, cnt = 0;
	}
	cout << ans;
}