ZROI #364. 【2018普转提day18专题】嘤嘤嘤

ZROI #364. 【2018普转提day18专题】嘤嘤嘤

直接贴代码

具体见注释

#include<stdio.h>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<cmath>
#include<iostream>
#include<queue>
#include<string>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<long long,long long> pll;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define rep(i,j,k)  for(register int i=(int)(j);i<=(int)(k);i++)
#define rrep(i,j,k) for(register int i=(int)(j);i>=(int)(k);i--)

ll read(){
    ll x=0,f=1;char c=getchar();
    while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0' && c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}

const int maxn=100100,maxm=55;
ll n,m,s,t;
ll tot[maxm][maxn],f[maxm][maxn][2];
ll a[maxn],q[maxn],lim,start;

ll dfs(int x,int y,int z,ll tmp=0){
    //当前确定了后x位，加的数是tmp时第x位上有y个数发生了进位，是否超过了限制lim（z=0/1) 
    if(x==m) return !z; //如果枚举完了，就看当前加的数是否可行 
    if(f[x][y][z]!=-1) return f[x][y][z];
    ll nw=0;
    int a=tot[x][y],b=y-a,c=tot[x][n]-a,d=n-y-c;    
    //a表示按照后x位排序后前y个数有几个第x+1位上是1，这时候a所代表的的数这一位为1且前一位发生了进位，那么这一位实际上会变成0 
    //b表示有几个数发生了进位但是这一位不是1 
    //c表示有几个数这一位是1但是没有进位，在当前情况下b和c代表的数是等价的 
    //d表示剩余的数（既没进位也不是1）的个数 
    int t=(s>>x)&1; //t表示s的这一位是啥 
    //现在有a+d个数这一位是0，b+c个数这一位是1 
    if(((b+c)&1)==t) nw+=dfs(x+1,a,((lim>>x)&1)?0:z,tmp);   //这一位放0 
    if(((a+d)&1)==t) nw+=dfs(x+1,a+b+c,((lim>>x)&1)?z:1,tmp+(1<<x));    //这一位放1 
    return f[x][y][z]=nw;
}

ll solve(ll x){
    lim=x;
    memset(f,-1,sizeof(f));
    return dfs(0,0,0);
}
int main(){
    n=read(),m=read(),s=read(),t=read();
    rep(i,1,n){
        a[i]=read();
        start^=a[i];
    }
    rep(i,0,m-1){
        rep(j,1,n)
            tot[i][j]=tot[i][j-1]+((a[j]>>i)&1);
        int nums=0;
        rep(j,1,n)
            if((a[j]>>i)&1) q[++nums]=a[j];
        rep(j,1,n)
            if(!((a[j]>>i)&1)) q[++nums]=a[j];
        swap(a,q);
        //q表示原来的数按照后i+1位排序之后的结果
        //tot表示原来的数按照后i位排序之后，第i位上前j个数里有几个1 
    }
    printf("%lld\n",solve((1ll<<m)-1)*(t>>m)+solve(t%(1ll<<m))-(start==s));
    return 0;
}

Review

为什么这么做？

首先按位考虑，这个很好想

然后我就没思路了

没有想到数位dp。。。

而且对于每个长度为j的后缀的处理很巧妙