2023河南萌新联赛第（六）场：河南理工大学 K神运水晶

题目大意:
$给定一个无穷大的k叉树，从根节点向下走，每个分支的点权为0至k-1,求满足下列要求的概率,答案取模998244353$
$1.从根节点到该点所经过的所有点的点权和为n$
$2.从根节点到该点所经过的所有点权所构成的序列非递减。$

题目分析:
$可以设d{p}_{x,d}表示经过的点权和为x,上一个点的点权为d，继续走，合法的概率$
$首先不难想到转移式d{p}_{x,d}={\sum }_{i=d}^{k-1}d{p}_{x+i,i}\ast \frac{1}{k},$$答案便为d{p}_{0,0},但是观察转移式会发现，当d=0时，转移式便存在一个问题，会无限递归,我们不妨在d=0且x>0时不进行转移，最后再转移$

$当d=0,x=0时d{p}_{0,0}={\sum }_{i=0}^{k-1}d{p}_{0+i,i}\ast \frac{1}{k},将等式右边的d{p}_{0,0}\ast \frac{1}{k}移项至左边，得\frac{k-1}{k}d{p}_{0,0}=\frac{1}{k}\ast {\sum }_{i=1}^{k-1}d{p}_{0+i,i},故d{p}_{0,0}=\frac{1}{k-1}\ast {\sum }_{i=1}^{k-1}d{p}_{0+i,i}$

参考代码:

#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int M=207;
const LL Mod=998244353;

int n,k;

LL dp[M][M];

LL qpow(LL x,LL tim){
	LL ret=1,tmp=x;
	while(tim){
		if(tim&1) ret=ret*tmp%Mod;
		tmp=tmp*tmp%Mod;
		tim>>=1;
	}
	return ret;
}
int main(){
	ios::sync_with_stdio(false);
	cin>>n>>k;
	for(int i=0;i<k;i++){
		dp[n][i]=1;
	}
    for(int x=n;x>=0;x--){
    	for(int d=1;d<=k;d++){
    		for(int l=d;l<k;l++){
    			dp[x][d]=(dp[x][d]+dp[l+x][l]*qpow(k,Mod-2))%Mod;
			}
		}
	}
	for(int i=1;i<k;i++){
		dp[0][0]=(dp[0][0]+dp[i][i])%Mod;
	}
	dp[0][0]=(qpow(k-1,Mod-2)*dp[0][0])%Mod;
	cout<<dp[0][0]<<"\n";
}