POJ 2763 Housewife Wind (LCA+树状数组)

10W节点的树,10W次操作:

1,:询问两点距离;

2:修改一条边权。

解法:如果没有修改,那么dis(a,b) = dep(a) + dep(b) - dep(lca(a,b)) * 2,可以用rmqLCA来O(1)做到。

但是有修改的话,基本就可以无脑考虑log级别的算法像树状数组线段树什么的了。

记一个dfs时间戳:进入节点记一次,退出节点记一次,那么可以得到边的两个时间序in[id]和out[id],以及点的第一次进入的时间序idn[u]。

然后把树按照时间序展开成序列C,对于边i以及边权val[i],C[in[i]] += val[i],C[out[i]] -= val[i],

然后你会发现,求节点u到根的距离,就是sigma(C[i]),0<i<=idn[u],前缀和嘛,树状数组就可以了。

修改边权的话,就把C数组的in[i]和out[i]位置更新到新的值就可以了。

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = 101000,INF = 0x3f3f3f3f;
int n,nq,s,head[N],etot,dep[N],plca[N][20],in[N],out[N],val[N],C[N<<1],tim,idn[N];
struct edge {int v,id,next;}g[N<<1];
void add_edge(int u,int v,int id) {
        g[etot].v = v, g[etot].id = id, g[etot].next = head[u], head[u] = etot ++;
}
void dfs(int u,int fa,int deep) {
        dep[u] = deep;
        plca[u][0] = fa;
        for (int i = 1; i < 20; i ++)
                plca[u][i] = plca[u][i-1]==-1 ? -1 : plca[plca[u][i-1]][i-1];
        for (int i = head[u]; i != -1; i = g[i].next) {
                edge &e = g[i];
                if (e.v==fa) continue;
                in[e.id] = idn[e.v] = ++tim;
                dfs(e.v,u,deep+1);
                out[e.id] = ++tim;
        }
}
int lca(int a,int b) {
        if (dep[a]<dep[b]) swap(a,b);
        for (int i = 0; i < 20; i ++)
                if (dep[a]-dep[b]>>i&1) a = plca[a][i];
        if (a!=b) {
                for (int i = 19; i >= 0; i --)
                        if (plca[a][i] != plca[b][i]) a = plca[a][i], b = plca[b][i];
                a = plca[a][0];
        }
        return a;
}
void modify(int p,int dt) {
        for (int i = p; i <= tim; i += i&-i) C[i] += dt;
}
int query(int p) {
        int ret = 0;
        for (int i = p; i > 0; i -= i&-i) ret += C[i];
        return ret;
}
int main() {
        while (~scanf("%d%d%d",&n,&nq,&s)) {
                memset(head,-1,sizeof(head)); etot = 0;
                tim = 0;
                for (int i = 1; i <= n-1; i ++) {
                        int a,b,c;
                        scanf("%d%d%d",&a,&b,&c);
                        add_edge(a,b,i);
                        add_edge(b,a,i);
                        val[i] = c;
                }
                dfs(1,-1,0);
                memset(C,0,sizeof(C));
                for (int i = 1; i <= n-1; i ++) {
                        modify(in[i],val[i]);
                        modify(out[i],-val[i]);
                }
                while (nq--) {
                        int op,a,b;
                        scanf("%d",&op);
                        if (op==0) {
                                scanf("%d",&a);
                                printf("%d\n",query(idn[a])+query(idn[s])-query(idn[lca(a,s)])*2);
                                s = a;
                        } else {
                                scanf("%d%d",&a,&b);
                                int dt = b-val[a];
                                modify(in[a],dt);
                                modify(out[a],-dt);
                                val[a] = b;
                        }
                }
        }
        return 0;
}


  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
POJ 2182是一道使用树状数组解决的题目,题目要求对给定的n个数进行排序,并且输出每个数在排序后的相对位置。树状数组是一种用来高效处理前缀和问题的数据结构。 根据引用中的描述,我们可以通过遍历数组a,对于每个元素a[i],可以使用二分查找找到a到a[i-1]中小于a[i]的数的个数。这个个数就是它在排序后的相对位置。 代码中的query函数用来求前缀和,add函数用来更新树状数组。在主函数中,我们从后往前遍历数组a,通过二分查找找到每个元素在排序后的相对位置,并将结果存入ans数组中。 最后,我们按顺序输出ans数组的元素即可得到排序后的相对位置。 参考代码如下: ```C++ #include <iostream> #include <cstdio> using namespace std; int n, a += y; } } int main() { scanf("%d", &n); f = 1; for (int i = 2; i <= n; i++) { scanf("%d", &a[i]); f[i = i & -i; } for (int i = n; i >= 1; i--) { int l = 1, r = n; while (l <= r) { int mid = (l + r) / 2; int k = query(mid - 1); if (a[i > k) { l = mid + 1; } else if (a[i < k) { r = mid - 1; } else { while (b[mid]) { mid++; } ans[i = mid; b[mid = true; add(mid, -1); break; } } } for (int i = 1; i <= n; i++) { printf("%d\n", ans[i]); } return 0; } ``` 这段代码使用了树状数组来完成题目要求的排序功能,其中query函数用来求前缀和,add函数用来更新树状数组。在主函数中,我们从后往前遍历数组a,通过二分查找找到每个元素在排序后的相对位置,并将结果存入ans数组中。最后,我们按顺序输出ans数组的元素即可得到排序后的相对位置。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [poj2182Lost Cows——树状数组快速查找](https://blog.csdn.net/aodan5477/article/details/102045839)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* [poj_2182 线段树/树状数组](https://blog.csdn.net/weixin_34138139/article/details/86389799)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值