2021中国大学生程序设计竞赛（CCPC）- 网络选拔赛（重赛）1006 Nun Heh Heh Aaaaaaaaaaa

添加链接描述

#include"bits/stdc++.h"
#define pii pair<ll,ll>
#define ll long long
using namespace std;
const ll N=1e5+10;
const ll mod=998244353;

ll dp[N][20];
char str[N];
char strt[20]="0nunhehheh";
ll suffix[N];
ll powtwo[N];
ll sum[N][20];

void INIT() {
  powtwo[0]=1;
  for(ll i=1; i<N; i++)
    powtwo[i]=(powtwo[i-1]<<1)%mod;
}

void init(ll len) {
  for(ll i=1; i<=len+10; i++) {
    suffix[i]=0;
    for(ll j=1; j<=20; j++)
      sum[i][j]=dp[i][j]=0;
  }
}

void solve() {
  scanf("%s",str+1);
  ll len=strlen(str+1);
  init(len);
  suffix[len]=(str[len]=='a');
  for(ll i=len; i>=1; i--)
    suffix[i]=suffix[i+1]+(str[i]=='a');
  for(ll i=1; i<=len; i++) {
    sum[i][1]=dp[i][1]=(str[i]==strt[1]);
  }
  for(ll i=2; i<=len; i++) {
    for(ll j=1; j<=min(i,9LL); j++) {
      if(str[i]==strt[j]) {
        dp[i][j]=(dp[i][j]+sum[i-1][j-1])%mod;
      }
      sum[i][j]=(sum[i-1][j]+dp[i][j])%mod;
    }
  }
  ll ans=0;
  for(ll i=1; i<=len; i++) {
    ll cnta=suffix[i+1];
    ans=(ans+dp[i][9]*((powtwo[cnta]-1+mod)%mod)%mod)%mod;
  }
  printf("%lld\n",ans);
}

int main() {
  ll t;
  INIT();
  scanf("%lld",&t);
  while(t--)
    solve();
  return 0;
}