码迷,mamicode.com
首页 > 其他好文 > 详细

P2986 [USACO10MAR]伟大的奶牛聚集Great Cow Gat…

时间:2019-01-01 23:55:56      阅读:212      评论:0      收藏:0      [点我收藏+]

标签:pre   lld   continue   mes   自己   names   nod   没有   div   

  一道比较经典的题(蒟蒻的我没有一次切掉)……

  首先这道题要优化的原因就是数据范围,要是O(n2)  “  显然  ” 会超时。。。

  那么我们很不情愿不得已不开心地最终被迫考虑一下优化(说实话好多算法都是为优化而生的啊)。

  首先记录所有点(记得点上有可能有好几头牛)到根节点的路径之和。

  那么每一次考虑,将汇聚的点从父亲改到现在到的这个儿子,那么,以儿子节点作为根的子树上的所有点,就都少走了从儿子到父亲的距离,那么除这些点以外的点,它们都要多走从儿子到父亲的距离。

  那么我们只要一开始随意找一个点作为root,dfs求出距离和,再来一次dfs从下往上不断更新状态就好,每次取最优解就可以了。

  我第一次没A是因为在第二次dfs的时候忘了继续往上dfs(我对自己表示emmm……),第二次是因为没开long long……

  代码如下(请不要学习黑科技):

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define maxn 10000005
#define int long long
int num[maxn],dis[maxn],head[maxn];
struct node
{
    int to,nxt,w;
} e[maxn];
int cnt,tot,n,d,ans;
void add(int a,int b,int c)
{
    e[++cnt].to=b;
    e[cnt].nxt=head[a];
    head[a]=cnt;
    e[cnt].w=c;
}
void dfs(int u,int fa)
{
    for(int i=head[u];i;i=e[i].nxt)
    {
        int v=e[i].to;
        if(v==fa) continue;
        dfs(v,u);
        num[u]+=num[v];
        d+=e[i].w*num[v];
    }
}
void dp(int u,int fa)
{
    for(int i=head[u];i;i=e[i].nxt)
    {
        int v=e[i].to;
        if(v==fa) continue;
        dis[v]=dis[u]-num[v]*e[i].w+(tot-num[v])*e[i].w;
        ans=min(ans,dis[v]);
        dp(v,u);
    }
}
main()
{
    scanf("%lld",&n);
    for(int i=1;i<=n;i++)
    {
        int am;
        scanf("%lld",&am);
        num[i]=am;
        tot+=am;
    }
    for(int i=1;i<n;i++)
    {
        int a,b,c; 
        scanf("%lld%lld%lld",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
    }
    dfs(1,0);
    dis[1]=d;
    ans=d;
    dp(1,0);
    printf("%lld",ans);
    return 0;
}

 

 

P2986 [USACO10MAR]伟大的奶牛聚集Great Cow Gat…

标签:pre   lld   continue   mes   自己   names   nod   没有   div   

原文地址:https://www.cnblogs.com/popo-black-cat/p/10206416.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!