C - Rencontre Gym - 102798C题解：

题目链接：C - Rencontre Gym - 102798C
参考题解：
这份题解用的是直接计算边用啦多少次的方式

思路：

三个人独立选择，那么按照树上分别计算距离即可，然后最后将总距离除以方案数，就是每两个人的期望距离。

一共使用啦三种做法，归根到底是一道树形dp的题目，是一道换根dp的模板题，然后因为需要一些简单的计算所以可以转化为非换根题来做。
sum[y]=sum[x]+(n−2×size[y])×len(x,y）.注意n是全集，换根的写法就是因为这里一直没调过去。

1.换根dp写法：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5+1000;
bool vis[maxn][3]; // 0代表0和1的交，1代表1和2的交，2代表2和0的交。
ll Size[maxn][3];
int n;
int rong[3];
int to[maxn<<1],h[maxn<<1],idx,ne[maxn<<1];
double w[maxn<<1];
void add(int a,int b,int ww)
{
    to[idx]=b,w[idx]=ww,ne[idx]=h[a],h[a]=idx++;
}
double sum[maxn][3];
double ans[3];  // 0代表0和1的交，1代表1和2的交，2代表2和0的交。
void dfs2(int u,int fa)
{
    // cout<<u<<" ";
    for(int i=0;i<3;i++)
    {
        int left=(i+2)%3;
        int right=(i+1)%3;
        if(vis[u][i])
        {
            // cout<<"now"<<i<<endl;
            // cout<<sum[u][right]<<" "<<sum[u][left]<<endl;
            ans[i]+=sum[u][right];
            // ans[left]+=sum[u][left]/2.0;
        }
    }
    for(int i=h[u];~i;i=ne[i])
    {
        int v=to[i];
        if(v==fa)continue;
        for(int j=0;j<3;j++)
        {
            sum[v][j]=sum[u][j]+(double)w[i]*(Size[1][j]-2*Size[v][j]);
        }
        dfs2(v,u);
    }
}
void dfs1(int u,int fa,double len)
{
    for(int i=0;i<3;i++)
    {
        if(vis[u][i])
        {
            sum[u][i]=len;
            Size[u][i]=1;
        }
        else Size[u][i]=0,sum[u][i]=0.0;
    }
    for(int i=h[u];~i;i=ne[i])
    {
        int v=to[i];
        if(v==fa)continue;
        dfs1(v,u,len+w[i]);
        for(int j=0;j<3;j++)
        {
            sum[u][j]+=sum[v][j];
            Size[u][j]+=Size[v][j];
        }
    }
}

int main()
{
    memset(h,-1,sizeof(h));
    scanf("%d",&n);
    int m=n-1;
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
    }
    for(int i=0;i<3;i++)
    {
        scanf("%d",&rong[i]);
        int y=rong[i];
        while(y--)
        {
            int x;
            scanf("%d",&x);
            vis[x][i]=1;
        }
    }
    for(int i=0;i<3;i++)
    {
        ans[i]=0;
    }
    dfs1(1,-1,0.0);
    dfs2(1,-1);
    // cout<<ans[0]<< " " <<ans[1]<<" "<<ans[2]<<endl;
    double res=ans[0]/((double)rong[0]*rong[1])+ans[1]/((double)rong[1]*rong[2])+ans[2]/((double)rong[2]*rong[0]);
    res/=2.0;
    printf("%.7lf\n",res);
    return 0;
}

2.第二种写法

