2021CCPC重现赛Nun Heh Heh Aaaaaaaaaaa

 题目大意：给定了一个字符串，问你在给定的字符串中可以找到多少个nunhehheh+个数>=1的a这样的字符串，看这个题之前先来做一道简单一点的题I-I love you_牛客小白月赛21 (nowcoder.com)

直接看代码

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int N =684694;
char s[N];
int mod=20010905;
long long dp[10];
int main()
{
	scanf("%s",s+1);
	int len=strlen(s+1);
	for(int i=1;i<=len;i++)
	{
		if(s[i]=='i'||s[i]=='I')
		dp[1]=(dp[1]+1)%mod;
		if(s[i]=='l'||s[i]=='L')
		dp[2]=(dp[1]+dp[2])%mod;
		if(s[i]=='o'||s[i]=='O')
		dp[3]=(dp[2]+dp[3])%mod;
		if(s[i]=='v'||s[i]=='V')
		dp[4]=(dp[3]+dp[4])%mod;
		if(s[i]=='e'||s[i]=='E')
		dp[5]=(dp[4]+dp[5])%mod;
		if(s[i]=='y'||s[i]=='Y')
		dp[6]=(dp[5]+dp[6])%mod;
		if(s[i]=='o'||s[i]=='O')
		dp[7]=(dp[6]+dp[7])%mod;
		if(s[i]=='u'||s[i]=='U')
		dp[8]=(dp[7]+dp[8])%mod;
	}
	cout<<dp[8]<<endl;
	return 0;
}

 这个题思路比较简单就不做详细的说明了，看代码就能看明白，我们还是回到原来的题目，可以发现我们只要能找到nunhehheh这个字符串，看这时该字符串的后面有几个a，就有2^(a的个数）-1个满足要求的字符串，所以到了这一步，我们就可以看出其实真正的难点就在于怎么求出nunhehheh字符串的个数，这就用到了我们上面代码中涉及到的知识了，比较容易理解的是以下代码

 for(int i=1;i<=s.size();i++)
        {
            dp[1][i]=(dp[1][i-1]+(s[i]=='n'))%mod;
            dp[2][i]=(dp[2][i-1]+dp[1][i-1]*(s[i]=='u'))%mod;
            dp[3][i]=(dp[3][i-1]+dp[2][i-1]*(s[i]=='n'))%mod;
            dp[4][i]=(dp[4][i-1]+dp[3][i-1]*(s[i]=='h'))%mod;
            dp[5][i]=(dp[5][i-1]+dp[4][i-1]*(s[i]=='e'))%mod;
            dp[6][i]=(dp[6][i-1]+dp[5][i-1]*(s[i]=='h'))%mod;
            dp[7][i]=(dp[7][i-1]+dp[6][i-1]*(s[i]=='h'))%mod;
            dp[8][i]=(dp[8][i-1]+dp[7][i-1]*(s[i]=='e'))%mod;
            dp[9][i]=(dp[8][i-1]*(s[i]=='h'))%mod;
            if(dp[9][i])res=res%mod+(dp[9][i]*qmi(a[i+1]))%mod;
            res%=mod;
        }

然后我们可以用滚动数组进行优化

for(int i=1;i<=len;i++)
		{
			if(s[i]=='n')
			{
			dp[3]=(dp[3]+dp[2])%mod;
			dp[1]=(dp[1]+1)%mod;
			}
			if(s[i]=='u')
			dp[2]=(dp[2]+dp[1])%mod;
			if(s[i]=='h')
			{
				int tmp=dp[9];
				dp[9]=(dp[8]+dp[9])%mod;
				dp[7]=(dp[7]+dp[6])%mod;
				dp[6]=(dp[6]+dp[5])%mod;
				dp[4]=(dp[4]+dp[3])%mod;
				if(dp[9])
				ans=(ans%mod+(dp[9]-tmp+mod)%mod*ksm(2,numa[i+1])%mod)%mod;
				ans=ans%mod;
			}
			if(s[i]=='e')
			{
			    dp[8]=(dp[7]+dp[8])%mod;
				dp[5]=(dp[4]+dp[5])%mod;
			}
		}

这里需要注意2点：

（1）、当字符串中有重复的字母时，一旦遇到该字母，要先更新靠后位置的字母，再更新前面的位置，否则会导致后面位置代表的dp数组的数值偏大，具体原因在于数组从2维优化到了1维。

（2）、计算到最后一个字母时，要事先把原来dp数组的值记录下来，最后答案加的时候要用当前的dp值减去之前的dp值，因为我们在答案中要加的是新凑出来的nunhehheh的个数，否则之前的答案会被重复计算，必过不了样例。

上完整代码

//nunhehheh
#include <iostream>
#include <cstring>
#include <cstring>
using namespace std;
const int N =1e5+10;
long long dp[11],numa[N];
int mod=998244353;
char s[N];
long long ksm(long long a,long long b)
{
	long long ans=1;
	while(b)
	{
		if(b&1)
		ans=(ans*a)%mod;
		b=b>>1;
		a=(a*a)%mod; 
	 } 
	 return ((ans-1)%mod+mod)%mod;
 } 
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		scanf("%s",s+1);
		memset(dp,0,sizeof dp);
		memset(numa,0,sizeof numa);
		int len=strlen(s+1);
		long long ans=0;
		for(int i=len;i>=1;i--)
		{
			if(s[i]=='a')
			numa[i]=numa[i+1]+1;
			else
			numa[i]=numa[i+1];
		}
		for(int i=1;i<=len;i++)
		{
			if(s[i]=='n')
			{
			dp[3]=(dp[3]+dp[2])%mod;
			dp[1]=(dp[1]+1)%mod;
			}
			if(s[i]=='u')
			dp[2]=(dp[2]+dp[1])%mod;
			if(s[i]=='h')
			{
				int tmp=dp[9];
				dp[9]=(dp[8]+dp[9])%mod;
				dp[7]=(dp[7]+dp[6])%mod;
				dp[6]=(dp[6]+dp[5])%mod;
				dp[4]=(dp[4]+dp[3])%mod;
				if(dp[9])
				ans=(ans%mod+(dp[9]-tmp+mod)%mod*ksm(2,numa[i+1])%mod)%mod;
				ans=ans%mod;
			}
			if(s[i]=='e')
			{
			    dp[8]=(dp[7]+dp[8])%mod;
				dp[5]=(dp[4]+dp[5])%mod;
			}
		}
		cout<<ans%mod<<endl;
	}
	return 0;
 }