CF1657E

Star MST

这个场一大堆 DP,我还都不会做/kk。。。

首先我们需要知道一个性质,对于一张菊花图,要想满足菊花图的最小的生成树,其他不与 1 相连的边 $(u,v)$ 必须满足:

$$\begin{cases}
w(1,u)+w(u,v)\ge w(1,v)+w(1,u)
\\
w(1,v)+w(u,v)\ge w(1,u)+w(1,v)
\end{cases}$$

即满足不管替换哪一条路径 $w(u,v)$ 都不会更优,上面的方程解一下就是 $w(u,v)\ge \max\{w(1,u),w(1,v)\}$。容易明白这个条件是充要的。

那么现在着手 DP。

定义状态 $f(i,j)$,表示已经定好了前 $i$ 个点和它们的连边,其中与 1 相连的边权最大为 $j$,一共有多少种方案。

那么我们考虑转移,我们的一个阶段显然是选择 $p$ 个点与 1 的连边边权是 $j$,不妨直接选择 $p$ 个点与 1 的连边权小于 $j$,并考虑从 $f(p,x)$($x<j$)转移。

很容易想到我们选择点的方案数是 $\binom{n-p}{i-p}$ 的(从剩下的 $n-p$ 个点选 $i-p$ 个点与 1 相连边权为 $j$)。

很容易想到连边的边数是 $t=(i-p)(p-1)+(i-p)(i-p-1)/2=(i-p)(i+p-3)/2$ 的(即这 $i-p$ 个点与除了 1 的前面的点的连边边权大于 $j$,这 $i-p$ 个点之间相互连边的边权大于 $j$)。

那么我们连边的方案数也就很显然了:$(k-j+1)^t$。

然后我们就有了转移方程:

$$f(i,j)=\sum_{p=1}^{i-1}\left[\left(\sum_{x=0}^{j-1}f(p,x)\right)\binom{n-p}{i-p}(k-j+1)^{(i+p-3)(i-p)/2}\right]$$

这个 $\sum_{x=0}^{j-1}f(p,x)$ 可以预处理,$\binom{n-p}{i-p}$ 可以预处理,$(k-j+1)^{(i+p-3)(i-p)/2}$ 可以快速幂或者光速幂。

初始状态 $f(1,0)=1$。

最后的答案是 $\sum_{i=1}^kf(n,i)$。

时间复杂度 $O(n^2k\log n)$。

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#include<iostream>
#include<cstdio>
#define ll long long
using namespace std;
namespace Ehnaev{
inline ll read() {
ll ret=0,f=1;char ch=getchar();
while(ch<48||ch>57) {if(ch==45) f=-f;ch=getchar();}
while(ch>=48&&ch<=57) {ret=(ret<<3)+(ret<<1)+ch-48;ch=getchar();}
return ret*f;
}
inline void write(ll x) {
static char buf[22];static ll len=-1;
if(x>=0) {do{buf[++len]=x%10+48;x/=10;}while(x);}
else {putchar(45);do{buf[++len]=-(x%10)+48;x/=10;}while(x);}
while(len>=0) putchar(buf[len--]);
}
}using Ehnaev::read;using Ehnaev::write;

const ll N=3e2,mo=998244353;

ll n,k;
ll comb[N+5][N+5],c[N+5][N+5],f[N+5][N+5];

inline ll Pow(ll b,ll p) {
ll res=1;while(p) {if(p&1) res=res*b%mo;b=b*b%mo;p>>=1;}return res;
}

inline void Init() {
for(ll i=0;i<=n||i<=k;i++) {
comb[i][0]=1;
for(ll j=1;j<=i;j++) {
comb[i][j]=(comb[i-1][j]+comb[i-1][j-1])%mo;
}
}
}

int main() {

n=read();k=read();

Init();

f[1][0]=1;
for(ll i=0;i<=k;i++) c[1][i]=1;
for(ll i=2;i<=n;i++) {
for(ll j=1;j<=k;j++) {
for(ll p=1;p<i;p++) {
ll tmp=c[p][j-1];
tmp=tmp*comb[n-p][i-p]%mo;
ll tmp1=k-j+1,tmp2=(i+p-3)*(i-p)/2;
tmp1=Pow(tmp1,tmp2)%mo;
tmp=tmp*tmp1%mo;
f[i][j]=(f[i][j]+tmp)%mo;
c[i][j]=(c[i][j-1]+f[i][j])%mo;
}
}
}

ll ans=0;
for(ll i=1;i<=k;i++) {ans=(ans+f[n][i])%mo;}

write(ans);

return 0;
}