首页 > 代码库 > POJ 2763 Housewife Wind LCA转RMQ+时间戳+线段树成段更新
POJ 2763 Housewife Wind LCA转RMQ+时间戳+线段树成段更新
题目来源:POJ 2763 Housewife Wind
题意:给你一棵树 2种操作0 x 求当前点到x的最短路 然后当前的位置为x; 1 i x 将第i条边的权值置为x
思路:树上两点u, v距离为d[u]+d[v]-2*d[LCA(u,v)] 现在d数组是变化的 对应每一条边的变化 他修改的是一个区间 用时间戳处理每个点管辖的区域 然后用线段树修改 线段树的叶子节点村的是根到每一个点的距离 求最近公共祖先没差别 只是堕落用线段树维护d数组
各种错误 4个小时 伤不起
#include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int maxn = 200010; struct edge { int u, v, w, next; }edges[maxn*2], e[maxn]; int E[maxn*2], H[maxn*2], I[maxn*2], L[maxn], R[maxn]; int dp[maxn*2][40]; int cnt, clock, dfn; int first[maxn]; int a[maxn<<2]; int b[maxn]; int add[maxn<<2]; int degree[maxn]; int vis[maxn]; void AddEdge(int u, int v, int w) { edges[cnt].u = u; edges[cnt].v = v; edges[cnt].w = w; edges[cnt].next = first[u]; first[u] = cnt++; edges[cnt].u = v; edges[cnt].v = u; edges[cnt].w = w; edges[cnt].next = first[v]; first[v] = cnt++; } void dfs(int u, int fa, int dep) { E[++clock] = u; H[clock] = dep; I[u] = clock; L[u] = ++dfn; b[dfn] = u; for(int i = first[u]; i != -1; i = edges[i].next) { int v = edges[i].v; if(v == fa) continue; if(vis[v]) continue; vis[v] = true; dfs(v, u, dep+1); E[++clock] = u; H[clock] = dep; } R[u] = dfn; } void RMQ_init(int n) { for(int i = 1; i <= n; i++) dp[i][0] = i; for(int j = 1; (1<<j) <= n; j++) for(int i = 1; i+(1<<j)-1 <= n; i++) { if(H[dp[i][j-1]] < H[dp[i+(1<<(j-1))][j-1]]) dp[i][j] = dp[i][j-1]; else dp[i][j] = dp[i+(1<<(j-1))][j-1]; } } int RMQ(int l, int r) { l = I[l], r = I[r]; if(l > r) swap(l, r); int len = r-l+1, k = 0; while((1<<k) <= len) k++; k--; if(H[dp[l][k]] < H[dp[r-(1<<k)+1][k]]) return E[dp[l][k]]; else return E[dp[r-(1<<k)+1][k]]; } void pushdown(int rt, int l, int r) { int k = (r-l+1); if(add[rt]) { a[rt<<1] += add[rt]*(k-(k>>1)); a[rt<<1|1] += add[rt]*(k>>1); add[rt<<1] += add[rt]; add[rt<<1|1] += add[rt]; add[rt] = 0; } } void build(int l, int r, int rt) { a[rt] = 0; add[rt] = 0; if(l == r) return; int m = (l + r) >> 1; build(l, m, rt<<1); build(m+1, r, rt<<1|1); } void update(int x, int y, int l, int r, int rt, int num) { if(l == x && r == y) { a[rt] += (r-l+1)*num; add[rt] += num; return; } pushdown(rt, l, r); int m = (l + r) >> 1; if(y <= m) update(x, y, l, m, rt<<1, num); else if(x > m) update(x, y, m+1, r, rt<<1|1, num); else { update(x, m, l, m, rt<<1, num); update(m+1, y, m+1, r, rt<<1|1, num); } a[rt] = a[rt<<1] + a[rt<<1|1]; } int query(int x, int l, int r, int rt) { if(l == r) { return a[rt]; } pushdown(rt, l, r); int m = (l + r) >> 1; int ans = 0; if(x <= m) ans = query(x, l, m, rt<<1); else ans = query(x, m+1, r, rt<<1|1); a[rt] = a[rt<<1] + a[rt<<1|1]; return ans; } int main() { int cas = 1; int T; //scanf("%d", &T); int s, to, root, n, q; while(scanf("%d %d %d", &n, &q, &s) != EOF) { memset(vis, 0, sizeof(vis)); memset(first, -1, sizeof(first)); memset(degree, 0, sizeof(degree)); clock = cnt = dfn = 0; build(1, n, 1); //for(int i = 1; i <= n; i++) // scanf("%d", &b[i]); for(int i = 1; i < n; i++) { int u, v, w; scanf("%d %d %d", &u, &v, &w); e[i].u = u; e[i].v = v; e[i].w = w; AddEdge(u, v, 0); degree[v]++; } for(int i = 1; i <= n; i++) if(!degree[i]) { vis[i] = true; dfs(i, -1, 0); root = i; break; } RMQ_init(2*n-1); //puts("1"); for(int i = 1; i < n; i++) { int u = e[i].u; int v = e[i].v; int w = e[i].w; //printf("***%d %d\n", L[v], R[v]); if(L[u] < L[v]) update(L[v], R[v], 1, n, 1, w); else update(L[u], R[u], 1, n, 1, w); } while(q--) { int x; scanf("%d", &x); if(!x) { scanf("%d", &to); int d1 = query(L[s], 1, n, 1); int d2 = query(L[to], 1, n, 1); int lca = RMQ(s, to); int d3 = query(L[lca], 1, n, 1); //printf("***%d %d %d\n", d1, d2, d3); printf("%d\n", d1+d2-2*d3); //printf("%d\n", dfn); s = to; } else { int i, w; scanf("%d %d", &i, &w); int x = w - e[i].w; e[i].w = w; int v = e[i].v; int u = e[i].u; if(L[u] < L[v]) update(L[v], R[v], 1, n, 1, x); else update(L[u], R[u], 1, n, 1, x); } } } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。