手机号码---数位dp

题目链接:https://ac.nowcoder.com/acm/problem/19945

题目描述
人们选择手机号码时都希望号码好记、吉利。比如号码中含有几位相邻的相同数字、不含谐音不吉利的数字等。手机运营商在发行新号码时也会考虑这些因素,从号段中选取含有某些特征的号 码单独出售。为了便于前期规划,运营商希望开发一个工具来自动统计号段中满足特征的号码数 量。
工具需要检测的号码特征有两个:号码中要出现至少3个相邻的相同数字,号码中不能同时出现8和4。号码必须同时包含两个特征才满足条件。满足条件的号码例如:13000988721、 23333333333、14444101000。而不满足条件的号码例如:1015400080、10010012022。 手机号码一定是11位数,前不含前导的0。工具接收两个数L和R,自动统计出[L,R]区间 内所有满足条件的号码数量。L和R也是11位的手机号码。

分析:
这道题我们设置dp数组为dp[pos][pre][ppre][ok][flag1][flag2],pos表示枚举到了第几位数,pre表示当前这位数的前面一位数,ppre表示pre的前面一位数,ok表示是否有三个相同的连续的数,flag1表示是否出现了4,flag2表示是否出现了8,但这道题要注意消除前导0,因为电话号码题目说了没有前导0

代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll a[12];
ll f[12][10][10][2][2][2];
int pre;
/*sum记录相同连续数的个数,flag1记录是否有4,flag2记录是否有8*/
ll dfs(int pos,int limit,int pre,int ppre,int ok,int flag1,int flag2,int first)
{
    
    
	if(!pos)
	{
    
    
		if(!(flag1&&flag2)&&ok) return 1;//有三个连续相同的数字,并且4和8没有同时出现 
		else return 0; 
	}
	if(!limit&&~f[pos][pre][ppre][ok][flag1][flag2]) return f[pos][pre][ppre][ok][flag1][flag2];
	int ed = a[pos];
	if(!limit) ed = 9;
	ll ans = 0;
	for(int i=0;i<=ed;i++)
	{
    
    
		if(first&&i==0) continue;//这里是消除前导0 
		ans+=dfs(pos-1,limit&&(i==ed),i,pre,ok||(i==pre&&i==ppre),flag1||(i==4),flag2||(i==8),0);
	
	}
	if(!limit)  f[pos][pre][ppre][ok][flag1][flag2]=ans;
	return ans;
} 
ll solve(ll n)
{
    
    
	if(n<1e10) return 0;//如果这个数没有11位,那么肯定不满足电话号码要去直接返回0 
	memset(a,0,sizeof(a));
	memset(f,-1,sizeof(f));
	int pos = 0;
	while(n)
	{
    
    
		a[++pos] = n%10;
		n/=10;
	}
//	pre = -1;
	ll ans = dfs(pos,1,0,0,0,0,0,1);
	return ans;
}
int main()
{
    
    
	ll l,r;
	cin>>l>>r;
	ll ans = solve(r)-solve(l-1);
	cout<<ans<<'\n';
	return 0;
}

猜你喜欢

转载自blog.csdn.net/TheWayForDream/article/details/116173391