2024杭电ACM-个人PK赛（1）：1004 合法数对

2024杭电ACM-个人PK赛（1）

1004 合法数对

Time Limit (Java / Others)
2000 / 1000 MS
Memory Limit (Java / Others)
327680 / 327680 K
Ratio (Accepted / Submitted)
12.06% (219/1816)

Problem Description

数对 $(x,y)$ 是好的，当且仅当 $x\oplus y=x\mid y$，其中 $\oplus$ 表示异或操作，$\mid$ 表示或运算。
给定正整数 $N$，请你求解有多少个好的数对 $(x,y)$，满足 $x,y\in [0,N]$​。
因为出题人小 $M$ 比较仁慈，所以 $N$ 将以二进制的形式给出。
答案对 $998244353$ 取模。

Input

输入共 $1$ 行，包含一个正整数 $N$，通过二进制的形式给出。

评测数据规模：
对于所有测评数据，$1\le N<2^{1234567}$。

Output

输出共 $1$ 行，输出 $1$ 个整数，表示最终答案，答案对 $998244353$ 取模。

Sample Input

$111$

Sample Output

$27$

思路分析

认识“异或”和“或”

• 异或 ( $\oplus$ )
  $0\oplus 0=0$
  $0\oplus 1=1$
  $1\oplus 0=1$
  $1\oplus 1=0$
• 或 ( $|$ )
  $0|0=0$
  $0|1=1$
  $1|0=1$
  $1|1=1$

分析与计算

易知当 $(x,y)=(1,1)$ 时不满足题意，由此可以延伸至 $N$ 的每一位。

通过计算得知，输入每增加 $1$，输出的变化量 $D$ 与 $0$ 的个数 $m$ 有关。
① 只有在与 $0$ 进行异或以及或运算的时候才会相等。
② 进行运算的位可以取值 $0$ 或 $1$ 两种，所以 $d=2^m$。
③ $(x,y)$ 与 $(y,x)$ 视为不同的数对，所以 $D=d\times 2=2^{m+1}$。

然后没什么思路了，打个表先，看看能不能找到一点规律。

Input	N.count(0)	D	Output	Funct
$0$	$1$	$1$	$1$
$1$	$0$	$2$	$3$	$1\times 3$
$10$	$1$	$4$	$7$
$11$	$0$	$2$	$9$	$3\times 3$
$100$	$2$	$8$	$17$
$101$	$1$	$4$	$21$	$7\times 3$
$110$	$1$	$4$	$25$
$111$	$0$	$2$	$27$	$9\times 3$
$1000$	$3$	$16$	$43$
$1001$	$2$	$8$	$51$	$17\times 3$
$1010$	$2$	$8$	$59$
$1011$	$1$	$4$	$63$	$21\times 3$
$1100$	$2$	$8$	$71$
$1101$	$1$	$4$	$75$	$25\times 3$
$1110$	$1$	$4$	$79$
$1111$	$0$	$2$	$81$	$27\times 3$
…	…	…	…	…
$xxx0$	$m$	$2^{m+1}$	$f(xxx)\times 3-2^m$
$xxx1$	$m-1$	$2^m$	$f(xxx)\times 3$	$f(xxx)\times 3$

不难看出，当末位为 $1$ 时，可由 $N$ 右移一位的结果乘 $3$ 得到答案。
当末位为 $0$ 时，可由 $N$ 右移一位的结果乘 $3$ 再减去 $d=2^m$ 得到答案。
这里采用减法是因为从 $xxx1$ 到 $xxx0$ 不涉及进位退位，便于计算。

便可以从末位向前遍历，得到的这个计算过程将是倒序的。
为了避免递归的写法爆栈，手动把计算过程压入堆栈（后进先出）达到反序的目的。
本代码使用了 STL 中的 vector 来模拟堆栈。

代码实现

#include<iostream>
#include<vector>
using namespace std;
typedef long long ll;
const ll mod = 998244353;
ll q_pow(ll a,ll b) { ll ans = 1; while(b) { if(b & 1) ans = ans * a % mod; a = a * a % mod; b >>= 1; } return ans % mod; }
vector<ll> v;
int main(){
    ll b = 0; string ss; cin >> ss;
    for(auto p : ss) if(p == '0') b++;          //记录0的个数
    while(!ss.empty()){
        if(*ss.rbegin()=='1') v.push_back(0);   //不需要减，用0占位
        else v.push_back(b--);                  //存入即时0的个数
        ss.pop_back();
    }ll ans=1;
    while(!v.empty()){
        ans *= 3;
        if(*v.rbegin()) ans -= q_pow(2, *v.rbegin()); 
        ans = (ans+mod)%mod;                    //保证ans>0
        v.pop_back();
    }cout << ans << "\n";
}

。。。我写复杂了，去看这篇吧，写得挺好的【2024杭电ACM-个人PK赛(1) 1004合法数对 - HDU7415】