SCU 2009(数位dp)

传送门:Zeros and Ones

题意:求总数位为n包含0和1个数相同且整除k的二进制数的个数。

分析:设dp[pos][num][md]表示还有pos位已包含num个1且模k余md的符合条件的二进制数的个数,裸数位dp题。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <queue>
#include <vector>
#define LL long long
#define N 3000010
using namespace std;
int n,k;
LL dp[][][];
LL dfs(int pos,int num,int md,int fzore)
{
if(!pos)
{
return num==n/&&!md;
}
if(~dp[pos][num][md])return dp[pos][num][md];
LL ans=;
for(int i=;i<=;i++)
{
if(fzore)
{
if(i==)ans+=dfs(pos-,num+,(md*+i)%k,);
}
else
{
if(i==)ans+=dfs(pos-,num+,(md*+i)%k,);
else ans+=dfs(pos-,num,(md*+i)%k,);
}
}
dp[pos][num][md]=ans;
return ans;
}
int main()
{
int T,cas=; scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&k);
printf("Case %d: ",cas++);
memset(dp,-,sizeof(dp));
if(k==||n%==)
{
puts("");continue;
}
printf("%lld\n",dfs(n,,,));
}
}
上一篇:github pages搭建个人网站如何添加导航


下一篇:linux内核分析第六周-分析Linux内核创建一个新进程的过程