XTU oj 1289 3的倍数

Multiples of Three

题目描述

一个正整数n,要求去掉其中的一些数码得到一个新正整数n′(n′不能有前导0)，使得n′能被3整除。比如n=1234，那么我们有5种方法，分别得到{3,12,24,123,234}这5个数。 现在给你一个整数，请问有多少种方法符合要求。

输入

存在不超过2017个样例，每行一个整数n(1≤n≤105000)。

输出

每行输出一个整数，由于这个数可能很大，请将其对于 109+7 取模。

样例输入

100000
111000
100011
100023
123456
123456789012345678901234567890

样例输出

0
8
8
17
23
357566622

---

​
#include <stdio.h>//DP
#include <string.h>
char s[5003];
#define mod 1000000007
int f[5003][3];
int main(){
	int n,i,t;
	while(scanf("%s",s+1)!=EOF){
		n=strlen(s+1);
		//memset(f,0,sizeof(f));
		for(i=0;i<=n;i++)f[i][0]=f[i][1]=f[i][2]=0;//将要用的n位初始化为0
		for(i=1;i<=n;i++){
			t=s[i]-'0';//得到第i个数
			f[i][0]=f[i-1][0];
			f[i][1]=f[i-1][1];
			f[i][2]=f[i-1][2];//将前面%3的数量放到对应这一位，前缀和
			f[i][(0+t)%3]+=f[i-1][0];
			f[i][(1+t)%3]+=f[i-1][1];
			f[i][(2+t)%3]+=f[i-1][2];//前一个与t组合的新数的种数加到%3后余数的地方
			f[i][0]%=mod;
			f[i][1]%=mod;
			f[i][2]%=mod;
			if(t!=0){//0不能是数的开头，单个也不是3都倍数，不加
				f[i][t%3]++;//t作为单个数%3的地方加1
				f[i][t%3]%=mod;
			}
		}
		printf("%d\n",f[n][0]);//%3为0的是三的倍数的个数
	}
	return 0;
}

​