洛谷 P6293

显然有一个性质:

我们剖成的链都是递增递减

如果一条链不是递增或递减的,那我们一定可以断成两条链,使得贡献不减

fi,0/1 表示以 i 为根的子树,划分成若干条链,其中 i 所在的那条链是从上往下递减/递增的答案。

设当前在点 u,设 sum=vsonumax(fv,0,fv,1)

先对 fu,0/1 赋初值为 sum,表示 u 新开一条链。

然后转移枚举接上哪个儿子,fu,0=maxvsonu,avau{summax(fv,0,fv,1)+fv,0+auav}fu,1 的转移同理。

最终答案是 max(f1,0,f1,1)

时间复杂度 O(n)

Code:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 100005;
int n;
int a[N]; ll f[N][2];
int head[N], ver[N], nxt[N], cnt;

void add(int u, int v) {
	ver[++cnt] = v, nxt[cnt] = head[u], head[u] = cnt;
}

void chkmax(ll &a, ll b) { if (a < b) a = b; }

void dfs(int u) {
	ll sum = 0;
	for (int i = head[u]; i; i = nxt[i]) {
		int v = ver[i];
		dfs(v), sum += max(f[v][0], f[v][1]);
	}
	f[u][0] = f[u][1] = sum;
	for (int i = head[u]; i; i = nxt[i]) {
		int v = ver[i];
		if (a[u] <= a[v]) chkmax(f[u][1], f[v][1] + a[v] - a[u] + sum - max(f[v][0], f[v][1]));
		if (a[v] <= a[u]) chkmax(f[u][0], f[v][0] + a[u] - a[v] + sum - max(f[v][0], f[v][1]));
	}
}

int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);
	for (int i = 1, u, v; i < n; ++i) scanf("%d%d", &u, &v), add(u, v);
	dfs(1);
	printf("%lld", max(f[1][1], f[1][0]));
	return 0;
}
posted @   Kobe303  阅读(25)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话
点击右上角即可分享
微信分享提示