【牛客】被3整除的子序列(区间DP)

链接:https://ac.nowcoder.com/acm/problem/21302
来源:牛客网

时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 524288K,其他语言1048576K
64bit IO Format: %lld
题目描述
给你一个长度为50的数字串,问你有多少个子序列构成的数字可以被3整除
答案对1e9+7取模
输入描述:
输入一个字符串,由数字构成,长度小于等于50
输出描述:
输出一个整数
示例1
输入
复制
132
输出
复制
3
示例2
输入
复制
9
输出
复制
1
示例3
输入
复制
333
输出
复制
7
示例4
输入
复制
123456
输出
复制
23
示例5
输入
复制
00
输出
复制
3
备注:
n为长度
子任务1: n <= 5
子任务2: n <= 20
子任务3: 无限制

思想:
dp[i][j][0]表示i,j之间 子序列值%30 的子序列个数
dp[i][j][1]表示i,j之间 子序列值%3
1 的子序列个数
dp[i][j][2]表示i,j之间 子序列值%3==2 的子序列个数
则很容易得到状态转移方程:

s[j + 1] - ‘0’ % 3 == 0时:

  •  		dp[i][j + 1][0] = 2*dp[i][j][0] + 1;
     		dp[i][j + 1][1] = 2*dp[i][j][1];
     		dp[i][j + 1][2] = 2*dp[i][j][2];
    

s[j + 1] - ‘0’ % 3 == 1时:

  •  		dp[i][j + 1][0] = dp[i][j][0] + dp[i][j][2];
     		dp[i][j + 1][1] = dp[i][j][1] + dp[i][j][0] + 1;
     		dp[i][j + 1][2] = dp[i][j][1] + dp[i][j][2];
    

s[j + 1] - ‘0’ % 3 == 2时:

  •  		dp[i][j + 1][0] = dp[i][j][0] + dp[i][j][1];
     		dp[i][j + 1][1] = dp[i][j][2] + dp[i][j][1];
     		dp[i][j + 1][2] = dp[i][j][0] + dp[i][j][2] + 1;
    

代码如下:

#include<bits/stdc++.h>

using namespace std;

const int mod = 1e9 + 7;

int dp[50][50][4];

int main(){
	string s;
	cin>>s;
	for(int i = s.length();i >= 1;i--){
		s[i] = s[i - 1];
	}
	for(int i = 1;i <= s.length();i++){
		dp[i][i][(s[i]-'0')%3]=1;
	}
	for(int i = 1;i <= s.length();i++){
		for(int j = i;j <= s.length();j++){
			if((s[j + 1]  - '0')% 3 == 0){
				dp[i][j + 1][0] = 2*dp[i][j][0] + 1;
				dp[i][j + 1][1] = 2*dp[i][j][1];
				dp[i][j + 1][2] = 2*dp[i][j][2];
			}else if((s[j + 1]  - '0')% 3 == 1){
				dp[i][j + 1][0] = dp[i][j][0] + dp[i][j][2];
				dp[i][j + 1][1] = dp[i][j][1] + dp[i][j][0] + 1;
				dp[i][j + 1][2] = dp[i][j][1] + dp[i][j][2];
			}else{
				dp[i][j + 1][0] = dp[i][j][0] + dp[i][j][1];
				dp[i][j + 1][1] = dp[i][j][2] + dp[i][j][1];
				dp[i][j + 1][2] = dp[i][j][0] + dp[i][j][2] + 1;
			}
			for(int k = 0;k < 3;k++){
				dp[i][j + 1][k] = dp[i][j + 1][k] % mod;
			}
		}
	}
	cout<<dp[1][s.length()][0];
	return 0;
} 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值