CodeForces - 17C Balance【DP】

最新推荐文章于 2026-03-26 16:15:34 发布

原创最新推荐文章于 2026-03-26 16:15:34 发布 · 2w 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

codeforces 同时被 2 个专栏收录

145 篇文章

订阅专栏

23 篇文章

订阅专栏

本文解析了一段C++代码，用于计算字符串s中满足特定条件的子串组合数量，通过动态规划方法实现，核心在于维护三个字符类别计数。

题目链接：https://codeforces.com/contest/17/problem/C

#include <iostream>
#include <cmath>
using namespace std;
static const int MOD=51123987;
static const int MAXN=150+10;
char s[MAXN];
int p[MAXN][3];
int dp[MAXN][55][55][55];
int n,res;
int main()
{
    scanf("%d%s",&n,s+1);
    p[n+1][0]=p[n+1][1]=p[n+1][2]=n+1;
    for(int i=n;i>=1;i--)
    {
        p[i][0]=p[i+1][0];
        p[i][1]=p[i+1][1];
        p[i][2]=p[i+1][2];
        if(s[i]=='a') p[i][0]=i;
        if(s[i]=='b') p[i][1]=i;
        if(s[i]=='c') p[i][2]=i;
    }
    int k=(n+2)/3;
    dp[1][0][0][0]=1;
    for(int i=1;i<=n;i++)
        for(int a=0;a<=k;a++)
            for(int b=0;b<=k;b++)
                for(int c=0;c<=k;c++)
                {
                    if(a+b+c==n && abs(a-b)<2 && abs(a-c)<2 && abs(b-c)<2)
                    {
                        res=(res+dp[i][a][b][c])%MOD;
                        continue;
                    }
                    dp[p[i][0]][a+1][b][c]=(dp[p[i][0]][a+1][b][c]+dp[i][a][b][c])%MOD;
                    dp[p[i][1]][a][b+1][c]=(dp[p[i][1]][a][b+1][c]+dp[i][a][b][c])%MOD;
                    dp[p[i][2]][a][b][c+1]=(dp[p[i][2]][a][b][c+1]+dp[i][a][b][c])%MOD;
                }
    printf("%d\n",res);
    return 0;
}