P4124 [CQOI2016]手机号码

传送门

简单的数位dp。就是dfs的参数有亿点点长。设 \(dfs(pos,pre1,pre2,pre4,pre8,flag,lim)\) 分别表示搜到了第几位,前一位数,前两位数,是否出现过4,是否出现过8,是否已经有三个连续的数,是否需要限制。再用一个 \(f\) 数组记忆化。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll l,r,len,f[12][11][11][2][2][2][2],a[12];

ll dfs(int pos,int pre1,int pre2,int pre4,int pre8,int flag,int lim){
	if(pre4&&pre8)return 0;
	if(pos==0)return flag;
	if(f[pos][pre1][pre2][pre4][pre8][flag][lim]!=-1)return f[pos][pre1][pre2][pre4][pre8][flag][lim];
	ll sum=0,mx=lim?a[pos]:9;
	for(int i=0;i<=mx;i++)sum+=dfs(pos-1,i,pre1,pre4||(i==4),pre8||(i==8),flag||(i==pre1&&i==pre2),lim&&i==mx);
	return f[pos][pre1][pre2][pre4][pre8][flag][lim]=sum;
}

ll work(ll num){
	if(num<1e10)return 0;
	memset(f,-1,sizeof(f));
	len=0;
	while(num)a[++len]=num%10,num/=10;
	ll sum=0;
	for(int i=1;i<=a[len];i++)sum+=dfs(10,i,0,i==4,i==8,0,i==a[len]);
	return sum;
}

int main(){
	scanf("%lld%lld",&l,&r);
	printf("%lld",work(r)-work(l-1));
	return 0;
}
posted @ 2022-11-13 20:16  ChenJHen  阅读(29)  评论(0)    收藏  举报