HDU 5965 [模拟+枚举+DP]

该篇博客主要介绍了一种使用枚举和动态规划的方法来解决数组中雷区分布的问题。博主通过初始化dp数组和num数组来记录状态,并通过循环更新dp数组,根据dp数组的状态判断是否满足条件,最后计算符合条件的方案数。代码中涉及到的主要技术包括动态规划、数组操作和条件判断。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目链接

//模拟枚举+dp 
//只要确定了第一位的1雷的个数,后面一的个数都是确定不变的,所以我们只需枚举第一位上雷的个数,通过dp就能得到最后结果
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll mod=100000007;
int dp[10005];//dp[i]第i位上有几个1 
int num[10005];//第i位上的雷的个数 
void init(){
	memset(dp,0,sizeof(dp));
	memset(num,0,sizeof(num));
}
int main(){
	int T;
	cin>>T;
	while(T--){
		char s[10005];
		scanf("%s",s);
		int n=strlen(s);
		for(int i=0;i<n;i++){
			num[i+1]=s[i]-'0';
		}
		//dp[i]=num[i-1]-dp[i-1]-dp[i-2]
		ll ans0=1,ans1=1,ans2=1;
		//分别代表dp[1]=0,1,2时的个数 
		//分别讨论 
		dp[1]=0;
		int k=1; 
		if(1){
			for(int i=2;i<=n;i++){
				//dp公式 
				dp[i]=num[i-1]-dp[i-1]-dp[i-2]; 
				//超出范围就不合适 
				if(dp[i]<0||dp[i]>2){
					k=0;break;
				}
			}
			//符合条件就计算 
			if(num[n]==dp[n]+dp[n-1]&&k){
				for(int i=1;i<=n;i++){
					if(dp[i]==1)ans0*=2;
					ans0%=mod;
				}
			}
			else ans0=0;
		}
		dp[1]=1;k=1;
		if(num[1]>=1){
			for(int i=2;i<=n;i++){
				dp[i]=num[i-1]-dp[i-1]-dp[i-2];
				if(dp[i]<0||dp[i]>2){
					k=0;break;
				}
			}
			if(num[n]==dp[n]+dp[n-1]&&k){
				for(int i=1;i<=n;i++){
					if(dp[i]==1)ans1*=2;
					ans1%=mod;
				}
			}
			else ans1=0;
		}
		else {
			ans1=0;
		}
		dp[1]=2;k=1;
		if(num[1]>=2){
			for(int i=2;i<=n;i++){
				dp[i]=num[i-1]-dp[i-1]-dp[i-2];
				if(dp[i]<0||dp[i]>2){
					k=0;break;
				}
			}
			if(num[n]==dp[n]+dp[n-1]&&k){
				for(int i=1;i<=n;i++){
					if(dp[i]==1)ans2*=2;
					ans2%=mod;
				}
			}
			else ans2=0;
		}
		else {
			ans2=0;
		}
		printf("%lld\n",(ans0+ans1+ans2)%mod);		
	}
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值