链接:https://www.nowcoder.com/acm/contest/107/E
来源:牛客网
题目描述
众所周知,Xieldy最常用的口令是******。
为了改变这一现状,他random了一个01串,并从中截取了一段作为自己的口令。
他选择的口令满足以下条件:
1. 口令串表示的二进制数在十进制下可以被表示为3k(k>=0)。
2. 口令串可以有前导零。
现已经random出了01串,他想知道有多少种口令方案可以选择(不同的子段即为不同)。
为了改变这一现状,他random了一个01串,并从中截取了一段作为自己的口令。
他选择的口令满足以下条件:
1. 口令串表示的二进制数在十进制下可以被表示为3k(k>=0)。
2. 口令串可以有前导零。
现已经random出了01串,他想知道有多少种口令方案可以选择(不同的子段即为不同)。
输入描述:
若干组数据,每组数据仅一行01串s,表示random出来的的串,|s|<=1e6。
输出描述:
输出口令的方案数。
今天遇到一道dp题。
思路:
dp[i][0] 以i位置结尾的 值%3余0的情况数。
dp[i][1] 以i位置结尾的 值%3余1的情况数。
dp[i][2] 以i位置结尾的 值%3余2的情况数。
当到达i位置时,i位置为1或0,有四种情况
1 从前一个位置 值%3余0状态转移而来。
2 从前一个位置 值%3余1状态转移而来。
3 从前一个位置 值%3余2状态转移而来。
4 从i位置,新开一种状态为1或0的情况。
最后把位置 值%3余0 的状态的情况数加和,就是把所有以i位置结尾的能够被3整除的数加和。
具体转移方程在代码里。
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 1000010
char a[maxn];
long long dp[maxn][3];
int main()
{
long long sum;
int i,l;
while(scanf("%s",a+1)!=EOF)
{
memset(dp,0,sizeof(dp));
l=strlen(a+1);
sum=0;
for(i=1;i<=l;i++)
{
if(a[i]=='0')
{
dp[i][0]=1;//新开一种情况
dp[i][0]+=dp[i-1][0];
dp[i][1]+=dp[i-1][2];
dp[i][2]+=dp[i-1][1];
}
else
{
dp[i][1] = 1;//新开一种情况
dp[i][0]+=dp[i-1][1];
dp[i][1]+=dp[i-1][0];
dp[i][2]+=dp[i-1][2];
}
sum=sum+dp[i][0];
}
printf("%lld\n",sum);
}
return 0;
}