codeforces 401D. Roman and Numbers 数位dp

题目链接

给出一个<1e18的数， 求将他的各个位的数字交换后， 能整除m的数的个数。

用状态压缩记录哪个位置的数字已经被使用了， 具体看代码。

 #include<bits/stdc++.h>
 using namespace std;
 #define pb(x) push_back(x)
 #define ll long long
 #define mk(x, y) make_pair(x, y)
 #define lson l, m, rt<<1
 #define mem(a) memset(a, 0, sizeof(a))
 #define rson m+1, r, rt<<1|1
 #define mem1(a) memset(a, -1, sizeof(a))
 #define mem2(a) memset(a, 0x3f, sizeof(a))
 #define rep(i, a, n) for(int i = a; i<n; i++)
 #define ull unsigned long long
 typedef pair<int, int> pll;
 const double PI = acos(-1.0);
 const double eps = 1e-;
 const int mod = 1e9+;
 const int inf = ;
 const int dir[][] = { {-, }, {, }, {, -}, {, } };
 int digit[], m, cnt;
 ll dp[<<][];
 ll dfs(int s, int num) {            //s是当前状态， num是模m剩余的数
     if(num==&&s==)
         return ;
     if(~dp[s][num])
         return dp[s][num];
     int used[];
     mem(used);
     ll ret = ;
     for(int i = ; i<cnt; i++) {
         if(!(s&(<<i)))             //如果这一位已经被使用了就直接continue
             continue;
         if(used[digit[i]])
             continue;
         used[digit[i]] = ;
         ret += dfs(s^(<<i), (num*+digit[i])%m);
     }
     return dp[s][num] = ret;
 }
 int main()
 {
     int num[];
     mem(num);
     mem1(dp);
     ll n;
     cin>>n>>m;
     while(n) {
         digit[cnt++] = n%;
         n/=;
     }
     ll ans = ;
     ll s = (<<cnt)-;          //初始全都没有被使用
     for(int i = ; i<cnt; i++) {
         if(digit[i]) {
             if(num[digit[i]])       //某一位上的数不能够重复计数
                 continue;
             num[digit[i]] = ;
             ans += dfs(s^(<<i), digit[i]%m);       //s^(1<<i)说明第i位被使用了
         }
     }
     cout<<ans<<endl;
     return ;
 }