符合要求的数困难版-数位dp

题目大意:

给定n,求[1,n]中满足要求的数有多少个。
满足要求的描述由输入给定，先给定m,再给定m个[0,9]互不相等的数，需要你求出[1,n]中有多少个数包含了所有的这m个数，结果对998244353取模。

输入：

输入测试数据组数t(t<=10)
输入n(1<=n<=1e15)
输入m(1<=m<=10)
输入m个[0,9]互不相等的数，代表这m个数要全部出现才符合要求。

n最大1e15所以不能暴力枚举很明显是到数位dp

数位dp：

利用枚举每一位从高位向低位枚举满足条件的利用记忆化进行计数

利用二进制保存状态输入的m个数用num存分别num=num|mi;

dp[50][1<<10] (第一个代表目前是第几位，第二个代表目前的值是多少)

数位dp:(54条消息) 数位dp总结 之 从入门到模板_wust_wenhao的博客-CSDN博客_数位dp

#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define int long long
#define PII pair<int,int>
const int N=2e5+10;
const int mod=998244353;
int lowbit(int x){return x&(-x);}
int dir[8][2]={{1,0},{-1,0},{0,1},{0,-1},{1,1},{1,-1},{-1,1},{-1,-1}};
int dp[50][(1<<10)+5];
int num=0;
int n,m;
vector<int>v;
int dfs(int pos,int val,bool is_zero,bool limit)
{
    if(pos==-1) return (val&num)==num;
    if(!limit&&dp[pos][val]!=-1)
    {
        return dp[pos][val];
    }
    int up=(limit?v[pos]:9);
    int ans=0;
    for(int i=0;i<=up;i++)
    {
        if(is_zero&&i==0)
        {
            ans+= dfs(pos-1,0,1,0);
            ans%=mod;
        }
        else
        {
            ans+= dfs(pos-1,val|(1<<i),0,limit&&i==up);
            ans%=mod;
        }
    }
    if(!limit)
    {
        dp[pos][val]=ans;
    }
    return ans;
}
int fun()
{
    int x=n;
    memset(dp,-1,sizeof dp);
    while(x)
    {
        v.emplace_back(x%10);
        x/=10;
    }
    return dfs(v.size()-1,0,1,1);
}
void solve()
{
//    freopen("D:\\c++\\in","r",stdin);
//    freopen("D:\\c++\\out1","w",stdout);
    cin>>n>>m;
    num=0;
    v.clear();
    for(int i=1;i<=m;i++)
    {
        int x;
        cin>>x;
        num|=(1<<x);
    }
    cout<<fun()<<endl;
}
signed main()
{
    IOS
    int _;cin>>_;while(_--)
        solve();
    return 0;
}
/*
 *
 */

代表目前是第几位，第二个代表当前的状态)