【LOJ #6094. 「Codeforces Round #418」归乡迷途】
题目大意:
传送门。
lca说的很明白就不重复了。
题解:
先膜一发lca。
大体读完题以后我们可以知道对于第i个节点最短路一定是连向1到i-1中的某个点。
然后我们考虑将到1距离(这里及以下均是最短路)相等的点放到同一层,显然最后的总体结构应该是一棵树,再加上在同一层/深度相同的点之间连接一些边的并。
很容易发现一层的转移只需要知道上一层度数为2/3的个数,以及当前层之间的相互连接。
先说一下$n^5$做法。
设$f_{i,p1,p2,u_1,u_2}$表示插入第i个点时,上一层度数为2/3还剩$p1/p2$,这一层度数为2/3还剩$u1/u2$,很显然这个dp转移是比较好(e)想(xin)的……
这里就不写了……
然后是lca大神的$n^3$。
我们发现在事实上,我们并不需要在向上连边的同时和同层的连边,我们只需要在把这一层向上连边结束后计算一下当前层状态到最后连边状态的方案即可。
设$f_{i,j}$表示前i个点中,最后j个点为最后1层,且前i-j个点度数已经满足要求,但最后j个点只考虑向上一层如何连边。
设$g_{i,j,k}$来表示新的一层插了i个的情况下,上一层有j个点度数为2,k个点度数为3。
所有有$f_{i,j}=\sum_k f_{i-j,k}*g_{j,c0,c1}$。其中c0/1表示上一层要求度数为2/3的个数。
假设我们知道了$g$那么这个DP就是$n^3$。
1.显然$g_{0,0,0}=1$。
2.考虑$g$当新的一层没有节点时,即$i==0$,显然此时状态含义为当前层最后有度数2的点为$j$,3的点为$k$的方案数。此时,当$j==0$时,因为不会有度数为2的点,所以这次一定是一堆仍未在同层连接的点和最后一个添加点连成一个环。由于我们此时需要知道这个环的大小,所以我们得到:
$g_{0,0,k}=\sum_{l=2}^{k-1}g_{0,0,k-l-1}C_{k-1}^lN_{l+1}$。
这个表示最后状态里有$k$个三个度数的点由$k-l-1$个点的递推。(应该很好理解)
$N_i$表示项链数,$i>=2$,即圆排列除以2。
3.考虑当$j!=0$时,我们最终状态可以由$g_{0,j-2/j,k/k-1}$递推而来,当我们这次选择连接一个实际度数为1的点时,我们会产生两个度数为2的点,当选择实际度数为2的点时,我们会得到的一个度数为3和一个度数为2的点同时失去一个度数为2的点。同时由于之前选取集合的不同,此时我们得到:
$g_{0,j,k}=(j-1)g_{0,j-2,k}+k*g_{0,j,k-1}$。
4.当$i!=0$时,此时$g$回归本来含义。我们很轻易地得出此时DP方程:
$g_{i,j,k}=j*g_{i-1,j-1,k}+k*g_{i-1,j+1,k-1}$
至此问题解决了。Orz lca大神犇。
代码:
1 #include "bits/stdc++.h" 2 3 using namespace std; 4 5 inline int read() { 6 int s=0,k=1;char ch=getchar(); 7 while (ch<'0'|ch>'9') ch=='-'?k=-1:0,ch=getchar(); 8 while (ch>47&ch<='9') s=s*10+(ch^48),ch=getchar(); 9 return s*k; 10 } 11 12 const int N=405,mod=1e9+7; 13 typedef long long ll; 14 15 int n,d[N]; 16 int f[N][N],g[N][N][N],fac[N],inv[N]; 17 18 #define C(a,b) (1ll*fac[a]*inv[b]%mod*inv[a-b]%mod) 19 20 inline void init() { 21 int i,j,k,s; 22 fac[0]=inv[1]=1; 23 inv[0]=1; 24 for (i=2;i<=n;++i) inv[i]=1ll*inv[mod%i]*(mod-mod/i)%mod; 25 for (i=1;i<=n;++i) fac[i]=1ll*fac[i-1]*i%mod; 26 for (i=1;i<=n;++i) inv[i]=1ll*inv[i-1]*inv[i]%mod; 27 g[0][0][0]=1; 28 for (i=0;i<=n;++i) 29 for (j=0;j<=n-i;++j) if(i||j){ 30 if (!i){ 31 for (k=2;k<j;++k)g[0][i][j]=(g[0][i][j]+C(j-1,k)*g[0][i][j-k-1]%mod*fac[k]%mod*inv[2])%mod; 32 }else { 33 if(i>=2)g[0][i][j]=(g[0][i][j]+(i-1ll)*g[0][i-2][j])%mod; 34 if(j>=1)g[0][i][j]=(g[0][i][j]+(ll)j*g[0][i-1][j-1])%mod; 35 } 36 } 37 38 for (i=1;i<n;++i) 39 for (s=1;s<n-i;++s) 40 for (k=0;k<=s;++k) { 41 j=s-k; 42 if (j) g[i][j][k]=(g[i][j][k]+(ll)j*g[i-1][j-1][k])%mod; 43 if (k) g[i][j][k]=(g[i][j][k]+(ll)k*g[i-1][j+1][k-1])%mod; 44 } 45 } 46 47 int main(){ 48 n=read(); 49 register int i,j,k; 50 int c[2]; 51 for (i=1;i<=n;++i) d[i]=read(); 52 init(); 53 f[d[1]+1][d[1]]=1; 54 for (i=d[1]+2;i<=n;++i) 55 for (j=1;j<=i-d[1]-1;++j) { 56 c[0]=c[1]=0; 57 for (k=1;k<i-j;++k) { 58 ++c[d[i-j-k+1]==3]; 59 f[i][j]=(f[i][j]+(ll)g[j][c[0]][c[1]]*f[i-j][k])%mod; 60 } 61 } 62 int ans=0; 63 c[0]=c[1]=0; 64 for (i=1;i<n;++i) { 65 ++c[d[n-i+1]==3]; 66 ans=(ans+(ll)f[n][i]*g[0][c[0]][c[1]])%mod; 67 } 68 printf("%d\n",ans); 69 }