牛客国庆比赛 K number(顺序思维)(dp)

最新推荐文章于 2024-11-02 14:05:25 发布

逐梦起航-带梦飞翔

最新推荐文章于 2024-11-02 14:05:25 发布

阅读量103

点赞数 1

文章标签： c++ 算法动态规划

本文链接：https://blog.csdn.net/A_Bright_CH/article/details/127191511

版权

选择一个子串，使其为300的倍数，特别地0,00,000以及不同位置的相同子串，都算。

f[i][0/1/2]表示以i结尾的，模3为0/1/2的子串个数。转移很好写，注意区分“转移”和“新增”。

遇00就加，注意：0和00的情况是不会被这种方法统计进去的，要特别计算。

一定要有数学归纳法的思想，这对于解所有编程问题都很重要。

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;

int f[N][5];
char s[N];

int main()
{
	scanf("%s", s + 1);
	int len = strlen(s + 1);
	long long ans = 0;
	for(int i = 1; i <= len; i++)
	{
		ans += s[i] == '0'; // 计算0
		f[i][(0 + s[i] - '0') % 3] = f[i - 1][0];
		f[i][(1 + s[i] - '0') % 3] = f[i - 1][1];
		f[i][(2 + s[i] - '0') % 3] = f[i - 1][2];
		f[i][(s[i] - '0') % 3]++;
	}
	for(int i = len - 1; i >= 1; i--)
	{
		if(s[i] == '0' && s[i + 1] == '0') ans += f[i - 1][0] + 1; // +1 是为了计算00
	}
	cout << ans << endl;
	return 0;
}