题目链接:https://nanti.jisuanke.com/t/42578
思路:
要保证i>=j 可以从最后一位开始记录方案数
设f[k][2][2] 第一维表示记录到第k位,第二维表示i为0或1 ,第三维表示j为0或1。
要注意的一点是,需要额外记录f[i][0][1] 的方案数,虽然在第k位下不合法,但如果之后记录到i=1,j=0的情况时候,就会变成合法状态。
代码如下:
#include <bits/stdc++.h>
using namespace std;
#define rep(x, y, z) for (int x = y; x <= z; x++)
#define dec(x, y, z) for (int x = y; x >= z; x--)
const int N = 1e5+10, mod = 1e9 + 7;
const double expp = 1e-10;
typedef long long ll;
//------------------------'
char s[N];
ll f[N][2][2];
/*
若n的这个位置为1 ,则i可以为1 也可以为0
若n的这个位置为0 ,则i必须为0
若i为1 ,则j必须为0
若i为0 ,则j可以为0 ,也可以为1(保证i>=j)
*/
int main() {
int T;
cin>>T;
while(T--) {
cin>>s+1;
int len = strlen(s+1);
if(s[len] == '1') {
f[len][1][0] = 1;
f[len][0][0] = 1;
f[len][0][1] = 1;
} else {
f[len][0][0] = 1;
f[len][0][1] = 1;
}
ll ans = 0;
ans += f[len][1][0] + f[len][0][0];
for(int i=len-1; i>=1; i--) {
if(s[i] == '1') {
f[i][1][0] = (f[i][1][0] + f[i+1][1][0] + f[i+1][0][0] + f[i+1][0][1])%mod;
f[i][0][0] = (f[i][0][0] + f[i+1][1][0] + f[i+1][0][0] + f[i+1][0][1])%mod;
f[i][0][1] = (f[i][0][1] + f[i+1][1][0] + f[i+1][0][0] + f[i+1][0][1])%mod;
} else {
f[i][0][0] = (f[i][0][0] + f[i+1][1][0] + f[i+1][0][0] + f[i+1][0][1])%mod;
f[i][0][1] = (f[i][0][1] + f[i+1][1][0] + f[i+1][0][0] + f[i+1][0][1])%mod;
}
ans = (ans+f[i][1][0])%mod;
}
cout<<ans%mod<<endl;
for(int i=0; i<=len; i++) {
f[i][0][1] = f[i][0][0] = f[i][1][1] = f[i][1][0] = 0;
}
}
return 0;
}