前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 1931. Painting a Grid With Three Different Colors(DP)

LeetCode 1931. Painting a Grid With Three Different Colors(DP)

作者头像
ShenduCC
发布2021-10-09 09:56:56
2530
发布2021-10-09 09:56:56
举报
文章被收录于专栏:算法修养

题目

题解,动态规划 因为m 最大只有5,所以我们可以枚举5个相邻的块,最多有多少种可能, 然后分析每个是否能与其他几个并排,然后就是简单的BP了

代码语言:javascript
复制
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);
            }
        }
    }
};
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021-09-30 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档