Codeforces 515(Div. 3) E (1066E)二进制求前缀和&&与运算

 题意:

现在,给你两个位数为 nn 和 mm 的两个二进制数aa,bb,现在,我们要进行如下操作:

  • 计算aa&bb
  • 答案累加上一个操作的值
  • bb右移一位,最后一位直接舍弃

现在,请你算出最终的答案,并输出,答案对998244353取模

Codeforces 515(Div. 3) E (1066E)二进制求前缀和&&与运算

这题的关键是这个前缀和,因为暴力是1-2e5的和肯定超时,想想如果仅仅是a,b求&怎么运算?肯定是跑一遍n。

算每次b右移的串利用的b高位的1在右移中一定会和a中比它位低的所有1与成1,所以就可以用前缀和计算这个位前面有多少个1,就知道这个a这个位产生了过多少次非零数,加和起来就是总数了。

#include<iostream>
using namespace std;
typedef long long ll;
const ll mod=998244353;
ll pow2[200010],bit[200010];
int main(){
	int n,m;
	string a,b;
	cin>>n>>m;
	cin>>a>>b;
	if(n<=m){
		a=string(m-n,'0')+a;
	}
	else{
		b=string(n-m,'0')+b;
	}
	n=max(n,m);
	pow2[0]=1;
	for(int i=1;i<2e5+10;i++){
		pow2[i]=pow2[i-1]*2%mod;
	}
	bit[0]=(b[0]=='1');
	for(int i=1;i<n;i++){
		bit[i]=bit[i-1]+(b[i]=='1');
	}
	ll sum=0;
	for(int i=0;i<n;i++){
		ll ans=(a[i]=='1');
		ans=ans*pow2[n-i-1]%mod;
		sum+=ans*bit[i]%mod;
		sum%=mod;
	}
	cout<<sum<<endl;
}