zoukankan      html  css  js  c++  java
  • CF1092F Tree with Maximum Cost

    洛咕

    伪双倍经验,这题是边权,但是做法一样

    题意:(n)个节点的树,每个节点有点权(a_i).定义(dist(x,y))(x)(y)的边数.选取一个点(v),使得(sum_{i=1}^ndist(i,v)*a_i)最大.

    分析:选取的那个点(v)不就是树的根?相当于要以每个点为根求一次贡献,然后取(max?)那不就是换根(dp?)转移手玩一下就行了.

    #include<iostream>
    #include<cstdio>
    #include<algorithm>
    #include<cstring>
    #include<cmath>
    #include<queue>
    #include<map>
    #include<set>
    #define ll long long
    using namespace std;
    inline int read(){
        int x=0,o=1;char ch=getchar();
        while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
        if(ch=='-')o=-1,ch=getchar();
        while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
        return x*o;
    }
    const int N=2e5+5;
    int n,a[N];ll ans,cost[N],f[N];
    int tot,head[N],nxt[N<<1],to[N<<1];
    inline void add(int a,int b){
    	nxt[++tot]=head[a];head[a]=tot;to[tot]=b;
    }
    inline void pre_dfs(int u,int fa){
    	cost[u]=a[u];
    	for(int i=head[u];i;i=nxt[i]){
    		int v=to[i];if(v==fa)continue;
    		pre_dfs(v,u);cost[u]+=cost[v];
    	}
    }
    inline void dfs(int u,int fa){
    	for(int i=head[u];i;i=nxt[i]){
    		int v=to[i];if(v==fa)continue;
    		dfs(v,u);f[u]+=f[v]+cost[v];
    	}
    }
    inline void dp(int u,int fa){
    	for(int i=head[u];i;i=nxt[i]){
    		int v=to[i];if(v==fa)continue;
    		f[v]=f[u]-cost[v]+cost[1]-cost[v];
    		ans=max(ans,f[v]);dp(v,u);
    	}
    }
    int main(){
    	n=read();for(int i=1;i<=n;++i)a[i]=read();
    	for(int i=1;i<n;++i){
    		int a=read(),b=read();
    		add(a,b);add(b,a);
    	}
    	pre_dfs(1,0);dfs(1,0);ans=f[1];dp(1,0);
    	printf("%lld
    ",ans);
        return 0;
    }
    
    
  • 相关阅读:
    纪念日给男(女)朋友的表白页面
    Vue组件的传值(非父子之间)
    express脚手架的安装和使用
    MongoDB数据库
    vuex状态
    MVVM框架的简单理解
    关于vue脚手架
    申请百度密钥
    svg
    微信小程序开发学习笔记
  • 原文地址:https://www.cnblogs.com/PPXppx/p/11845449.html
Copyright © 2011-2022 走看看