lightoj 1348 (树链剖分入门)
来源:互联网 发布:苍南软件开发 编辑:程序博客网 时间:2024/04/30 05:35
题意:修改一个节点上的点权,然后查询两个节点之间的路上的所有的节点的权值和。
思路:用树链剖分一下,然后直接写一个线段树维护一下区间和,更新直接在线段树上更新就行。
树链剖分只是把一些点映射到一个区间(一个重链相当于映射到了一个区间,轻链这是映射到了孤立的一些点),然后需要线段树或者rmq之类的数据结构基础来维护一些值
#include<bits/stdc++.h>using namespace std;typedef long long ll;typedef pair<int,int> P;#define fi first#define se second#define INF 0x3f3f3f3f#define clr(x,y) memset(x,y,sizeof x)#define PI acos(-1.0)#define ITER set<int>::iteratorconst int Mod = 1e9 + 7;const int maxn = 30000 + 10;struct Edge{ int to,next;}edge[maxn * 3];int head[maxn],edge_num,top[maxn],fa[maxn],son[maxn],id[maxn],deq[maxn],siz[maxn],num;int tree[maxn<<2],a[maxn];void add(int x,int y){ edge[edge_num] = (Edge){y,head[x]}; head[x] = edge_num++;}//第一次搜索找出siz[],son[],deq[],fa[]void dfs1(int rt,int pa,int d){ deq[rt] = d; son[rt] = 0; fa[rt] = pa; siz[rt] = 1; for(int i=head[rt]; i!=-1; i=edge[i].next){ int t = edge[i].to; if(t != fa[rt]){ dfs1(t,rt,d+1); siz[rt] += siz[t]; if(siz[son[rt]] < siz[t]){//如果父节点的son[rt]的iz[]小于子节点的siz[]更新son[] son[rt] = t; } } }}//第二次搜索找到w[],top[]void dfs2(int rt,int tp){ top[rt] = tp; id[rt] = ++num; if(son[rt] != 0) dfs2(son[rt],tp);//如果当前父节点纯在son 那么为了让链在线段树中连续 先对son[]进行搜索,并且重边顶端的点相同。 for(int i=head[rt]; i!=-1; i=edge[i].next){ int t = edge[i].to; if(t != fa[rt] && son[rt] != t){ dfs2(t,t); } }}void update(int x,int val,int l,int r,int rt){ if(l == r){tree[rt] = val;return ;} int mid = (l + r) >> 1; if(mid >= x) update(x,val,l,mid,rt << 1); else update(x,val,mid + 1,r,rt << 1 | 1); tree[rt] = tree[rt << 1] + tree[rt << 1| 1];}int query(int l,int r,int L,int R,int rt){ if(l == L && r == R)return tree[rt]; int mid = (L + R) >> 1; int ret = 0; if(r <= mid)ret += query(l,r,L,mid,rt << 1); else if(l >= mid + 1)ret += query(l,r,mid + 1,R,rt << 1 | 1 ); else ret += query(l,mid,L,mid,rt << 1),ret += query(mid + 1,r,mid + 1,R,rt << 1|1); return ret;}int solve(int u,int v){ int f1=top[u],f2=top[v],tmp=0; while(f1!=f2){ if(deq[f1]<deq[f2]){ swap(f1,f2); swap(u,v); } tmp+=query(id[f1],id[u],1,num,1); u=fa[f1];f1=top[u]; } if(deq[u]>deq[v]) swap(u,v); tmp+=query(id[u],id[v],1,num,1); return tmp;}int main(){ int Tcase;scanf("%d",&Tcase); for(int ii = 1;ii <= Tcase; ii ++) { int n,m;scanf("%d",&n);edge_num = 0;clr(tree,0);clr(head,-1); for(int i = 1; i <= n;i ++)scanf("%d",&a[i]); for(int i = 1;i < n;i ++){int x,y;scanf("%d%d",&x,&y);++ x,++y;add(x,y);add(y,x);} num = 0;dfs1(1,1,1);dfs2(1,1);// for(int i = 1;i <= n;i ++)cout << i << " -> " << id[i] << endl; for(int i = 1;i <= n;i ++)update(id[i],a[i],1,num,1); scanf("%d",&m);printf("Case %d:\n",ii); while(m --) { int x,y,z;scanf("%d%d%d",&x,&y,&z); if(x == 1)update(id[y + 1],z,1,num,1); else printf("%d\n",solve(++ y,++ z)); } } return 0;}
阅读全文
0 0
- lightoj 1348 (树链剖分入门)
- LightOJ 1348 Aladdin and the Return Journey【树链剖分入门题】
- LightOj 1348(树链剖分)
- LIGHTOJ 1348 树链剖分
- lightoj 1164 - Horrible Queries 线段树入门
- lightoj 1422 - Halloween Costumes 【区间dp入门】
- lightOJ Discovering Gold(期望DP入门)
- LightOJ
- LightOJ
- LightOJ
- LightOJ
- LightOJ
- [LightOJ
- LightOJ
- LightOJ
- LightOJ
- LightOJ
- LightOJ
- iOS与内存管理
- ABAP自定义截取字符串长度函数
- ITUNES账号骗局!今年被骗子欺骗人数已超1200人
- Eclipse链接Android 7.0的手机无法输出logcat
- 实验二:命令行菜单小程序V1.0
- lightoj 1348 (树链剖分入门)
- c++ 一维数组与指针
- NetworkOnMainThreadException异常 --Android
- haslayout & bfc
- java多线程
- 第三周项目一---顺序表的基本运算
- 用iso文件快速搭建本地临时yum源
- HDU 6214 Smallest Minimum Cut【最小割的最小边数】
- H5--video