【ZROI8102 D2T2】配对

题目描述

给定一棵 n 个点的边有长度的无根树，小 A 的班里一共有 m 个男生和 m 个女生，他们各自会等概率出现在树上 n 个点中的某一个，(注意同一个点上可能会出现多个人)。

然后小 A 会将他们配对成 m 对男女，设$dis[i]$ 是第 i 对男女在树上的最短距离，小 A 会选择使得 $\sum^m_{i=1} dis[i]$ 最大的配对方案。

现在小 A 想知道，他选择的配对方案中 $\sum^m_{i=1} dis[i]$ 的期望，由于答案可能过大，你只需要输出答案$*n^{2m}$ 后对 1e9+7 取模的值，这个值显然是一个整数

样例输入1

3 1
1 2 1
2 3 1

样例输出1

8

样例输入2

2 2
1 2 1

样例输出2

20

数据范围

对于 20% 的数据，有 1≤n,m≤4，1≤n,m≤4
对于 50% 的数据，有 1≤n,m≤300，1≤n,m≤300
另有 20% 的数据，满足最多只有一条边的长度 >0
对于 100% 的数据，有 1≤n,m≤2500，1≤n,m≤2500 ，0≤w≤1e4

自述心路历程+题解

考试时看到这道题时，我的内心是崩溃的。最讨厌的就是数学和树上问题，结果这场考试全都遇到了orz
乱想了两个多小时，感觉毫无思路，就去拿暴力分。看一眼数据范围，感觉前20%和特殊的20%是可以拿的，其余的好像没思路。。。
敲完暴力水过样例，懒得检查，走人。。。
然后这道题我就爆零了啊啊啊啊啊。。。DFS太久没写，两个都写错了，原本能得40分的。。。

好的，我们来讲一讲正解
出题人吨老师表示，可以从特殊数据中找到思路。。。
我们来观察一下如何分组能使答案最大。如图。。。

我们可以将紫色边下面的两点配对，剩下两点配对，但这显然不是最优解。我们发现，最优解刚好多了紫色边权值的两倍。。。。
于是。。。进行大胆的猜想吧（“证明是留给出题人的。”）。一个图内，一种解比另一种更优，较劣的解所用到的每一条边，较优解都是用了的。。简而言之，对于每一条边，做一遍处理20%特殊数据的算法，是没有问题的（因为贪心原则成立嘛，能走则走）。。
dfs每个点。。。假设某一条边的儿子房为子树的节点个数为si，那么这条边对答案的贡献为：
$\sum_{i=1}^m \sum_{j=1}^m Min(i+j,2m-i-j)*(^m_i)*(^m_j)*si^{i+j}*(n-si)^{2m-i-j}*v$
i和j枚举某一端男女生数量。。。
时间复杂度$O(nm^2)$，要凉。。。
令a=i+j，我们可以发现，对于每个儿子节点，它对各个a的值的影响是可以$O(n)$计算的。。。所以我们加一个数组A[]，在dfs时存储贡献，上边那个式子改到主函数去计算，将最后3项改为$*A[i+j]$，就行了。
时间复杂度$O(nm+m^2)$，具体实现见代码。。。

代码

#include <cstdio>
#include <iostream>
#include <cmath>
#define ll long long
using namespace std;
char buf[1<<20],*p1,*p2;
#define GC (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<20,stdin),p1==p2)?0:*p1++)
inline int R()
{
    int p=0;
    char o=GC;
    while(o<48||o>57)o=GC;
    while(o<=57&&o>=48){
        p=(p<<3)+(p<<1)+o-48;
        o=GC;
    }
    return p;
}
const int Q=5005;
int las[Q],e[Q],nn[Q],v[Q],inc=0;
void add(int x,int y,int z)
{
    e[++inc]=y;
    v[inc]=z;
    nn[inc]=las[x];
    las[x]=inc;
}
const int MOD=1e9+7;
inline int add(int a,int b)
{a+=b;return a>=MOD?a-MOD:a;}
inline int ch(int a,int b)
{return 1LL*a*b%MOD;}
int ans=0,n,m,A[5555];
int owo[2555][5555],c[2555][2555];
int dfs1(int x,int fa,int fv)
{
    int si=1;
    for(register int t=las[x];t;t=nn[t]){
        int y=e[t];
        if(y!=fa)si+=dfs1(y,x,v[t]);
    }
    for(int a=0;a<=(m<<1);a++)A[a]=add(A[a],ch(ch(owo[si][a],owo[n-si][2*m-a]),fv));
    return si;
}
void solved()
{
    int i,j;
    for(i=0;i<=n;i++)owo[i][0]=1;
    for(i=1;i<=n;i++)
        for(j=1;j<=(m<<1);j++)
            owo[i][j]=ch(owo[i][j-1],i);
    for(i=0;i<=m;i++){
        c[i][0]=1;
        for(j=1;j<=i;j++)
            c[i][j]=add(c[i-1][j-1],c[i-1][j]);
    }
    dfs1(1,0,0);
    for(register int i=0;i<=m;i++)
        for(register int j=0;j<=m;j++)
            ans=add(ans,ch(ch(min(i+j,2*m-i-j),A[i+j]),ch(c[m][i],c[m][j])));
    printf("%d",ans);
}
int main()
{
    int i,x,y,z;
    n=R(),m=R();
    for(i=1;i<n;i++)
        x=R(),y=R(),z=R(),add(x,y,z),add(y,x,z);
    solved();
    return 0;
}