对每一位数字考虑,假设横坐标和纵坐标为x和y,对于每一位的(x,y)都有4种情况。
令f/g[i][a][b][c]表示第i位上:
a=0:x小于n a=1:x等于n
b=0:y小于m b=1:y等于m
c=0:x^y大于p c=1:x^y等于p
f:符合要求的数的个数,g:符合要求的x^y的和。
程序里面是从(a,b,c)状态推到(A,B,C)状态,这个转移的确有点复杂。有一些状态是不符合条件的,比如当前第i位a=1表示前i位目前卡着n转移,这个时候如果n的第i+1位是0,x的第i+1位就不可能是1,否则这个新状态就大于n了。这个判断情况还是挺多的,但是刚好用一个位运算可以套出来。
PS:这个题有一个k=0和k=1的部分分,很容易就会先用打表找规律,然后如果你足够强的话就可以用一种看起来相当显然和简单的方法完爆恶心的dp啦!
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<iomanip>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#define ll long long
#define inf 1000000000
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
#define N 100
using namespace std;
int T;
ll n,m,k,mo,res;
int i,a,b,c,x,y,z,p,q,t,A,B,C;
ll ex[N],f[N][3][3][3],g[N][3][3][3];
int main()
{
scanf("%d",&T);
while (T--)
{
scanf("%lld%lld%lld%lld",&n,&m,&k,&mo);
n--; m--; res = 0; ex[0] = 1;
fo(i,1,63) ex[i] = (ll)ex[i-1] * 2 % mo;
memset(f,0,sizeof(f)); memset(g,0,sizeof(g));
f[0][1][1][1] = 1; g[0][1][1][1] = 0;
fo(i,0,63)
fo(a,0,1)
fo(b,0,1)
fo(c,0,1)
if (f[i][a][b][c] > 0)
{
int p=((n&((ll)1<<(63-i)))==0)?0:1; //判断n、m、k的第i位
int q=((m&((ll)1<<(63-i)))==0)?0:1;
int t=((k&((ll)1<<(63-i)))==0)?0:1;
fo(x,0,1)
fo(y,0,1)
{
if (a && x > p) continue;
if (b && y > q) continue;
z = x ^ y;
if (c && z < t) continue;
A = (a && x == p) ? 1 : 0;
B = (b && y == q) ? 1 : 0;
C = (c && z == t) ? 1 : 0;
f[i+1][A][B][C]=(f[i+1][A][B][C]+f[i][a][b][c])%mo;
g[i+1][A][B][C]=(g[i+1][A][B][C]+g[i][a][b][c]+((z==0)?0:ex[63-i])*f[i][a][b][c]%mo)%mo;
}
}
k = k % mo;
fo(a,0,1)
fo(b,0,1)
fo(c,0,1)
res+=((g[64][a][b][c] - k*f[64][a][b][c])%mo+mo)%mo;
printf("%lld\n",res%mo);
}
return 0;
}