【BZOJ 4513】【SDOI 2016】储能表

http://www.lydsy.com/JudgeOnline/problem.php?id=4513

设\(f(i,0/1,0/1,0/1)\)和\(g(i,0/1,0/1,0/1)\)分别表示dp到第i位xx卡不卡n的上界，yy卡不卡m的上界，xx xor yy卡不卡k的下界的数字总和和数字个数。

转移时枚举上一位的状态和这一位的数字来转移，这一位的状态可以通过前两者算出来。

当有多个数的时候可以把它们放在一起dp！！！

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;

ll f[64][2][2][2], g[64][2][2][2];

int main() {
	int T, x, y, z, zz, aa, bb, cc; scanf("%d", &T);
	ll n, m, k, p;
	while (T--) {
		memset(f, 0, sizeof(f));
		memset(g, 0, sizeof(g));
		scanf("%lld%lld%lld%lld", &n, &m, &k, &p);
		g[63][1][1][1] = 1;
		for (int i = 62; i >= 0; --i) {
			x = (n >> i) & 1;
			y = (m >> i) & 1;
			z = (k >> i) & 1;
			for (int xx = 0; xx < 2; ++xx)
				for (int yy = 0; yy < 2; ++yy) {
					zz = xx ^ yy;
					for (int a = 0; a < 2; ++a)
						for (int b = 0; b < 2; ++b)
							for (int c = 0; c < 2; ++c) {
								if (a && xx > x) continue;
								if (b && yy > y) continue;
								if (c && zz < z) continue;
								aa = a ? xx == x : 0;
								bb = b ? yy == y : 0;
								cc = c ? zz == z : 0;
								(f[i][aa][bb][cc] += ((f[i + 1][a][b][c] + (1ll * (zz - z + p) % p * ((1ll << i) % p) % p * g[i + 1][a][b][c] % p)) % p)) %= p;
								(g[i][aa][bb][cc] += g[i + 1][a][b][c]) %= p;
							}
				}
		}
		printf("%lld\n", f[0][0][0][0]);
	}
}