【枚举区间思想+DP】子串的子序列

港控/mmm° 2024-03-16 20:02 61阅读 0赞

[F-子串的子序列\_牛客小白月赛62 (nowcoder.com)][F-_62 _nowcoder.com]

题意：

![45d4a87408594a30ac7cc222f3ce00e6.png][]

![3c9bb7b619bb44e288c936db0b34db45.png][]

思路：

复盘一下应该有的思路：

首先n^2枚举肯定超时，我们枚举的是一个区间

**枚举区间有一些trick：**

**1.枚举其中一个右（左）端点，O(1)或O(logn)计算满足条件的左（右）端点个数 ，可以组合数学 or DP 计算**

**2.单独计算每个元素的贡献，O(1)或O(logn)计算合法区间的左端点和右端点匹配个数，这种一般是乘法原理**

在这里我们用到的是第一种思想

我们去枚举右端点r，然后计算满足条件的左端点的个数

我就想到这里，甚至连DP都没想到

这里的DP并不是传统意义的子序列DP，也就是说，并不是枚举两个指针然后转移(LIS)，也不是统计前缀满足某个性质的子序列的最值（接龙数列 or 绝世好题），这里用的是状态机DP

设dp\[i\]\[0/1\]\[0/1\]表示，前i个字符，a的数量的奇偶性是0/1，ac的数量的奇偶性是0/1的**子串个数**

为什么是这样设计的，其实一开始想的应该只是ac的数量的奇偶性，但是发现这样不能算贡献，考虑多一个字符c，贡献还与前缀a的数量的奇偶性有关，所以应该还有加上一维

**这里告诉我们，在考虑计算贡献的DP时，考虑多一格，然后思考如何计算贡献**

然后转移的时候注意这是子串数量，所以应该从i-1转移

（看代码）为什么一开始要++，因为考虑新加的字符自成一个区间

还有一个坑点：

然后统计答案的时候会把ac子序列个数为0的区间个数算进去，因此我们需要减去这些区间

这个怎么做？同样也是那个枚举区间的trick，我们去枚举左区间，右侧ac的c左边那格-i+1就是区间个数了，统计一下即可

Code：

#include <bits/stdc++.h>
    
    #define int long long
    
    using namespace std;
    
    using i64 = long long;
    
    const int mxn=5e5+10;
    const int mxe=5e5+10;
    const int mod=1e9+7;
    
    string s;
    
    int N;
    int pre_a[mxn],pre_c[mxn];
    int dp[mxn][2][2];
    
    int no_ac(){
        int res=0;
        for(int i=1;i<=N;i++){
            if(s[i]=='a'){
                pre_a[i]=i;
                pre_c[i]=pre_c[i-1];
            }else if(s[i]=='c'){
                pre_c[i]=i;
                pre_a[i]=pre_a[i-1];
            }else{
                pre_a[i]=pre_a[i-1];
                pre_c[i]=pre_c[i-1];
            }
        }
        for(int r=1;r<=N;r++) res+=r-(pre_a[pre_c[r]]+1)+1;
        return res;
    }
    void solve(){
        cin>>s;
        N=s.size();
        s=" "+s;
        int ans=0;
        for(int i=1;i<=N;i++){
            if(s[i]=='a'){
                dp[i][1][0]++;
                dp[i][1][1]+=dp[i-1][0][1];
                dp[i][1][0]+=dp[i-1][0][0];
                dp[i][0][1]+=dp[i-1][1][1];
                dp[i][0][0]+=dp[i-1][1][0];
            }else if(s[i]=='c'){
                dp[i][0][0]++;
                dp[i][1][1]+=dp[i-1][1][0];
                dp[i][1][0]+=dp[i-1][1][1];
                dp[i][0][1]+=dp[i-1][0][1];
                dp[i][0][0]+=dp[i-1][0][0];
            }else{
                dp[i][0][0]++;
                for(int j=0;j<2;j++){
                    for(int k=0;k<2;k++){
                        dp[i][j][k]+=dp[i-1][j][k];
                    }
                }
            }
            ans+=dp[i][1][0]+dp[i][0][0];
        }
        cout<<ans-no_ac()<<'\n';
    }
    signed main(){
        ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
        int __=1;//cin>>__;
        while(__--)solve();return 0;
    }

[F-_62 _nowcoder.com]: https://ac.nowcoder.com/acm/contest/47266/F
[45d4a87408594a30ac7cc222f3ce00e6.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/15/60496944da8a40108d126b0f422dce8c.png
[3c9bb7b619bb44e288c936db0b34db45.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/15/e851d48a86d74e688ff7ce2a0777d00f.png