CF1426F Number of Subsequences(简单线性DP)

题意:

给出一个带问号和a b c的序列,询问在不同的子串中出现子序列abc个数的总和。

题解:

简单DP,在问号的地方要实时记录当前序列种类的数量。

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+100;
const int mod=1e9+7;
typedef long long ll;
ll a[maxn][3];
ll c[maxn][3];
//分别表示当第i位的数字取j时的前后缀a c 的数量
int n;
string s;
int main () {
    scanf("%d",&n);
    cin>>s;
    ll cnt=1;
    for (int i=1;i<=n;i++) {
        if (s[i-1]=='a') {
            a[i][0]=a[i-1][0]+a[i-1][1]+a[i-1][2]+cnt;a[i][0]%=mod;
        }
        else if (s[i-1]=='b') {
            a[i][1]=a[i-1][0]+a[i-1][1]+a[i-1][2];a[i][1]%=mod;
        }
        else if (s[i-1]=='c') {
            a[i][2]=a[i-1][0]+a[i-1][1]+a[i-1][2];a[i][2]%=mod;
        }
        else {
            a[i][0]=a[i-1][0]+a[i-1][1]+a[i-1][2]+cnt;a[i][0]%=mod;
            a[i][1]=a[i-1][0]+a[i-1][1]+a[i-1][2];a[i][1]%=mod;
            a[i][2]=a[i-1][0]+a[i-1][1]+a[i-1][2];a[i][2]%=mod;
            cnt*=3;
            cnt%=mod;
        }
    }
    cnt=1;
    for (int i=n;i>=1;i--) {
        if (s[i-1]=='a') {
            c[i][0]=c[i+1][0]+c[i+1][1]+c[i+1][2];c[i][0]%=mod;
        }
        else if (s[i-1]=='b') {
            c[i][1]=c[i+1][0]+c[i+1][1]+c[i+1][2];c[i][1]%=mod;
        }
        else if (s[i-1]=='c') {
            c[i][2]=c[i+1][0]+c[i+1][1]+c[i+1][2]+cnt;c[i][2]%=mod;
        }
        else {
            c[i][0]=c[i+1][0]+c[i+1][1]+c[i+1][2];c[i][0]%=mod;
            c[i][1]=c[i+1][0]+c[i+1][1]+c[i+1][2];c[i][1]%=mod;
            c[i][2]=c[i+1][0]+c[i+1][1]+c[i+1][2]+cnt;c[i][2]%=mod;
            cnt*=3;
            cnt%=mod;
        }
    }
    ll ans=0;
    for (int i=1;i<=n;i++) {
        if (s[i-1]=='b'||s[i-1]=='?') {
            ans+=(a[i-1][0]+a[i-1][1]+a[i-1][2])*(c[i+1][0]+c[i+1][1]+c[i+1][2])%mod;
            ans%=mod;
        }
    }
    printf("%lld\n",ans);
} 

 

posted @ 2020-09-29 20:37  zlc0405  阅读(240)  评论(0编辑  收藏  举报