【NOI2017模拟6.29】呵呵

题目

$1\le n\le2000$

解法

考虑一个特定形态的树的贡献，设点i的度数为d[i]，那么答案就是：

$$\sum(\prod w_i^{d_i}\cdot d_i)$$
考虑prufer序，一个度数为d[i]的点出现的次数是d[i]-1，那么就可以得到一个很显然的DP，f[i][j]表示前i个点的度数为i+j：
$$f_{i,j}=\sum_{d=0}^{j}f_{i-1,j-d}\cdot \begin{pmatrix}d\\n-2-(j-d)\end{pmatrix}\cdot w_i^{d+1}\cdot (d+1)$$
这个DP是 $O(n^3)$的
即答案要求的是这样一个东西：
$$\text{Ans}=\sum_{d_1+d_2+\cdots+d_n=n-2}\frac{(n-2)!}{\prod d_i!}\cdot\prod w_i^{d_i+1}\cdot (d_i+1)\\ \text{Ans}=\prod w_i\cdot(n-2)!\sum_{d_1+d_2+\cdots+d_n=n-2}\frac{1}{\prod d_i!}\cdot\prod w_i^{d_i}\cdot (d_i+1)$$
然后考虑将 $(d_i+1)$拆开，对于每个排列 $1\le p_1<p_2<\cdots p_k\le n$考虑 $\prod_{i=1}^k d_{p_i}$的贡献：
$$\sum_{d_1+d_2+\cdots+d_n=n-2}\frac{1}{\prod d_i!}\cdot\prod w_i^{d_i}\cdot\prod d_{p_i}$$
将 $d_{p_i}$与 $\frac{1}{d_{p_i}!}$中的 $\frac{1}{d_{p_i}}$消掉，得到：
$$\prod w_{p_i}(\sum_{d_1+d_2+\cdots+d_n=n-2-k}\prod\frac{w_i^{d_i}}{d_i!})$$
这样有什么好处呢？可以看到后面的式子变成了这样的形式： $\frac{x^i}{i!}$
考虑生成函数：
$$f_i(x)=e^{w_i\cdot x}=\sum_{i=0}^\infty\frac{{w_i}^i}{i!}\cdot x^i$$
那么前面的式子可以变成：
$$\prod w_{p_i}([x^ {n-2-k}]\prod f_i(x))$$
有：
$$\prod w_{p_i}([x^ {n-2-k}]e^{(\sum w_i)\cdot x})$$
$$\prod w_{p_i}(\frac{(\sum w_i)^{n-2-k}}{(n-2-k)!})$$
综上所述有：
$$\text{Ans}=\prod w_i\cdot(n-2)!\cdot\sum_{k=0}^{n-2}(\prod _{1\le p_1<p_2<\cdots<p_k\le n}w_{p_i})\frac{(\sum w_i)^{n-2-k}}{(n-2-k)!}$$
中间那个可以直接 $O(n^2)$的DP处理（其实FFT也是可以的）
所以总时间复杂度是 $O(n^2)$的

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<set>
#include<bitset>
#include<map>

#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)

using namespace std;

typedef long long LL;
typedef double db;

int get(){
    char ch;
    while(ch=getchar(),(ch<'0'||ch>'9')&&ch!='-');
    if (ch=='-'){
        int s=0;
        while(ch=getchar(),ch>='0'&&ch<='9')s=s*10+ch-'0';
        return -s;
    }
    int s=ch-'0';
    while(ch=getchar(),ch>='0'&&ch<='9')s=s*10+ch-'0';
    return s;
}

const int N = 2010;
const int mo = 1e+9+7;

int n;
LL a[N],js[N],ny[N],mi[N];
LL f[N];

LL quickmi(LL x,LL tim){
    LL ans=1;
    for(;tim;tim/=2,x=x*x%mo)
    if (tim&1)ans=ans*x%mo;
    return ans;
}

int main(){
    freopen("hehe.in","r",stdin);
    freopen("hehe.out","w",stdout);
    n=get();
    fo(i,1,n)a[i]=get();
    js[0]=1;
    fo(i,1,n)js[i]=js[i-1]*i%mo;
    ny[n]=quickmi(js[n],mo-2);
    fd(i,n-1,0)ny[i]=ny[i+1]*(i+1)%mo;
    mi[0]=1;
    fo(i,1,n)mi[1]=(mi[1]+a[i])%mo;
    fo(i,2,n)mi[i]=mi[i-1]*mi[1]%mo;
    f[0]=1;
    fo(i,1,n)
        fd(j,n,0)
        f[j+1]=(f[j+1]+f[j]*a[i]%mo)%mo;
    LL ans=0;
    fo(k,0,n-2)
    ans=(ans+f[k]*mi[n-2-k]%mo*ny[n-2-k]%mo)%mo;
    ans=ans*js[n-2]%mo;
    fo(i,1,n)ans=ans*a[i]%mo;
    printf("%lld\n",ans);
    fclose(stdin);
    fclose(stdout);
    return 0;
}