ABC 129 E - Sum Equals Xor

E - Sum Equals Xor
f[i][1]表示前j位，等于给定数L的方案数
f[i][1]表示前i位，严格小于给定数L的方案数
有状态转移方程：
p[i]=='1':

f[i][0]=f[i-1][0]*2%mod;  //第一个数*2+1,第二个数*2或者反过来
f[i][1]=(f[i-1][0]+f[i-1][1]*3)%mod; //要小于当前数，则为前一位的相等的数加上前一位小于的数中的两个分别*2+1或者都只*2

p[i]=='0'

f[i][0]=f[i-1][0];
f[i][1]=f[i-1][1]*3%mod;

Code:

#define int LL
const int N=200010,M=N*2,mod=1e9+7;
string p;
LL f[N][2],n;
//f[i][0] a+b==L  f[i][1] a+b<L
void solve(){
	cin>>p;
	n=p.size(); 
	p=" "+p;
	f[0][0]=1,f[0][1]=0;
	for(int i=1;i<=n;++i){
		if(p[i]=='1'){
			f[i][0]=f[i-1][0]*2%mod;
			f[i][1]=(f[i-1][0]+f[i-1][1]*3)%mod;
		}
		else{
			f[i][0]=f[i-1][0];
			f[i][1]=f[i-1][1]*3%mod;
		}
	}
	print(Mod(f[n][0]+f[n][1], mod));
}