Atcoder abc194_F - Digits Paradise in Hexadecimal（状压+数位dp）

题目传送门

题意： 给你一个长度为$2e5$的16进制数$N$，问你在$[1,N]$范围内的16进制数中，有多少个数的数位上有且仅有$K$个不同的数。

思路： 数位dp，f[pos][cnt]表示搜到第pos位，已经有cnt个不同的数时，到达终点的方案数，每次转移的时候有两种情况，第一是目前搜到的全是0且下一个还是0，第二是其他情况。

对于全是0的情况，如果再碰到0，这个0并不能计入贡献。
对于其他情况，我们使用二进制数去压缩已经搜过的数。

最后搜完的时候，我们只需要判断当前二进制数中1的个数是否与k相等。

代码：

#define int long long
const int mod = 1e9+7;
const int INF = 0x3f3f3f3f;
const int N = 2e5+10;

string a="0123456789ABCDEF",s;
int k,f[N][17];
int dfs(int pos,int st,int lim,int pre,int cnt)
{
    if(pos==sz(s)) return (cnt==k);
    if(!lim&&f[pos][cnt]!=-1&&!st) return f[pos][cnt];
    char high = lim?s[pos]:'F';
    int res=0;
    for(int i=0;i<sz(a);i++)
    {
        if(st&&i==0)
            res = (res+dfs(pos+1,st&&i==0,lim&&a[i]==high,pre,cnt))%mod;
        else
            res = (res+dfs(pos+1,st&&i==0,lim&&a[i]==high,pre|(1<<i),cnt+(pre!=(pre|(1<<i)))))%mod;
        if(a[i]==high) break;
    }
    if(!lim&&!st) f[pos][cnt] = res;
    return res;
}
signed main()
{
    cin>>s>>k;
    mem(f,-1);
    cout<<dfs(0,1,1,0,0)<<endl;
}