NC54584 小琛和他的学校

题目链接

题目链接

题意

小琛是一所学校的校长。
他的学校有n个校区（编号1~n），被n-1条双向道路连接，呈树形结构。
第i个校区共有Ai个学生。
第i天早上，所有的学生会沿最短路走到第i个校区参加活动，晚上再原路返回。
一个人通过第j条通道一次（即一人次），需要小琛支付wj的维护费用。
小琛想知道第n天结束之后，对于每一条通道，他总共需要支付多少费用。
对于100%的数据，1≤ n ≤ 200,000，1≤ A[i]≤ 10,000，1≤ w[i] ≤ 10,000。

思路

• 一条边可以把一个树分成两部分
• 该边上面的人数访问该边下面的子节点，(tot-sum[v])*point[v]
• 该边下面的人数访问改变上面的节点，sum[v]*(n-point[v]
• 则该边的费用为(sum[v]*(n-point[v])+(tot-sum[v])*point[v])*spe
• 注意开long long

参考代码

#include<bits/stdc++.h>
using namespace std;
long long  n,tot=0;
long long aa[200100],ans[200100],sum[200100],point[200100];
struct node
{
    long long  order,des,spe;
    node(long long  _order, long long  _des, long long  _spe)
    {
        order=_order;
        des=_des;
        spe=_spe;
    }
};
vector<node> road[200100];
void dfs(long long  u,long long  fa)
{
    sum[u]=aa[u];
    point[u]=1;
    for(int i=0; i<road[u].size(); i++)
    {
        long long  v=road[u][i].des;
        if(v!=fa)
        {
            long long  spe=road[u][i].spe;
            long long  order=road[u][i].order;
            dfs(v,u);
            sum[u]+=sum[v];
            point[u]+=point[v];
            ans[order]=(sum[v]*(n-point[v])+(tot-sum[v])*point[v])*spe;
        }
    }
}
int main()
{
    cin>>n;
    for(int i=1; i<=n; i++)
        {cin>>aa[i];tot+=aa[i];}
    for(int i=1; i<n; i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        road[u].push_back(node(i,v,w));
        road[v].push_back(node(i,u,w));
    }
    dfs(1,0);
    for(int i=1; i<n; i++)
    {
        cout<<ans[i]*2<<endl;
    }
}