2019牛客暑期多校训练营（第九场）I KM and M

传送门

思路：因为最后按位与的是一个常数，所以只需要看这个常数对应为1的位置，在M、2M、3M...NM这N个数字中，有多少个仍然是1。用个数乘以对应位的2的幂次即可。那么现在问题变成了如何求这个个数。我们考虑对于一个数字iM，如果求它二进制下第j位是否是0。显然，我们可以先把iM右移j位得到x，然后再把iM右移j+1位得到y，再把y左移1位得到z，z-x的值即为iM第j位的数值。而这个过程，就相当于iM除以2的j次方减去iM除以2的j+1次方的两倍。而我们的iM有多个 用类欧几里得来求。类欧几里得证明  

#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;
typedef __int128 ll;
const int mod=1e9+7;

void scan(__int128 &x)//输入
{
    x = 0;
    int f = 1;
    char ch;
    if((ch = getchar()) == '-') f = -f;
    else x = x*10 + ch-'0';
    while((ch = getchar()) >= '0' && ch <= '9')
        x = x*10 + ch-'0';
    x *= f;
}
void print(__int128 x)//输出
{
    if(x < 0)
    {
        x = -x;
        putchar('-');
    }
     if(x > 9) print(x/10);
    putchar(x%10 + '0');
}

ll f(ll a, ll b, ll c, ll n)// 类欧几里得求f
{
    if(a == 0)
        return (n+1) * (b/c);
    if(a < c && b < c)
    {
        ll m = (a*n+b)/c;
        if(m == 0)
            return 0;
        return n*m - f(c, c-b-1, a, m-1);
    }
    return f(a%c, b%c, c, n) + (n+1)*(b/c) + (n+1)*n/2*(a/c);
}
int main()
{
	ll n,m;
	scan(n);
	scan(m);
	ll ans=0;
	for(ll i=1;i<=n*m;i+=i)
	{
		if(m&i)
		{
			ans+=(f(m,m,i,n-1)%mod-f(m,m,i+i,n-1)*2%mod+mod)*i%mod;
		}
	}
	ans=ans%mod;
	print(ans);
	return 0;
}