P4882 lty loves 96!
题目背景
众所周知,(lty)非常喜欢(96)这两个数字(想歪的现在马上面壁去),更甚于复读(人本复)!
题目描述
由于爱屋及乌,因此,(lty)对于那些含有(96)的数也非常喜欢,而这里的含有不是一般的含有,而是具有以下性质的含有(三条都需要满足):
这个数为一个(N)位数,且没有前置零
这个数中至少要出现(M)次(9)和(6)(例:(986996)中出现了(5)次,(9)出现了(3)次,(6)出现了(2)次,共计(5)次)
这个数存在任意连续的从高到低三位(A,B,C),满足下面任意一条
(A+B+C)为(9)或(6)
((A^2+B^2)\%C)为(9)或(6),如果(C)为(0),则该条件视为不满足
输入输出格式
输入格式:
一行,两个数(N),(M)
输出格式:
一个数,表示这样的数的个数
说明
对于10%的数据,(N<=6)
对于40%的数据,(N<=18)
对于100%的数据,(N<=50,0<=M<=N)
人生第一道数位DP
(dp_{i,j,k,l,q})从高到低前(i)位第(i)位为(j),第(i-1)位为(k),一共有(l)个(6)或(9),是否已经满足条件的方案数
注意前导0,注意要压位高精。
Code:
#include <cstdio>
#include <cstring>
int max(int x,int y){return x>y?x:y;}
struct node
{
int num[20];
node()
{
memset(num,0,sizeof(num));
num[0]=1;
}
node(int i)
{
memset(num,0,sizeof(num));
num[0]=1,num[1]=i;
}
node friend operator +(node n1,node n2)
{
int len=max(n1.num[0],n2.num[0]);
node n3;
n3.num[0]=len;
for(int i=1;i<=len;i++)
{
n3.num[i]+=n1.num[i]+n2.num[i];
n3.num[i+1]=n3.num[i]/10000;
n3.num[i]%=10000;
}
if(n3.num[len+1]) ++n3.num[0];
return n3;
}
}dp[51][10][10][51][2];
int n,m;
int main()
{
scanf("%d%d",&n,&m);
for(int i=0;i<=9;i++)
for(int j=1;j<=9;j++)
{
if((i==6||i==9)&&(j==6||j==9))
dp[2][i][j][2][0]=node(1);
else if(i==6||i==9)
dp[2][i][j][1][0]=node(1);
else if(j==6||j==9)
dp[2][i][j][1][0]=node(1);
else
dp[2][i][j][0][0]=node(1);
}
for(int i=3;i<=n;i++)
for(int l3=0;l3<=9;l3++)
for(int l2=0;l2<=9;l2++)
for(int l1=0;l1<=9;l1++)
{
int s1=l1+l2+l3,s2=l1?(l3*l3+l2*l2)%l1:0;
for(int j=0;j<=i;j++)
{
if(s1==6||s1==9||s2==6||s2==9)
{
if(l1==6||l1==9)
{
if(!j) continue;
dp[i][l1][l2][j][1]=dp[i][l1][l2][j][1]
+dp[i-1][l2][l3][j-1][1]
+dp[i-1][l2][l3][j-1][0];
}
else
{
dp[i][l1][l2][j][1]=dp[i][l1][l2][j][1]
+dp[i-1][l2][l3][j][1]
+dp[i-1][l2][l3][j][0];
}
}
else
{
if(l1==6||l1==9)
{
if(!j) continue;
dp[i][l1][l2][j][0]=dp[i][l1][l2][j][0]
+dp[i-1][l2][l3][j-1][0];
dp[i][l1][l2][j][1]=dp[i][l1][l2][j][1]
+dp[i-1][l2][l3][j-1][1];
}
else
{
dp[i][l1][l2][j][0]=dp[i][l1][l2][j][0]
+dp[i-1][l2][l3][j][0];
dp[i][l1][l2][j][1]=dp[i][l1][l2][j][1]
+dp[i-1][l2][l3][j][1];
}
}
}
}
node ans=node(0);
for(int i=0;i<=9;i++)
for(int j=0;j<=9;j++)
for(int l=m;l<=n;l++)
ans=ans+dp[n][i][j][l][1];
printf("%d",ans.num[ans.num[0]]);
for(int i=ans.num[0]-1;i;i--)
{
if(ans.num[i]<1000) printf("0");
if(ans.num[i]<100) printf("0");
if(ans.num[i]<10) printf("0");
printf("%d",ans.num[i]);
}
return 0;
}
2018.9.9