bzoj千题计划277：bzoj4513: [Sdoi2016]储能表

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

f[i][0/1][0/1][0/1] 从高到低第i位，是否卡n的上限，是否卡m的上限，是否卡k的下限 的方案数

g[i][0/1][0/1][0/1] 对应 f 的和

#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;

int f[][][][],g[][][][];

template<typename T>
void read(T &x)
{
    x=; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*+c-''; c=getchar(); }
}

void solve()
{
    long long n,m,k; int p;
    read(n); read(m); read(k); read(p);
    memset(f,,sizeof(f));
    memset(g,,sizeof(g));
    f[][][][]=;
    int A,B,C,z;
    int ni,mi,ki;
    for(int i=;i>=;--i)
        for(int a=;a<=;++a)
            for(int b=;b<=;++b)
                for(int c=;c<=;++c)
                    if(f[i+][a][b][c])
                    {
                        ni=n>>i&; mi=m>>i&; ki=k>>i&;
                        for(int x=;x<=;++x)
                            if(!a || x<=ni)
                                for(int y=;y<=;++y)
                                    if(!b || y<=mi)
                                    {
                                        z=x^y;
                                        if(c && z<ki) continue;
                                        A=a&x==ni; B=b&&y==mi; C=c&&z==ki;
                                        f[i][A][B][C]=(f[i][A][B][C]+f[i+][a][b][c])%p;
                                        g[i][A][B][C]=(g[i][A][B][C]+g[i+][a][b][c])%p;
                                        if(z) g[i][A][B][C]=(g[i][A][B][C]+(1LL<<i)%p*f[i+][a][b][c]%p)%p;
                                    }
                    }
    printf("%d\n",(g[][][][]-k%p*f[][][][]%p+p)%p);
}

int main()
{
    int T;
    read(T);
    while(T--) solve();
    return ;
}