[BZOJ1827] [Usaco 2010 Mar]gather 젖소 대집회(나무형dp+유지보수)

3280 단어 dpbzoj

제목 설명


전송문

문제풀이


우선 dp는 각 점자 트리의 권한과 이미 1을 집회점으로 하는 불편 지수를 구한다.나무의 부자 관계에 따라 차치로 지키는 사상으로 나무에서 각 점의 불편 지수를 유지하면 된다.

코드

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define LL long long
const int max_n=1e5+5;
const int max_e=max_n*2;
const LL INF=1e18;
int n,x,y;
LL z,val[max_n],len[max_n],size[max_n],cost,ans=INF;
int tot,point[max_n],next[max_e],v[max_e];LL c[max_e];

inline void add(int x,int y,LL z){++tot,next[tot]=point[x],point[x]=tot,v[tot]=y,c[tot]=z;}
inline void dfs(int x,int fa){
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        len[v[i]]=len[x]+c[i];
        cost+=len[v[i]]*val[v[i]];
        dfs(v[i],x);
        size[x]+=size[v[i]];
      }
}
inline void calc_ans(int x,int fa,LL cost){
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        LL t=cost-size[v[i]]*c[i]+(size[1]-size[v[i]])*c[i];
        ans=min(ans,t);
        calc_ans(v[i],x,t);
      }
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<=n;++i) scanf("%lld",&val[i]),size[i]=val[i];
    for (int i=1;i<n;++i) scanf("%d%d%lld",&x,&y,&z),add(x,y,z),add(y,x,z);
    dfs(1,0); ans=min(ans,cost);
    calc_ans(1,0,cost);
    printf("%lld
"
,ans); }

좋은 웹페이지 즐겨찾기