来源:牛客网
题目描述
众所周知,Xieldy最常用的口令是******。
为了改变这一现状,他random了一个01串,并从中截取了一段作为自己的口令。
他选择的口令满足以下条件:
1. 口令串表示的二进制数在十进制下可以被表示为3k(k>=0)。
2. 口令串可以有前导零。
现已经random出了01串,他想知道有多少种口令方案可以选择(不同的子段即为不同)。
为了改变这一现状,他random了一个01串,并从中截取了一段作为自己的口令。
他选择的口令满足以下条件:
1. 口令串表示的二进制数在十进制下可以被表示为3k(k>=0)。
2. 口令串可以有前导零。
现已经random出了01串,他想知道有多少种口令方案可以选择(不同的子段即为不同)。
输入描述:
若干组数据,每组数据仅一行01串s,表示random出来的的串,|s|<=1e6。
输出描述:
输出口令的方案数。
题解: 每次保存取模等于0,1,2的情况数,然后迭代更新。 代码: #include<bits/stdc++.h> using namespace std; #define ll long long const int maxn=1e6+5; char t[maxn]; int a[maxn][3]; int main() { while(~scanf("%s",&t)) { int L=strlen(t); ll a0=0,a1=0,a2=0,b0=0,b1=0,b2=0,la=0,ans=0; for(int i=1,j=L-1; j>=0; j--,i++) { if(t[j]=='0') { a0++; swap(a0,b0);swap(a1,b1);swap(a2,b2); ans=ans+a0+b0;la=a0+b0; } else { swap(a0,a1);swap(a0,a2); swap(b0,b2);swap(b0,b1);a1++; swap(a0,b0);swap(a1,b1);swap(a2,b2); la=a0+b0;ans+=la; } } printf("%lld\n",ans); } return 0; }