BZOJ 1827 [Usaco2010 Mar]gather 奶牛大集会(树形DP)

【题目链接】 http://www.lydsy.com/JudgeOnline/problem.php?id=1827

【题目大意】

  给出一棵有点权和边权的树,
  请确定一个点,使得每个点到这个点的距离乘上该点乘积的总和最小。

【题解】

  定1为根,我们先计算当这个点为1的时候的值,同时记录每个子树的size
  之后我们再做一遍dfs,计算出每个点作为中心时候的答案
  我们发现当一个点从父节点往子节点移动的时候
  对于答案的变化是ans+=(size[1]-2*size[x])*len(fx->x)
  所以我们dfs计算,保留最小值即可。

【代码】

#include <cstdio>
#include <algorithm>
#include <vector>
#include <utility>
using namespace std;
const int N=100010;
typedef pair<int,int> P;
typedef long long LL;
vector<P> v[N];
int d[N],size[N],n;
LL ans,ans1;
void dfs(int x,int fx){
    for(int i=0;i<v[x].size();i++){
        int y=v[x][i].first,z=v[x][i].second;
        if(y!=fx){
            d[y]=d[x]+z;
            ans+=(LL)size[y]*d[y];
            dfs(y,x);
            size[x]+=size[y];
        }
    }
}
void dfs2(int x,int fx){
    for(int i=0;i<v[x].size();i++){
        int y=v[x][i].first,z=v[x][i].second;
        if(y!=fx){
            ans1+=(LL)(size[1]-2*size[y])*z;
            if(ans1<ans)ans=ans1;
            dfs2(y,x);
            ans1-=(LL)(size[1]-2*size[y])*z;
        }
    }
}
int main(){
    while(~scanf("%d",&n)){
        ans=ans1=0;
        for(int i=1;i<=n;i++)scanf("%d",&size[i]);
        for(int i=1;i<n;i++){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            v[a].push_back(P(b,c));
            v[b].push_back(P(a,c));
        }dfs(1,-1);ans1=ans;
        dfs2(1,-1);
        printf("%lld\n",ans);
    }return 0;
}
时间: 2024-10-08 22:41:09

BZOJ 1827 [Usaco2010 Mar]gather 奶牛大集会(树形DP)的相关文章

【树形DP/搜索】BZOJ 1827: [Usaco2010 Mar]gather 奶牛大集会

1827: [Usaco2010 Mar]gather 奶牛大集会 Time Limit: 1 Sec  Memory Limit: 64 MBSubmit: 793  Solved: 354[Submit][Status][Discuss] Description Bessie正在计划一年一度的奶牛大集会,来自全国各地的奶牛将来参加这一次集会.当然,她会选择最方便的地点来举办这次集会.每个奶牛居住在 N(1<=N<=100,000) 个农场中的一个,这些农场由N-1条道路连接,并且从任意一个

BZOJ 1827: [Usaco2010 Mar]gather 奶牛大集会( dp + dfs )

选取任意一个点为root , size[ x ] 表示以 x 为根的子树的奶牛数 , dp一次计算出size[ ] && 选 root 为集会地点的不方便程度 . 考虑集会地点由 x 点向它的子节点 son 转移 , 那么以 son 为集会地点比以 x 为集会地点要多 dist( x , son ) * ( tot - size[ x ] ) - dist( x , son ) * size[ x ] = dist( x , son ) * ( tot - 2 * size[ x ] )

BZOJ 1827: [Usaco2010 Mar]gather 奶牛大集会

Description 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个农场中的任意一个举行.另外

【BZOJ】1827: [Usaco2010 Mar]gather 奶牛大集会

[算法]树型DP [题解] 两遍DFS,第一次得到所有节点子树的路径和,第二次给出除了该子树外其它部分的路径和,时时计算答案. long long!!! #include<cstdio> #include<cstring> #include<algorithm> #include<cctype> #define ll long long using namespace std; const int maxn=100010; struct edge{int v

【BZOJ 1827】 [Usaco2010 Mar]gather 奶牛大集会

1827: [Usaco2010 Mar]gather 奶牛大集会 Time Limit: 1 Sec  Memory Limit: 64 MB Submit: 722  Solved: 314 [Submit][Status][Discuss] Description Bessie正在计划一年一度的奶牛大集会,来自全国各地的奶牛将来参加这一次集会.当然,她会选择最方便的地点来举办这次集会.每个奶牛居住在 N(1<=N<=100,000) 个农场中的一个,这些农场由N-1条道路连接,并且从任意

BZOJ1827 [Usaco2010 Mar]gather 奶牛大集会

题意:给定一棵树,求出树上的一点,使得树上的全部点到该点的距离之和最小. 思路:暴力显然是O(N^2)等死对吧. 我们首先将无根树转化为有根树,然后一边dfs求出f[i],size[i]. f[i]表示以i为根的子树中全部的点到i的距离之和,size[i]表示以i为根的子树的点数. 以下開始脑洞大开: 如今对于我们一開始的那个root,我们已经知道了答案.问题就是怎样高速的推知别的点作为根时的答案. 我们又一次进行一次dfs,当找到x时,我们用dp[fa[x]]+padis[x]*size[fa

BZOJ 1827 gather 奶牛大集会

Description 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个农场中的任意一个举行.另

BZOJ 1657: [Usaco2006 Mar]Mooo 奶牛的歌声( 单调栈 )

从左往右,从右往左 扫两遍,单调栈维护... ---------------------------------------------------------------------- #include<cstdio> #include<cstring> #include<algorithm> #include<stack> #include<iostream> #define rep( i , n ) for( int i = 0 ; i &

[BZOJ] 1657: [Usaco2006 Mar]Mooo 奶牛的歌声

1657: [Usaco2006 Mar]Mooo 奶牛的歌声 Time Limit: 5 Sec  Memory Limit: 64 MBSubmit: 858  Solved: 603[Submit][Status][Discuss] Description Farmer John's N (1 <= N <= 50,000) cows are standing in a very straight row and mooing. Each cow has a unique height