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);
            }
        }
    }
};