当前位置: 代码迷 >> 综合 >> 【hdu 5692 Snacks】【dfs序】【线段树】【好题】
  详细解决方案

【hdu 5692 Snacks】【dfs序】【线段树】【好题】

热度:97   发布时间:2024-01-04 11:43:45.0

【链接】

http://acm.hdu.edu.cn/showproblem.php?pid=5692

【题意】

树上两种操作:

0 x y 将x的权重变为y

1 x 求出从0出发,经过x的路径的最距离和

【分析】

其实就是求一棵子树x里到0点路径权值和最大的点的那个权值和,因为是对一棵子树的所有值,所以容易想到用DFS序来处理。

dfs序处理出2*n的数组,每个节点出现两次,以x为根的子树被包含在x出现的两次的区间中,第二次出现表示不选当前这个点。

每个节点第一次出现的位置记录一个它的价值v[i],在第二次出现的位置记录一个-v[i]。那么子树中到根距离最多的点就是x那个区

间中前缀和最大的点。用线段树维护sum,mx(左前缀)

【代码】

#pragma comment(linker, "/STACK:1024000000,1024000000") 
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 2e5 + 6;
vector<int>v[maxn];
int in[maxn<<1], out[maxn<<1];
ll a[maxn<<1];
int cnt = 0;void dfs(int x,int f) {in[x] = ++cnt;for (int y : v[x])if(f!=y) {dfs(y, x);}out[x] = ++cnt;
}struct node {int l, r;ll sum, mx;
}t[maxn << 2];void build(int p, int l, int r) {t[p].l = l; t[p].r=r;if (l == r) { t[p].sum = t[p].mx = a[l]; return; }int mid = l + r >> 1;build(p << 1, l, mid);build(p << 1 | 1, mid + 1, r);t[p].sum = t[p << 1].sum + t[p << 1 | 1].sum;t[p].mx = max(t[p << 1].mx, t[p << 1].sum + t[p << 1 | 1].mx);
}void change(int p, int x, ll val) {if (t[p].l == t[p].r) {t[p].mx=t[p].sum = val;return;}int mid = t[p].l + t[p].r >> 1;if (x <= mid)change(p << 1, x, val);else change(p << 1 | 1, x, val);t[p].sum = t[p << 1].sum + t[p << 1 | 1].sum;t[p].mx = max(t[p << 1].mx, t[p << 1].sum + t[p << 1 | 1].mx);
}ll query(int p, int x, int y) {if (t[p].l==x&&t[p].r==y) return t[p].mx;int mid = t[p].l + t[p].r >> 1;if (y <= mid)return query(p << 1, x, y);else if (x > mid)return t[p<<1].sum+query(p << 1 | 1, x, y);else return max(query(p << 1, x, mid), t[p << 1].sum + query(p << 1 | 1, mid + 1, y));
}int main() {int t;scanf("%d", &t);int ca = 0;while (t--) {printf("Case #%d:\n", ++ca);int n, m;scanf("%d%d", &n, &m);for (int i = 1; i <= n; i++)v[i].clear();for (int i = 1; i < n; i++) {int x, y;scanf("%d%d", &x, &y); x++; y++;v[x].push_back(y);v[y].push_back(x);}cnt = 0;dfs(1,1);for (int i = 1; i <= n; i++) {ll y;scanf("%lld", &y);a[in[i]] = y; a[out[i]] = -y;}build(1, 1, 2 * n);while (m--) {int op, x;ll y;scanf("%d%d", &op, &x);x++;if (op == 0) {scanf("%lld", &y);change(1, in[x], y);change(1, out[x], -y);}else {printf("%lld\n", query(1,in[x],out[x]-1));//注意out[x]表示已经走出x以及x的子树了}}}
}