[CCF 201312-4] 有趣的数 (数位DP) Apare_xzc

[CCF 201312-4] 有趣的数


题面


分析:

1. 一看就是数位DP

2. 这个数位DP只求满足条件的n位数的个数,所以比传统的规定了上下界的题目要简一点儿

3. 不让有前导零看似比较麻烦

4. 但是根据限制,0必须在1之前,2必须在3之前,于是我们可以得到这个有用的信息:首位一定是2

5. 还规定0,1,2,3都要至少出现一次

6. 于是我们可以开几个变量记录数字0,1,2,3在高位是否出现过,命名为has0,has1,has2,has3

7. 令dp[pos][has0][has1][has2][has3]表示从0到最大的pos+1位数,前面出现了0,1,2,3对应的数的个数(瞎扯,自己理解好了)

8. 然后开心地记忆化搜索


代码:

#include <bits/stdc++.h>
using namespace std;
const int mod = 1000000007;
#define LL long long 
LL dp[1050][2][2][2][2];
LL dfs(int pos,bool has0,bool has1,bool has2,bool has3,bool isFirst) 
{
	if(pos==-1)
	{
		if(has0&&has1&&has2&&has3) return 1;
		else return 0; 
	} 
	if(dp[pos][has0][has1][has2][has3]!=-1)
	{
		return dp[pos][has0][has1][has2][has3];
	}
	if(isFirst)
	{
		return dp[pos][has0][has1][has2][has3] = dfs(pos-1,has0,has1,true,has3,false);
	}
	LL ans = 0;
	//该位填0
	if(!has1)
	{
		ans = (ans+dfs(pos-1,true,has1,has2,has3,false))%mod;
	}
	
	//该位填1 
	if(has0)
	{
		ans = (ans+dfs(pos-1,has0,true,has2,has3,false))%mod;		
	} 
	
	//该位填2 
	if(!has3)
	{
		ans = (ans+dfs(pos-1,has0,has1,true,has3,false))%mod;
	}
	//该位填3 
	if(has2)
	{
		ans = (ans+dfs(pos-1,has0,has1,has2,true,false))%mod;
	}
	
	dp[pos][has0][has1][has2][has3] = ans;
	return ans;	
} 
LL solve(int n)
{
	if(n<=3) return 0;
	return dfs(n-1,false,false,false,false,true);
}

int main()
{
	int n;
	memset(dp,-1,sizeof(dp));	
	while(cin>>n) 
		printf("%lld\n",solve(n));
	
	return 0;
}

100分

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务