题解 | #被3整除的子序列#

被3整除的子序列

https://ac.nowcoder.com/acm/problem/21302

如果一个数可以被33整除,那么这个数的每一位之和一定是33的倍数
证明:题解 | 3的倍数
定义:数组dp[i][j]dp[i][j]表示以ii为结尾组成的子序列,数位之和对33取余为jj的方案数
动态转换方程:
num=(s[i]0)%3num=(s[i]-'0')\%3 表示本位取余33
dp[i][(num+j)%3]+=k=0k1dp[k][j]dp[i][(num+j)\%3]+=\sum_{k=0}^{k-1} dp[k][j]
AC代码

#include <bits/stdc++.h>
using namespace std;
const long long mod = 1e9 + 7;
long long dp[100][5];
int main(void)
{
	string s;
	cin >> s;
	int n = s.size();
	for (int i = 0; i < n; i++)
	{
		int num = (s[i] - '0') % 3;
		dp[i][num]++;
		for (int j = 0; j < 3; j++)
		{
			for (int k = 0; k < i; k++)
			{
				dp[i][(num + j) % 3] = (dp[i][(num + j) % 3] + dp[k][j]) % mod;
			}
		}
	}
	long long ans = 0;
	for (int i = 0; i < n; i++)
		ans = (ans + dp[i][0]) % mod;
	cout << ans << endl;
}
全部评论

相关推荐

耀孝女:就是你排序挂了
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务