cf 515 div3 E

传送门:http://codeforces.com/contest/1066/problem/E

题意:两个二进制数a、b ,每次对a&b的结果求和,b右移一位,继续对a&b的结果求和,直至b==0

          a、b长度 <= 2e5,结果对998244353取模

这题思路和某次刷题思路一致,很快就想到了,

a&b,按位与,当a [ i ] == 1 && b [ j ] == 1 时加上该位 2 的幂次权重

e.g.  

     543210

a   100101

b    10111

ans += 2 ^ 0 + 2 ^ 2

b 右移一位

   543210

a   100101

b      1011

   ……

发现 a 不动,对于 a 的每个“1”位,统计 b 中从0到与之对应位的 “1” 的个数,乘上对应位的 2 的幂次就可以啦

对 b 搞一个后缀和,遍历 a,统计求和即可

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<list>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> P;
typedef long double ld;
#define mem(x) memset(x, 0, sizeof(x))
#define me(x) memset(x, -1, sizeof(x))
#define fo(i,n) for(i=0; i<n; i++)
#define sc(x) scanf("%I64d", &x)
#define sca(n,m) scanf("%lld%lld", &n, &m)
#define pr(x) printf("%I64d\n", x)
#define pri(x) printf("%lld ", x)
#define lowbit(x) x&-x
const ll MOD = 998244353;
const ll oo = 1e18;
const ll N = 2e6 + 5;
ll sum[N];
int main()
{
    ll i, j, k;
    ll n, m, t;
    cin>>n>>m;
    string s,p;
    cin>>s>>p;
    for(i=m-1; i>=0; i--)
    {
        if(p[i]=='1') k=1;
        else k=0;
        sum[i]=sum[i+1]+k;
    }
    k=1;
    ll ans=0;
    for(i=n-1, j=m-1; i>=max(0ll,n-m); i--,j--)
    {
        if(s[i]=='1')
        ans+=(sum[0]-sum[j+1])*k, ans%=MOD;
        k*=2;k%=MOD;
    }
    cout<<ans<<endl;
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/op-z/p/11345901.html