USACO 2010 Mar Gold 1.Great Cow Gathering 树形dp

来源:互联网 发布:新疆如何绕过网络管制 编辑:程序博客网 时间:2024/05/17 07:15

事实证明,人类的脑残是没有极限的 他的意思就是简单的要死的动规题 即使你已经搞出了动态转移方程也会只拿70分

原题:Bessie正在计划一年一度的奶牛大集会,来自全国各地的奶牛将来参加这一次集会。当然,她会选择最方便的地点来举办这次集会。每个奶牛居住在 N(1<=N<=100,000) 个农场中的一个,这些农场由N-1条道路连接,并且从任意一个农场都能够到达另外一个农场。道路i连接农场A_i和B_i(1 <= A_i <=N; 1 <= B_i <= N),长度为L_i(1 <= L_i <= 1,000)。集会可以在N个农场中的任意一个举行。另外,每个牛棚中居住者C_i(0 <= C_i <= 1,000)只奶牛。在选择集会的地点的时候,Bessie希望最大化方便的程度(也就是最小化不方便程度)。比如选择第X个农场作为集会地点,它的不方便程度是其它牛棚中每只奶牛去参加集会所走的路程之和,(比如,农场i到达农场X的距离是20,那么总路程就是C_i*20)。帮助Bessie找出最方便的地点来举行大集会。

首先 这一看就是一道简单的树形DP
对于百分之50的数据
我们可以枚举根,进行转移

f[i]iisize[i]

size[i]=size[j]+size[i]json[i]f[i]=(f[j]+size[j]val[j])json[i]

这就是正常的50分做法。

对于100分算法,我们可以发现,其实真正的耗时的地方在于对于每一个点都要进行O(n)的验证,显然这是十分缓慢的,容易发现,从一个节点到其他的儿子进行跟的转移,只需要将中间的边进行处理。

g[i]i

g[x]=g[fa]+(sum2size[x])(val)xson[fa]

所以两次DFS就可以了
下面是代码

#include <stdio.h>#include <string.h>#include <iostream>#include <algorithm>#include <stdlib.h>#include <time.h>#define min(a,b) ((a)<(b)?(a):(b))#define N 100000+5#define M 200000+5using namespace std;long long ans=0x3f3f3f3f3fll;int head[N],n,a[N],size[N];long long f[N];bool vis[N];long long g[N];struct graph{    int next,to;    long long val;    graph (){}    graph (int _next,int _to,int _val)    :next(_next),to(_to),val(_val){}}edge[M];struct point{    int x;    int val;    bool operator<(const point &z)const    {        return val<z.val;    }}P[N];inline void add(int x,int y,int z){    static int cnt = 0;    edge[++cnt]=graph(head[x],y,z);    head[x]=cnt;    edge[++cnt]=graph(head[y],x,z);    head[y]=cnt;}inline int read(){    int x = 0, f = 1; char ch = getchar();    while (ch < '0' || ch > '9') { if (ch == '-') f = -1; ch = getchar(); }    while (ch >= '0' && ch <= '9') { x = x * 10 + ch - '0'; ch = getchar(); }    return x * f;}bool flag=false;long long sum;void DFS(int x,int fa){    size[x]=a[x];    f[x]=0;    for(int i=head[x];i;i=edge[i].next)        if(edge[i].to^fa)        {            DFS(edge[i].to,x);            size[x]+=size[edge[i].to];            f[x]+=(long long)size[edge[i].to]*(long long)edge[i].val+f[edge[i].to];        }}void reDFS(int x,int fa,int val){    if(x!=1)        g[x]=g[fa]+(sum-2*size[x])*(val);    else        g[x]=f[x];    for(int i=head[x];i;i=edge[i].next)        if(edge[i].to^fa)            reDFS(edge[i].to,x,edge[i].val);}int main(){    freopen("meet.in", "r", stdin);    freopen("meet.out", "w", stdout);    cin>>n;    for(int i=1;i<=n;++i)        a[i]=read(),P[i].x=i,P[i].val=a[i];    for(int i=1;i<n;++i)    {        int tmp1=read(),tmp2=read(),tmp3=read();        add(tmp1,tmp2,tmp3);    }    DFS(1,-1);    ans=f[1];    sum=size[1];    reDFS(1,-1,0);    sort(g+1,g+n+1);    cout<<g[1];}

美化博客环境

0 0