LeetCode 1931. Painting a Grid With Three Different Colors(DP)
题解,动态规划
因为m 最大只有5,所以我们可以枚举5个相邻的块,最多有多少种可能,
然后分析每个是否能与其他几个并排,然后就是简单的BP了
class Solution {
public:
int dp[50][5];
int bp[5];
int num[50];
long long int ans[50];
long long int ans2[50];
int pat[50][50];
int n,m;
int p;
const long long int mod = 1000000007;
int colorTheGrid(int m, int n) {
this->n=n;
this->m=m;
fun(0);
memset(num,0,sizeof(num));
for(int i=0;i<p;i++)
{
for(int j=0;j<p;j++)
{
if(j==i)
continue;
bool res = true;
for(int k=0;k<m;k++)
{
if(dp[i][k]==dp[j][k])
{
res=false;
break;
}
}
if(res)
{
pat[i][num[i]] = j;
num[i]++;
}
}
}
for(int i=0;i<p;i++)
{
ans[i]=1;
}
for(int i=1;i<n;i++)
{
for(int j=0;j<p;j++)
{
long long int res=0;
for(int k=0;k<num[j];k++)
{
res+=ans[pat[j][k]];
res%=mod;
}
ans2[j] = res;
}
for(int j=0;j<p;j++)
{
ans[j]=ans2[j];
}
}
long long int ans2=0;
for(int i=0;i<p;i++)
{
ans2+=ans[i];
ans2%=mod;
}
return ans2;
}
void fun(int pos)
{
if(pos==m)
{
for(int i=0;i<m;i++)
{
dp[p][i] = bp[i];
}
p++;
return;
}
for(int i=0;i<3;i++)
{
if(pos==0 || bp[pos-1] != i)
{
bp[pos]=i;
fun(pos+1);
}
}
}
};