ZZNUOJ 2141: 2333 求子串是3的倍数的子串个数(前缀和+规律(On)防T)

题目链接：http://47.93.249.116/problem.php?id=2141

题意是给你给一个数字串，问该数字串有多少个连续子串能整除3

---

因为能被3整除的数字串的和一定是3的倍数，一开始的思路是记录前缀和，再对每一个前缀和遍历之后的前缀和用后面的前缀和减去前面的前缀和如果差值为3的倍数则ans++
但是时间复杂度为O(n^2)，而n<=1e6，肯定会T，这道题的正确思路是

思考：这个规律要记住，如果问数字串有多少个连续子串是k的倍数，该方法直接套用时间复杂度O(n)

#include <iostream>
#include <cstring>
#include <cstdio>
#include <math.h>
#include <string>
#include <time.h>
#include <algorithm>
using namespace std;

int main()
{
    char str[1000005];
    while(scanf("%s",str)!=EOF)
    {
        long long sum[3]={0};
        int s=0;
        int n=strlen(str);
        for(int i=0;i<n;i++)
        {
            s+=str[i]-'0';
            s%=3;
            sum[s]++;//统计0, 1, 2个数
        }

        long long ans;

        //求组合数
        ans=(sum[0]+1)*sum[0]/2+sum[1]*(sum[1]-1)/2+sum[2]*(sum[2]-1)/2;

        printf("%lld\n",ans);
    }

    return 0;
}