【学无止境，每天一题】三倍子串

题目：
三倍子串
题目描述 第三届上海青少年算法竞赛T4
时间限制:1000ms 空间限制:256mb
给定一个十进制正整数n，请问可以从n中截取多少种不同的子串，使得子串构成的数字是3的倍
数。
例如：当n=1234，有且仅有3，12，123，234这四个子串是3的倍数。
输入格式
单个整数：表示输入的数字n
输出格式
单个整数：表示3的倍数的子串数量。
数据范围
对于20%的数据，1≤n≤10^9
对于50%的数据，1≤n≤10^100
对于70%的数据，1≤n≤10^1000
对于100%的数据，1≤n≤10^100000
样例数据
输入:
95764
输出:
6
说明:
子串6,9,57,576,957,9576是3的倍数
输入:
1111
输出:
2
说明:
有两个111都是3的倍数

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define ll long long
using namespace std;
const int maxn=1e6+10;
string s;
ll a[3];
int main()
{
	ios::sync_with_stdio(false);cin.tie(0);
	cin >> s;
	int n = s.size();
	int sum = 0;
	for(int i=0;i<n;i++){
		sum += s[i]-'0';
		sum %= 3;
		a[sum]++;
	}
	ll ans=a[1]*(a[1]-1)/2+a[2]*(a[2]-1)/2+a[0]*(a[0]-1)/2+a[0];
	cout<<ans;
	return 0;
}