AtCoder&Luogu[ABC129E] Sum Equals Xor 题解

一看题目，就是一道 dp 题。

我们不妨来确定一个设数：设 $d{p}_{i,0/1}$ 表示前 $i$ 个数与 $l$ 前 $i$ 个数的关系，那就有两种情况：

• 如果 $j=0$ 则表示这前 $i$ 个数小于 $l$ 前 $i$ 个数。

• 如果 $j=1$ 则表示这前 $i$ 个数等于 $l$ 前 $i$ 个数。

那我们就可以得出两种情况：

• 如果 $l_i=1$ ，则:

  • $d{p}_{i,0}=d{p}_{i-1,0}\times 3+d{p}_{i-1,1}$，解释：前一个的方案既然等于前 $l$ 的 $i$ 个数，那就有 $(1,0),(0,1),(0,0)$，三种情况，以及之前为等于它的前 $i$ 位有 $(0,0)$ 一种方法，因为 $(1,0),(0,1)$ 会使之相等，所以不行。

  • $d{p}_{i,1}=d{p}_{i-1,1}\times 2$，解释：只有两种方案，放 $(1,0),(0,1)$。

• 如果 $l_i=0$ ,则：
  • $d{p}_{i,0}=d{p}_{i-1,0}\times 3$，解释：三种方案，同上，只不过不能再之前相等的时候填 $(0,0)$ 会导致这个答案相等，不是 $d{p}_{i,0}$。
  • $d{p}_{i,1}=d{p}_{i-1,1}$，解释：之前相等了，只有一种方案 $(0,0)$，所以只有一个 $d{p}_{i-1,1}$。

代码

#include<iostream>
#define int long long
using namespace std;
int n,dp[105000][2];
string s,s1;
const int mod=1e9+7;
signed main(){
	cin>>s1;
	n=s1.size();
	s=' ',s+=s1;
	dp[0][1]=1;//初始化
	for(int i=1;i<=n;i++){
		dp[i][0]=(dp[i-1][0]*3)%mod;
		if(s[i]=='1'){
			dp[i][0]=(dp[i][0]+dp[i-1][1])%mod;
			dp[i][1]=(dp[i-1][1]*2)%mod;
		}
		else dp[i][1]=dp[i-1][1];
	}
	cout<<(dp[n][0]+dp[n][1])%mod;
	return 0;
}