动态规划--牛客多校number

动态规划--牛客多校number

题意：计算字符串中能被300整除的子串个数

思路：能被3整除的数，各位和能被3整除。算包含‘00’前一位数的前缀%3的个数，就是除0外能被300整除的个数

LL a[5];
int main()
{
    string s;cin >> s;
    int len = s.size();
    LL sum = 0, ans = 0;
    a[0] = 1;
    for(int i = 0;i < len;++i)
    {
        if(s[i] == '0' && s[i+1] == '0')
            ans += a[sum];
        sum = (sum+s[i]+'0')%3;
        a[sum]++;
    }
    for(int i = 0;i < len;++i)
        if(s[i] == '0') ans++;
    cout << ans << endl;
    return 0;
}