2021-2022-1 ACM集训队每周程序设计竞赛（4） - 问题 A:异或&加法: - 题解

题意：
给定一个二进制表示的数 $n$，问有多少对自然数 $(x,y)$ 满足 $x+y⩽n$且$x+y==x\oplus y$。

思路：
这是一道数位$dp$。
由于异或是无进位相加，所以 $x+y==x\oplus y$ 等价于 $x$ 和 $y$ 转化为二进制之后，每一位都不会出现两个${}^{\prime }{1}^{\prime }$。
设 $x_i$ 表示 $x$ 的二进制表示形式的第 $i$ 位。

首先考虑一下特殊情况，如果$n$的二进制表示形式全部是都是由${}^{\prime }{1}^{\prime }$组成的，那么可以想象到，$x,y,x+y$ 均 $⩽n$，即对于 $i(1⩽i⩽length(n))$，都可能出现 x i = = 0 & y i = = 0 ( { 0 , 0 } ) x_{i}==0\&y_{i}==0 (\{0,0\}) xi​==0&yi​==0({0,0})、 x i = = 1 & y i = = 0 ( { 1 , 0 } ) x_{i}==1\&y_{i}==0 (\{1,0\}) xi​==1&yi​==0({1,0})、 x i = = 0 & y i = = 1 ( { 0 , 1 } ) x_{i}==0\&y_{i}==1 (\{0,1\}) xi​==0&yi​==1({0,1}) 这三种情况，因此 $(x,y)$ 总共有 $3^{length(n)}$ 种。

但是如果$n$的二进制表示形式中有 ${}^{\prime }{0}^{\prime }$的话就会比较难办了，因为 $3^{length(n)}$ 种情况中必然有一些使 $x+y>n$ 等不合法的自然数对出现。 既然这样，我们只需要将 $x+y<n$ 和 $x+y=n$ 的情况均求出来即可。
设 $d[i][0]$ 表示 $(x+y{)}_j==n_j(1⩽j⩽i)$ 的情况数，$d[i][1]$ 表示 $1⩽j⩽i$ 中存在至少一位 $j$，使得 $(x+y{)}_j<n_j$ 的情况数。

那么当 $n_i=1$ 时，对于相等的情况，有 { 1 , 0 } 、 { 0 , 1 } \{1,0\}、\{0,1\} {1,0}、{0,1} 这两种可能，因此$d[i][0]=d[i-1][0]\ast 2$ ；而对于小于的情况，可以是前$i-1$位小于的方案数 $\ast 3$，表示 { 0 , 0 } 、 { 1 , 0 } 、 { 0 , 1 } \{0,0\}、\{1,0\}、\{0,1\} {0,0}、{1,0}、{0,1} 三种情况都合法，即$d[i][1]=d[i-1][1]\ast 3$；也可以是前$i-1$位均相等的方案，再加上 { 0 , 0 } \{0,0\} {0,0} ，即在第 $i$ 位时开始小于，$d[i][1]+=d[i-1][0]$。
而当 $n_i=0$ 时，小于的情况只可能是前$i-1$位小于的方案数 $\ast 3$，$d[i][1]=d[i-1][1]\ast 3$，而等于也只有 { 0 , 0 } \{0,0\} {0,0} 一种情况，$d[i][0]=d[i-1][0]$。

注意初始化时，$d[1][0]=1,d[1][1]=0$。
下面代码中 $d[i][0]$ 表示为 $res$、 $d[i][1]$ 表示为 $cnt$。

时间复杂度：$O（n）$

#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9 + 7;

int main() {
	string n;
	cin >> n;

	long long res = 1, cnt = 0;
	for (long long i = 0; i < n.size(); i++) {
		cnt = (cnt * 3) % mod;//d[i][1]=d[i-1][1]*3

		if (n[i] == '1') {
			cnt = (cnt + res) % mod;//d[i][1]+=d[i-1][0]
			res = (res * 2) % mod;//d[i][0]=d[i-1][0]*2
		}
	}

	cout << (res + cnt) % mod;//d[length(n)][0]+d[length(n)][1]
	return 0;
}