codeforces 1066E. Binary Numbers AND Sum

前綴和

題目傳送門

瘋狂划水

題目大意： 兩個很大的二進制數$a,b$，每次把答案加上$a\&b$並把$b$左移一位。求最終答案。

考慮$a$的每一位對答案的貢獻。如果這一位是$0$沒有貢獻，否則貢獻爲$2^i*sum[i]$，其中$sum[i]$爲$b$從最高位到第$i$位的1的個數。掃一遍算一下就好了。

代碼：

#include<cstdio>
#include<cstring>
#include<algorithm>
#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;
}