思路
考虑对于每一个 \(a\) 上数位进行分析。令 \(a_i\) 表示 \(a\) 在二进制表示中从左往右数的第 \(i\) 位上的数字,\(b_i\) 同理。
分类讨论一下 \(a_i\) 的取值对于答案的贡献:
- 如果 \(a_i = 0\),对于这一位无论如何都不会拥有贡献。
- 如果 \(a_i = 1\),因为 \(b\) 会向右移,所以能产生贡献的一定是 \(b_{j} = 1\) 的位置,其中 \(j \in [1,i]\)。那么可以用前缀和维护这个东西,令 \(s_i\) 表示 \(b_{1 \sim i} = 1\) 的数量。所以贡献就是 \(s_i \times 2^{n - i}\)。
当然你需要使 \(a,b\) 位数相同,才好处理,所以在较短的数前面加前导零即可。
Code
#include <bits/stdc++.h>
#define int long long
#define re register
using namespace std;
const int N = 2e5 + 10,mod = 998244353;
int n,m,ans;
int s[N];
string a,b;
inline int Add(int a,int b){
return (a + b) % mod;
}
inline int Mul(int a,int b){
return a * b % mod;
}
inline int qmi(int a,int b){
int res = 1;
while (b){
if (b & 1) res = Mul(res,a);
a = Mul(a,a);
b >>= 1;
}
return res;
}
signed main(){
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
string za,zb;
cin >> n >> m >> a >> b;
for (re int i = n;i < m;i++) za += '0';
for (re int i = m;i < n;i++) zb += '0';
if (za.size()) a = ' ' + za + a;
else a = ' ' + a;
if (zb.size()) b = ' ' + zb + b;
else b = ' ' + b;
n = m = max(n,m);
for (re int i = 1;i <= m;i++) s[i] = s[i - 1] + (b[i] == '1');
for (re int i = 1;i <= n;i++){
if (a[i] == '1') ans = Add(ans,Mul(s[i],qmi(2,n - i)));
}
cout << ans;
return 0;
}
标签:return,int,题解,CF1066E,za,zb,re,Numbers,res
From: https://www.cnblogs.com/WaterSun/p/18264020