“科林明伦杯”哈尔滨理工大学第十届程序设计竞赛(同步赛) 点对最大值 dp

链接:https://ac.nowcoder.com/acm/contest/5758/A
来源:牛客网

题目描述
这里有一棵树,每个点和每条边都存在一个价值。对于树上点对的价值,包括点对的起点和终点以及路径上边权值之和,不包括路径上其他点值。
求这颗树上最大的点对价值为多少。点对至少需要两个点。

输入描述:
输入t,代表有t组样例。每组样例第一行输入n,代表有n个点。接下来有n-1行,第i行有a[i]和b[i],代表a[i]节点与i节点存在一条边,且边的值为b[i],2<=i<=n。接下来一行有n个值c[j],代表每个节点j的价值,1<=j<=n。 (t<=10,n>1,n<1e6,a[i]<i,-500<=b[i]<=500,-500<=c[j]<=500)
输出描述:

输出最大的点对价值

示例1
输入
复制

1
4
1  -2
1   2
1   3
2 -2 3 4

输出
复制

12

推荐大佬博客:https://blog.nowcoder.net/n/dfe65f344b3b4058bc371c602c996175

  • d p [ u ] dp[u] dp[u]表示以 u u u为根节点的子树的最大价值
  • a n s = m a x ( a n s , d p [ u ] + d p [ v ] + e d g e u v ) ans = max(ans, dp[u]+dp[v]+edge_{uv}) ans=max(ans,dp[u]+dp[v]+edgeuv)
  • d p [ u ] = m a x ( d p [ u ] , d p [ v ] + e d g e u v ) dp[u]=max(dp[u], dp[v]+edge_{uv}) dp[u]=max(dp[u],dp[v]+edgeuv)
#define debug
#ifdef debug
#include <time.h>

#include "/home/majiao/mb.h"
#endif

#include <math.h>
#include <string.h>

#include <algorithm>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <vector>

#define MAXN ((int)1e5 + 7)
#define ll long long int
#define INF (0x7f7f7f7f)
#define fori(lef, rig) for (int i = lef; i <= rig; i++)
#define forj(lef, rig) for (int j = lef; j <= rig; j++)
#define fork(lef, rig) for (int k = lef; k <= rig; k++)
#define QAQ (0)

using namespace std;

#define show(x...)                             \
    do {                                       \
        cout << "\033[31;1m " << #x << " -> "; \
        err(x);                                \
    } while (0)

void err() { cout << "\033[39;0m" << endl; }
template <typename T, typename... A>
void err(T a, A... x) {
    cout << a << ' ';
    err(x...);
}

namespace FastIO {

char print_f[105];
void read() {}
void print() { putchar('\n'); }

template <typename T, typename... T2>
inline void read(T &x, T2 &... oth) {
    x = 0;
    char ch = getchar();
    ll f = 1;
    while (!isdigit(ch)) {
        if (ch == '-') f *= -1;
        ch = getchar();
    }
    while (isdigit(ch)) {
        x = x * 10 + ch - 48;
        ch = getchar();
    }
    x *= f;
    read(oth...);
}
template <typename T, typename... T2>
inline void print(T x, T2... oth) {
    ll p3 = -1;
    if (x < 0) putchar('-'), x = -x;
    do {
        print_f[++p3] = x % 10 + 48;
    } while (x /= 10);
    while (p3 >= 0) putchar(print_f[p3--]);
    putchar(' ');
    print(oth...);
}
}  // namespace FastIO
using FastIO::print;
using FastIO::read;

int n, m, Q, K, W[MAXN], dp[MAXN];

struct Edge {
	int v, w;
} ;
vector<Edge> G[MAXN];

/*
   --dp[u] 表示不包括 u 点点权的u到某个子节点的最大价值
   dp[u]表示以u为根的树的最大价值


   设v为u的某个子节点
   ans = max(ans, dp[u]+dp[v]+edge_uv)
   dp[u] = max(dp[u], dp[v]+edge_uv)
 */

int ans = 0;
void dfs(int u, int fa) {
	dp[u] = W[u];
	for(auto ed : G[u]) {
		int v = ed.v, tw = ed.w;
		if(v == fa) continue ;
		
		dfs(v, u); //先递归子树
		ans = max(ans, dp[u]+dp[v]+tw); //注意这里不能W[u]+dp[v]+tw

		dp[u] = max(dp[u], dp[v]+tw); //1.只要u点权   2.子状态v+uv边权  
	}
}

void init() {
	ans = -0x3f; //注意不要赋值成0
	for(int i=1; i<=n; i++) G[i].clear();
	memset(dp, -0x3f, sizeof(dp));
}

int main() {
#ifdef debug
    freopen("test", "r", stdin);
    clock_t stime = clock();
#endif
	read(Q);
	while(Q--) {
		read(n);
		init();
		for(int i=1, x, w; i<n; i++) {
			read(x, w);
			int u = x, v = i+1;
			G[u].push_back({v, w}), G[v].push_back({u, w});
		}
		for(int i=1; i<=n; i++) read(W[i]);
		dfs(1, -1);
		printf("%d\n", ans);
	}


#ifdef debug
    clock_t etime = clock();
    printf("rum time: %lf 秒\n", (double)(etime - stime) / CLOCKS_PER_SEC);
#endif
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值