洛谷 P6293
显然有一个性质:
我们剖成的链都是递增或递减的
如果一条链不是递增或递减的,那我们一定可以断成两条链,使得贡献不减
设 fi,0/1 表示以 i 为根的子树,划分成若干条链,其中 i 所在的那条链是从上往下递减/递增的答案。
设当前在点 u,设 sum=∑v∈sonumax(fv,0,fv,1)。
先对 fu,0/1 赋初值为 sum,表示 u 新开一条链。
然后转移枚举接上哪个儿子,fu,0=maxv∈sonu,av≤au{sum−max(fv,0,fv,1)+fv,0+au−av},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;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话