《ybtoj高效进阶》第五部分第五章例题3 涂抹果酱

题目大意

n*m的网格，有1~3任意填入，不能空，已填好第k行，问四联通不同的方案数%1e6的值。

思路

设$f_{i,j}$表示第i行填j号状态的方案数，有方程：
$f_{i,j}={\sum }_{1<=k<=总数且k不与j冲突}{f}_{i-1,k}$
code:

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
int n,m,pw[10],k,kk,tot,f[100001][1001],s[1001][10],tmp[10],velix[1001][1001]; 
int main()
{
	cin>>n>>m>>kk;
	pw[0]=1;
	for (int i=1;i<=m;i++) pw[i]=pw[i-1]*3;
	for (int i=1;i<=m;i++) cin>>tmp[i],k=k*3+tmp[i]%3;
	memset(tmp,0,sizeof(tmp));
	for (int i=0;i<pw[m];i++)
	{
		int lyf=i,l=0;
		bool wj=0;
		tmp[0]=-1;
		while (lyf)
		{
			tmp[++l]=lyf%3;
			lyf/=3;
		}
		for (int j=1;j<=m;j++) if (tmp[j]==tmp[j-1]) wj=1;
		if (wj==0)
		{
			if (i==k) k=-tot-1;
			++tot;
			for (int j=1;j<=m;j++) s[tot][j]=tmp[j];
		}
	}
	if (k>0)
	{
		cout<<0;
		return 0;
	}
	else k=-k;
	for (int i=1;i<=tot;i++)
	{
		velix[i][i]=0;
		for (int j=i+1;j<=tot;j++)
		{
			bool wj=1;
			for (int e=1;e<=m&&wj;e++) if (s[i][e]==s[j][e]) wj=0;
			velix[i][j]=velix[j][i]=wj;
		}
	}
	if (kk==1) f[1][k]=1;
	else
	{
		for (int i=1;i<=tot;i++) f[1][i]=1;
	}
	for (int i=2;i<=n;i++)
	{
		if (i==kk)
		{
			for (int j=1;j<=tot;j++) if (velix[k][j]) f[i][k]=(f[i-1][j]%1000000+f[i][k])%1000000;
		}
		else
		{
			for (int j=1;j<=tot;j++) for (int kkk=1;kkk<=tot;kkk++) if (velix[j][kkk]) f[i][j]=(f[i][j]+f[i-1][kkk]%1000000)%1000000;
		}
	}
	int ans=0;
	for (int i=1;i<=tot;i++) ans=(ans+f[n][i])%1000000;
	cout<<ans;
	return 0;
}