题目描述
小明对数学饱有兴趣,并且是个勤奋好学的学生,总是在课后留在教室向老师请教一些问题。一天他早晨骑车去上课,路上见到一个老伯正在修剪花花草草,顿时想到了一个有关修剪花卉的问题。于是当日课后,小明就向老师提出了这个问题:
一株奇怪的花卉,上面共连有NN朵花,共有N-1N−1条枝干将花儿连在一起,并且未修剪时每朵花都不是孤立的。每朵花都有一个“美丽指数”,该数越大说明这朵花越漂亮,也有“美丽指数”为负数的,说明这朵花看着都让人恶心。所谓“修剪”,意为:去掉其中的一条枝条,这样一株花就成了两株,扔掉其中一株。经过一系列“修剪“之后,还剩下最后一株花(也可能是一朵)。老师的任务就是:通过一系列“修剪”(也可以什么“修剪”都不进行),使剩下的那株(那朵)花卉上所有花朵的“美丽指数”之和最大。
老师想了一会儿,给出了正解。小明见问题被轻易攻破,相当不爽,于是又拿来问你。
输入输出格式
输入格式:
第一行一个整数N(1 ≤ N ≤ 16000)N(1≤N≤16000)。表示原始的那株花卉上共NN朵花。
第二行有NN个整数,第II个整数表示第II朵花的美丽指数。
接下来N-1N−1行每行两个整数a,ba,b,表示存在一条连接第aa 朵花和第bb朵花的枝条。
输出格式:
一个数,表示一系列“修剪”之后所能得到的“美丽指数”之和的最大值。保证绝对值不超过21474836472147483647。
输入输出样例
输入样例#1: 复制
7 -1 -1 -1 1 1 1 0 1 4 2 5 3 6 4 7 5 7 6 7
输出样例#1: 复制
3
说明
【数据规模与约定】
对于60\%60%的数据,有N≤1000N≤1000;
对于100\%100%的数据,有N≤16000N≤16000。
因为题目没有要求取的个数 所以直接开一维即可
#include<bits/stdc++.h> using namespace std; //input by bxd #define rep(i,a,b) for(int i=(a);i<=(b);i++) #define repp(i,a,b) for(int i=(a);i>=(b);--i) #define RI(n) scanf("%d",&(n)) #define RII(n,m) scanf("%d%d",&n,&m) #define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k) #define RS(s) scanf("%s",s); #define ll long long #define pb push_back #define REP(i,N) for(int i=0;i<(N);i++) #define CLR(A,v) memset(A,v,sizeof A) ////////////////////////////////// #define inf 0x3f3f3f3f const int N=16000+5; const int M=50005; int head[M],pos; struct Edge { int nex,to,v; }edge[M]; void add(int a,int b) { edge[++pos].nex=head[a]; head[a]=pos; edge[pos].to=b; } int n,m; int dp[N]; int maxx=-inf; void dfs(int u,int fa) { for(int i=head[u];i;i=edge[i].nex) { int v=edge[i].to; if(v==fa)continue; dfs(v,u); dp[u]=max(dp[u],dp[u]+dp[v]); } maxx=max(maxx,dp[u]); } int main() { RI(n); rep(i,1,n) RI(dp[i]); rep(i,1,n-1) { int a,b;RII(a,b); add(a,b);add(b,a); } dfs(1,0); cout<<maxx; return 0; }
原文地址:https://www.cnblogs.com/bxd123/p/10835900.html
时间: 2024-10-09 23:56:05