codeforces 1066E. Binary Numbers AND Sum

前綴和

題目傳送門

瘋狂划水

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

考慮aa的每一位對答案的貢獻。如果這一位是00沒有貢獻,否則貢獻爲2isum[i]2^i*sum[i],其中sum[i]sum[i]bb從最高位到第ii位的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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章