2021.8.10提高B组模拟2T4 数字（dp）

Posted 2022-01-10 SSL_LKJ

数字

题目大意

输入样例

2
0987654321

输出样例

1240

题目数据

解题思路

这题就是 dp

设 $f_i{,}_j$为长度为 i，和为 j 的方案数

可得 $f_i{,}_j=f_i{,}_j+f_i{}_{-}{}_1{,}_j{}_{-}{}_a{}_{[}{}_k{}_{]}$（ a数组 为 数字集合S中的所有数字）

最后答案就是分别符合

$它前n位之和与后n位之和相等$ 和 $它奇数位之和与偶数位之和相等$ 情况的

减去 两种情况都符合的

注：开 long long 和 随时Mod

AC代码

#include<iostream>
#include<cstdio>
using namespace std;
const long long Mod=999983;
long long a[15];
long long n,ans1,ans2,ans3,f[1005][10005];
string s;
int main()

	scanf("%lld",&n);
	cin>>s;
	long long len=s.size();
	for(long long i=0;i<len;i++)
		a[i]=s[i]-48;
	f[0][0]=1ll;
	for(long long i=1;i<=n;i++)//dp
		for(long long j=0;j<=i*9;j++)
			for(long long k=0;k<len;k++)
				if(j>=a[k])
					f[i][j]=(f[i][j]+f[i-1][j-a[k]])%Mod;
	for(long long i=0;i<=n*9;i++)//求出分别符合两种情况的方案数
		ans1=(ans1+2*f[n][i]*f[n][i]%Mod)%Mod;
	for(long long i=0;i<=(n/2)*9;i++)//求两种情况都符合的方案数
		ans2=(ans2+f[n/2][i]*f[n/2][i]%Mod)%Mod;
	for(long long i=0;i<=(n-n/2)*9;i++)//同上一个for
		ans3=(ans3+f[n-n/2][i]*f[n-n/2][i]%Mod)%Mod;
	printf("%lld",((ans1-ans2*ans3%Mod)%Mod+Mod)%Mod);//求答案
	return 0;

谢谢