C. The Fair Nut and String 递推分段形dp

C. The Fair Nut and String 递推分段形dp

题意

给出一个字符串选择一个序列\({p_1,p_2...p_k}\)使得
对于任意一个\(p_i\) , \(s[p_i]==a\)
对于任意一个\(p_{i}<j<p_{i+1}\)来说 \({\exists}s[p_j]==b\)

思路

所以我们可以得知 我们需要选择一系列a 使得a和a之间只能是b
那么我们就可以对a进行分段处理
例如aaaabaaaa 右面与前面组合 只能选择后面一大串a的前缀和前面一大串a的后缀组合才合法
也就是一共有 num[1]*num[2]中组合 其中num[i] 表示第i段有多少个a
而如果多段组合的话 第三段既可以和第二段组合 也可以和第二段+第一段组合
第二段+第一段就是 在算第三段之前的合法选择数，而第三段和第二段 同第一段和第二段的算法相同
所以递推求解即可 而分段的时候如果字母不是b 直接忽略就行了（因为只是\({\exists}s[p_j]==b）\)所以不会干扰a和a的组合可以忽略，相当于归到一段a中但是这一段a的数量不变
ps:记得开long long

#include<bits/stdc++.h>
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;i++)
#define MS(arr,arr_value) memset(arr,arr_value,sizeof(arr)) 
#define F first 
#define S second
#define pii pair<int ,int >
#define mkp make_pair
#define pb push_back
#define arr(zzz) array<ll,zzz>
using namespace std;
typedef long long ll;
const int maxn=3e5+5;
int n,m,y;
const int mod=1e9+7;
char s[maxn];
int main(){
    
    scanf("%s",s+1);
    int len=strlen(s+1);
    long long ans=0;
    for(int i=1;i<=len;i++){
        if(s[i]=='a')ans++;
    }
    ll flag=0;
    ll tmp=0;
    int cnt=0;
    ll x=0;
    ll fixsum=0;
    ll lastsum=0;
    int ok=0;
    for(int i=1;i<=len;i++){
        if(s[i]=='a'){
            flag++;
        }
        else if(s[i]=='b'){
            if(flag){
                cnt++;
            tmp+=flag*fixsum;
            tmp%=mod;
            //ll zzzz=lastsum;
            //lastsum=tmp;
            tmp+=lastsum*flag;
            tmp%=mod;
            lastsum=tmp;
            fixsum+=flag;
            fixsum%=mod;
            flag=0;
            }
        }
        if(s[i]=='b')ok=0;
        else if(s[i]=='a')ok=1;
    }
    if(ok){
                cnt++;
            tmp+=flag*fixsum;
            tmp%=mod;
            //ll zzzz=lastsum;
            //lastsum=tmp;
            tmp+=lastsum*flag;
            tmp%=mod;
            lastsum=tmp;
            fixsum+=flag;
            fixsum%=mod;
            flag=0;
    }
    if(cnt>=2)ans+=tmp;
    ans%=mod;
    cout<<ans<<endl;
    return 0;
}