关闭

LightOJ 1021 Painful Base(状压DP)

标签: 状压dp数位dp
150人阅读 评论(0) 收藏 举报
分类:

题意:对一个base进制数,给出它可以包含的数字,对这个数进行全排列,问多少种可以整除k。

思路:base不超过16,看到这个数据量可以想到状压,再联系数位DP。得到转移方程dp[i][nmod] += dp[i - (1<<j)][mod],i表示当前状态,其中i & (1 << j);nmod = mod * base + num[j]。然后进行刷表

#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <cmath>
#include <set>
#include <map>
using namespace std;

typedef long long LL;
#define mem(a, n) memset(a, n, sizeof(a))
#define rep(i, n) for(int i = 0; i < (n); i ++)
#define REP(i, t, n) for(int i = (t); i < (n); i ++)
#define FOR(i, t, n) for(int i = (t); i <= (n); i ++)
#define ALL(v) v.begin(), v.end()
#define si(a) scanf("%d", &a)
#define sii(a, b) scanf("%d%d", &a, &b)
#define siii(a, b, c) scanf("%d%d%d", &a, &b, &c)
#define pb push_back
#define eps 1e-8
const int inf = 0x3f3f3f3f, N = 20, MOD = 1e9 + 7;

int T, cas = 0;
int n, m;
int base, k, num[N];
LL dp[1 << 17][20];
char str[N];

int main(){
#ifdef LOCAL
    freopen("/Users/apple/input.txt", "r", stdin);
//	freopen("/Users/apple/out.txt", "w", stdout);
#endif
	
    si(T);
    while(T --) {
    	sii(base, k);
    	scanf("%s", str);
    	for(int i = 0; i < base; i ++ ) {
    		if(str[i] <= 'F' && str[i] >= 'A') num[i] = str[i] - 'A' + 10;
    		else num[i] = str[i] - '0'; 
    	}
    	mem(dp, 0);
    	n = strlen(str);
    	int all = (1 << n) - 1;	
    	dp[0][0] = 1;

    	for(int i = 1; i <= all; i ++) {
    		for(int j = 0; j < n; j ++) {
    			if(i & (1 << j)) {
    				int s = i - (1 << j);
    				for(int mod = 0; mod < k; mod ++) {
    					int nmod = (mod * base + num[j]) % k;
    					dp[i][nmod] += dp[s][mod];
    				}
    			}
    		}
    	}
    	printf("Case %d: %lld\n", ++ cas, dp[all][0]);
    }
    
    return 0;
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:24463次
    • 积分:1339
    • 等级:
    • 排名:千里之外
    • 原创:115篇
    • 转载:2篇
    • 译文:0篇
    • 评论:0条