codeforces 1066E. Binary Numbers AND Sum

前缀和

题目传送门

疯狂划水

题目大意： 两个很大的二进制数$a,b$，每次把答案加上$a\&b$并把$b$左移一位。求最终答案。

考虑$a$的每一位对答案的贡献。如果这一位是$0$没有贡献，否则贡献为$2^i\ast sum[i]$，其中$sum[i]$为$b$从最高位到第$i$位的1的个数。扫一遍算一下就好了。

代码：

#include
#include
#include
#define N 200005
using namespace std;
const int p=998244353;
int n,m,ans,pw[N],s[N];
char s1[N],s2[N];
int main(){
	scanf("%d%d%s%s",&n,&m,s1+1,s2+1);
	for (int i=pw[0]=1;i<=m;i++) pw[i]=pw[i-1]*2%p;
	for (int i=1;i<=m;i++) s[i]=s[i-1]+s2[i]-48;
	for (int i=max(1,n-m+1);i<=n;i++)
		ans=1ll*(ans+1ll*pw[n-i]*s[i+m-n]*(s1[i]-48))%p;
	return printf("%d\n",ans),0;
}