“蔚来杯“2022牛客暑期多校训练营7 Melborp Elcissalc

原题题面：https://ac.nowcoder.com/acm/contest/33192/J

题目大意：已知$k(0<k<65)$,对于每个仅包含$[0,k-1]$区间内的整数的数组，定义其优美度为非空连续子数组之和为$k$的倍数的数量。求有多少长度为$n$的数组，其优美度为$t$ ,答案对$998224353$取模

题目分析：
因为是求连续子数组之和，我们应该会情不自禁地想起前缀和
$su{m}_i=su{m}_{i-1}+a_i$

因为题目只需知道其是否为$k$的倍数,并不关心其值，所以不妨改动一下：
$su{m}_i=(su{m}_{i-1}+a_i)$ % $k$

显然，只要保证$（su{m}_y-su{m}_x）=0，即（su{m}_x=su{m}_y）$,就可说明在$[x+1,y]$内的值的和为$k$的倍数

为了统计，不妨设$numbe{r}_i为有多少su{m}_x=i$

两两搭配为$k$的倍数的方案数为$C_{numbe{r}_i}^2$

以上便是做这道题的基础

正题：
读完题目，应该敏锐地想到动态规划

不妨设$d{p}_{x,y,z}$表示使用$[0,x]$区间内的数，填充了$y$个位置，优美度为$z$的方案数

枚举用$i:s$次时的贡献，所以可得
$$d{p}_{x,y,z}=\sum _{s=0}^{y}d{p}_{x-1,j-s,z-C_s^2}\ast C_y^s$$

总结：
以上代码总时间复杂度为$O(2\ast n+lo{g}_n+n^5)\approx 2^{30}+200$ 勉强可通过
空间复杂度为$O(n^4)$

Tip: 这道题时间很吃紧，代码要比较优秀，补题时超时了好几次

代码：

#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int MXN=70;
const LL mod=998244353;

LL fac[MXN],ifac[MXN];
LL dp[MXN][MXN][MXN*MXN];
LL c[MXN][MXN];

LL fast(LL x,LL y){
	LL res=1,tmp=x;
	while(y){
		if(y&1) res=res*tmp%mod;
		tmp=tmp*tmp%mod;
		y>>=1;
	}
	return res;
}

void Init(){//预处理阶乘
	fac[0]=1;
	for(int i=1;i<=66;i++){
		fac[i]=fac[i-1]*(LL)i%mod;
	}
	ifac[66]=fast(fac[66],mod-2);
	for(int i=65;i>=0;i--){
		ifac[i]=ifac[i+1]*(LL)(i+1)%mod;
	}
}

LL C(int n,int m){//求组合数
    if(n<m) return 0;
	if(n==m) return 1;
    if(c[n][m]) return c[n][m]; //记忆化，不然会超时
	return c[n][m]=((fac[n]*ifac[m])%mod*ifac[n-m])%mod;
}

int main(){
	int n,k,t,i,j,d,s;
	LL tmpp;
	Init();
	scanf("%d%d%d",&n,&k,&t);
    for(i=1;i<=n;i++){
    	dp[0][i][i*(i+1)/2]=1;
    }
    for(i=0;i<k;i++){
    	dp[i][1][0]=i; dp[i][1][1]=1; dp[i][0][0]=1;
    }
    for(i=1;i<k;i++){
    	for(j=2;j<=n;j++){
    		for(d=0;d<=min(t,j*(j+1)/2);d++){
    			for(s=0;s<=j;s++){
    				tmpp=C(s,2);
    				if(tmpp>d) continue;
    				dp[i][j][d]=(dp[i][j][d]+dp[i-1][j-s][d-tmpp]*C(j,s)%mod)%mod;
    			}
    		}
    	}
    }
   	printf("%lld\n",dp[k-1][n][t]);
}