Codeforces Round #235 (Div. 2) D. Roman and Numbers

http://codeforces.com/problemset/problem/401/D

给一段数字串和m 求数字除以m余0 的组合数

状态压缩dp

#include<iostream>
#include<cstring>
#include<cstdio>

using namespace std;

typedef __int64 LL;

char str[20];
LL dp[1<<18][100];
LL mu[20]; 
LL len;
int bit[20];
int c[20];
int mod;
void init()
{
	len=strlen(str);
	mu[0]=1;                                   //阶乘
	for(int i=0;i<len;i++)
	{
		bit[i]=str[i]-'0';
		mu[i+1]=mu[i]*(i+1);
		c[str[i]-'0']++;
		if(bit[i])
			dp[1<<i][bit[i]%mod]=1;    //0不能出现在第一位
	}
}

int main()
{
	scanf("%s%d",str,&mod);
	init();
	for(LL i=1;i<(1<<len);i++)
	{
		for(int j=0;j<mod;j++)
		{
			if(dp[i][j]==0)
				continue;
			for(LL k=0;k<len;k++)
			{
				if((i&(1<<k))==0)
					dp[i|(1<<k)][(10*j+bit[k])%mod]+=dp[i][j];         //将bit[k]排在最后
			}
		}
	}
	LL ans=dp[(1<<len)-1][0];
	for(int i=0;i<10;i++)
		if(c[i])
			ans/=mu[c[i]];
	printf("%lld\n",ans);
//	system("pause");
	return 0;
}