power oj 3296 iloveswust的真相 动态规划

题目传送门：3296: iloveswust的真相 -- Power OJ

 在题iloveswust之后，这是iloveswust这题背后的真相。为了判断大家在iloveswust题中提交的代码是否正确，就需要计算给定的字符串中有多少个iloveswust。

答案可能很大，请对998244353取模

input

一个字符串ss，(0<|s|<100000)

output

输出字符串中"iloveswust"的个数

样例输入：iloveswustiloveswust

样例输出：13

题目大意：

        就是给你一个字符串，判断它的字串中有多少个"iloveswust"，对998244353取模。

解题思路：

        很明显，这是个动态规划的题，当时打比赛的时候，被某F题给坑住了，导致这题最后十几分钟匆匆忙忙写出来，模数都搞错了就冲，最后一分钟才过，惊险AC，F题简直意难平啊😭。这题呢，很明显嘛，用dp[i][j]表示字符串前i字符串的字串中，以"iloveswust"第j个字母结尾的字串个数。那么对应的状态转移方程为：

,值得注意的是，如果是第一个字母，直接加1就行，当然也可以初始化dp[i][0]=1，还有就是s显然有两个，所以需要算两个状态转移，看代码。（代码没做任何优化，当时比赛写的，显得很长，运行起来也很慢，显然此题有很多优化的地方，比如用滚动数组，不用哈希，后面有优化版，优化版代码是copy的我队友的，嘿嘿，万能的群友）。

AC代码：

#pragma GCC optimize(2)
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <queue>
#include <map>
#include <unordered_map>
using namespace std;
#define LL unsigned long long
#define ll (now<<1|1)
#define rr ll+1
#define mid ((l+r)>>1)
#define gc getchar
#define p 998244353
const int N=1e5+10;
inline LL rd(){
  LL x=0; bool f=0; char c=gc();
  while(!isdigit(c)){if(c=='-')f=1;c=gc();}
  while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=gc();}
  return f?-x:x;
}
char ss[15]={"iloveswust"};
unordered_map<char, int> buk;
char s[N];
LL dp[N][15];
int main()
{
    scanf("%s",s);
    for(int i=0;i<10;i++) buk[ss[i]]=i+1;
    for (int i=0; i<strlen(s); i++) {
        for (int j=1; j<=10; j++) {
            dp[i][j]=dp[i-1][j]%p;
        }
        if(s[i]=='i')
        {
            dp[i][1]++;
            dp[i][1]%=p;
        }else if (s[i]=='s')
        {
            dp[i][6]+=dp[i-1][5];
            dp[i][6]%=p;
            dp[i][9]+=dp[i-1][8];
            dp[i][9]%=p;
        }else{
            dp[i][buk[s[i]]]+=dp[i-1][buk[s[i]]-1];
            dp[i][buk[s[i]]]%=p;
        }
    }
    printf("%lld\n",dp[strlen(s)-1][10]%p);
    return 0;
}

优化版：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
#define ll long long
const ll mod=998244353;
char ss[15]="0iloveswust";
char a[100010];
ll dp[15];
int main()
{
    scanf("%s",a);
    for(int i=0;i<strlen(a);i++)
    {
        char t=a[i];
        dp[0]=1;
        for(int j=1;j<=10;j++)
        {
            if(t==ss[j]){
                dp[j]=(dp[j]+dp[j-1])%mod;
            }
        }
    }
    printf("%lld\n",dp[10]);
    return 0;
}