jzoj5894 同余方程 容斥

Description

---

Solution

---

NOIp要是考成这shi样可就凉了呀。。

首先可以想到令solve(a,b)为x<=a,y<=b的答案，那么答案就是solve(r1,r2)-solve(l1,r2)-solve(l2,r1)+solve(l1,l2)

现在看怎么搞定solve。考虑枚举i、j分别表示x的前i位等于a，y的前j位等于b，也就是说我们把数字分成了三段。其中第一段x、y都固定了，第二段x、y的其中一个固定了，第三段两个都随便选。于是乎就可以统计方案了。

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)

typedef long long LL;
const int MOD=998244353;
const int N=68;

LL bin[N]; LL m;

LL solve(LL a,LL b) {
	LL ret=0;
	drp(i,59,0) if (a&bin[i]) {
		drp(j,59,0) if (b&bin[j]) {
			int c=std:: max(i,j),d=std:: min(i,j);
			LL v=((((i==j)?0:bin[c])^a^b)|(bin[c]-1))^(bin[c]-1);
			if (v) ret=(ret+(((bin[c]+v-1)/m-(v-1)/m)%MOD+MOD)%MOD*(bin[d]%MOD)%MOD)%MOD;
			else ret=(ret+((bin[c]-1)/m+1)%MOD*(bin[d]%MOD)%MOD)%MOD;
		}
	}
	return ret;
}

int main(void) {
	bin[0]=1; rep(i,1,60) bin[i]=bin[i-1]*2LL;
	LL l1,r1,l2,r2;
	scanf("%lld%lld%lld%lld%lld",&l1,&r1,&l2,&r2,&m);
	LL ans=solve(r1+1,r2+1)-solve(l1,r2+1)-solve(r1+1,l2)+solve(l1,l2);
	printf("%lld\n", (ans%MOD+MOD)%MOD);
	return 0;
}