Leecode 第 292 场周赛 6058. 统计打字方案数 DP

最新推荐文章于 2024-10-25 13:33:57 发布

自信的小螺丝钉

最新推荐文章于 2024-10-25 13:33:57 发布

阅读量148

点赞数

分类专栏： Leetcode Leetcode周赛文章标签：动态规划算法 leetcode c++ 数据结构

本文链接：https://blog.csdn.net/qq_45791939/article/details/124650478

版权

Leetcode 同时被 2 个专栏收录

337 篇文章 0 订阅

订阅专栏

Leetcode周赛

16 篇文章 0 订阅

订阅专栏

这篇博客主要讲解了一种利用动态规划和矩阵快速幂的方法来解决LeetCode 70题的变种问题6058.统计打字方案数。博主详细介绍了代码实现过程，包括两个动态规划数组f和g的初始化及更新，以及如何根据字符出现情况进行计算。最后，博主给出了完整的C++代码实现。

摘要由CSDN通过智能技术生成

原题链接：6058. 统计打字方案数

做这道题之前先了解一下这道题：Leecode 70. 爬楼梯 DP/矩阵快速幂
在这里插入图片描述

class Solution {
public:
    const int mod=1e9+7;
    int countTexts(string s) {
        int n=s.size();
        vector<int> f(n+1);
        vector<int> g(n+1);
        f[0]=1; g[0]=1;
        for(int i=1;i<=n;i++)
        {
            for( int j=1 ; j<=3 && j<=i; j++ ) f[i]=(f[i]+f[i-j])%mod;
            for( int j=1 ; j<=4 && j<=i; j++ ) g[i]=(g[i]+g[i-j])%mod;
        }
        int cnt=0;
        char last='0';
        long long res=1;
        for(int i=0;i<n;i++)
        {
            if(s[i]!=last)
            {
                if(last=='7' || last=='9') res=(res*g[cnt])%mod;
                else res=(res*f[cnt])%mod;
                cnt=0;last=s[i];
            }
            cnt++;
        }
        if(last=='7' || last=='9') res=(res*g[cnt])%mod;
        else res=(res*f[cnt])%mod;
        return res;
    }
};