牛客月赛43-满意的集合-(数字类dp)

E

题意:
给你前9个数每个数出现的次数,然后满意的集合定义为,你选择几个数组成的集合,这几个数按某种排列成为10进制数可以被3整除,那么就是好的。问你现在有多少种方案是合法的,同时空集合看作0也是合法的

思考:
看到这题不知道咋想刚开始,看到取模就像dp,然后这种类型的dp没咋见过。就是定义为dp[i][j],前i个数中,现在取模后为j的总方案数。为什么可以这样,因为一个数如果可以被3整除,那么肯定包含369即可。然后包含369的可以从别的数转移过来,这就是转移方程了。

代码:

int T,n,m,k;
int va[N];
int dp[M][M];

signed main()
{
	IOS;
	n = 9;
	for(int i=1;i<=n;i++) cin>>va[i];
	dp[0][0] = 1;
	for(int i=1;i<=n;i++)
	{
		int vb[3] = {0,0,0}; //代表%3后为0,1,2的方案数。
		if(i%3==1) //选择多少个不改变当前的%后的值
		{
			vb[0] = (va[i]+3)/3; //选择0 3 6个...
			vb[1] = (va[i]+2)/3; //选择1 4 7个...
			vb[2] = (va[i]+1)/3; //选择2 5 8个...
		}
		if(i%3==2) 同理
		{
			vb[0] = (va[i]+3)/3;
			vb[1] = (va[i]+1)/3;
			vb[2] = (va[i]+2)/3;
		}
		if(i%3==0) vb[0] = va[i]+1;
		//下面就是经典的转移方程了
		dp[i][0] = (dp[i-1][0]*vb[0]%mod+dp[i-1][1]*vb[2]%mod+dp[i-1][2]*vb[1]%mod)%mod;
		dp[i][1] = (dp[i-1][0]*vb[1]%mod+dp[i-1][1]*vb[0]%mod+dp[i-1][2]*vb[2]%mod)%mod;
		dp[i][2] = (dp[i-1][0]*vb[2]%mod+dp[i-1][1]*vb[1]%mod+dp[i-1][2]*vb[0]%mod)%mod;
	}
	cout<<dp[n][0]%mod;
	return 0;
}

总结:
多多掌握思考推一推这种dp的方程式

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值