是将1号点作为所有两点经过的点，计算距离，然后 还是以1进行dfs枚举每一个点作为a，b需要经过的中间节点也就是lca节点，那么他们就不需要在往上走经过1号点，所以每次减下两次deep[u]即可。这种做法有个坑就是一个点上同时出现两种科学家时也需要减少deep[u]两次。
代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5+1000;
bool vis[maxn][3];
double deep[maxn];
ll Size[maxn][3];
int n;
int rong[3];
int to[maxn<<1],h[maxn<<1],idx,ne[maxn<<1];
double w[maxn<<1];
void add(int a,int b,int ww)
{
    to[idx]=b,w[idx]=ww,ne[idx]=h[a],h[a]=idx++;
}
double ans[3];
double res[3];  // 0代表0和1的交，1代表1和2的交，2代表2和0的交。
void dfs(int u,int fa)
{
    for(int j=0;j<3;j++)
    {
        if(vis[u][j]){
            for(int k=0;k<j;k++)
            {
                if(vis[u][k])
                {
                    if(k==(j+1)%3)
                    {
                        res[j]-=deep[u]*2;
                    }
                    if(k==(j-1+3)%3)
                    {
                        res[k]-=deep[u]*2;
                    }
                }
            }
            Size[u][j]=1;ans[j]+=deep[u];
        }
        else Size[u][j]=0;
    }
    for(int i=h[u];~i;i=ne[i])
    {
        int v=to[i];
        if(v==fa)continue;
        deep[v]=deep[u]+w[i];
        dfs(v,u);
        for(int j=0;j<3;j++)
        {
            int left=j,right=(j+1)%3;
            res[j]-=deep[u]*Size[u][left]*Size[v][right]*2;
            // cout<<j<<" "<<deep[u]*Size[u][left]*Size[v][right]*2<<endl;
            res[j]-=deep[u]*Size[u][right]*Size[v][left]*2;
            // cout << j << " " << deep[u] * Size[u][right] * Size[v][left] * 2<<endl;
        }
        for(int j=0;j<3;j++)
        {
            Size[u][j]+=Size[v][j];
        }
    }
}
int main()
{
    memset(h,-1,sizeof(h));
    scanf("%d",&n);
    int m=n-1;
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
    }
    for(int i=0;i<3;i++)
    {
        scanf("%d",&rong[i]);
        int y=rong[i];
        while(y--)
        {
            int x;
            scanf("%d",&x);
            vis[x][i]=1;
        }
    }
    deep[1]=0.0;
    for(int j=0;j<3;j++)
    {
        ans[j]=0.0;
        res[j]=0.0;
    }
    dfs(1,-1); 
    for(int j=0;j<3;j++)
    {
        // cout<<"asdfadfasdf";
        // cout<<ans[j]<<endl;
        int left=j,right=(j+1)%3;
        res[j]=ans[left]*(rong[right])+ans[right]*rong[left]+res[j];
    }
    double res1=res[0]/((double)rong[0]*rong[1])+res[1]/((double)rong[1]*rong[2])+res[2]/((double)rong[2]*rong[0]);
    res1/=2.0;
    printf("%.7lf\n",res1);
    return 0;
}

3.直接计算每条边利用多少次

基本比照开头博客中的写的，但是这种的思路最直接利于理解和操作：其中dp数组可以直接销掉，直接往ans[i]上加就行。
代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5+1000;
bool vis[maxn][3];
double dp[maxn][3];//0代表0和1的交，1代表1和2的交，2代表2和0的交。
ll Size[maxn][3];
int n;
int rong[3];
int to[maxn<<1],h[maxn<<1],idx,ne[maxn<<1];
double w[maxn<<1];
void add(int a,int b,int ww)
{
    to[idx]=b,w[idx]=ww,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int fa)
{
    for(int i=0;i<3;i++)
    {
        if(vis[u][i]) Size[u][i]=1;
        else Size[u][i]=0;
    }
    for(int i=h[u];~i;i=ne[i])
    {
        int v=to[i]; if(v==fa)continue;
        int ww=w[i];
        dfs(v,u);
        for(int j=0;j<3;j++)
        {
            int left=j,right=(j+1)%3;
            dp[u][j]+=(double)ww*Size[v][left]*(rong[right]-Size[v][right]);
            dp[u][j]+=(double)ww*Size[v][right]*(rong[left]-Size[v][left]);
            Size[u][j]+=Size[v][j];
        }
    }
}
int main()
{
    // ios::sync_with_stdio(false);
    memset(h,-1,sizeof(h));
    // cin>>n;
    scanf("%d",&n);
    int m=n-1;
    while(m--)
    {
        int a,b,c;
        // cin>>a>>b>>c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
    }
    for(int i=0;i<3;i++)
    {
        // cin>>rong[i];
        scanf("%d",&rong[i]);
        int y=rong[i];
        while(y--)
        {
            int x;
            scanf("%d",&x);
            // cin>>x;
            vis[x][i]=1;
        }
    }
    dfs(1,-1);
    double ans[3]={0.0,0.0,0.0};
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<3;j++)
        {
            ans[j]+=dp[i][j];
        }
    }
    ans[0]/=(double)rong[0]*rong[1];
    ans[1]/=(double)rong[1]*rong[2];
    ans[2]/=(double)rong[2]*rong[0];
    double res=ans[0]+ans[1]+ans[2];
    res/=2.0;
    printf("%.7lf\n",res);
    return 0;
}