当前位置: 代码迷 >> 综合 >> POJ 2763 Housewife Wind (树链剖分)
  详细解决方案

POJ 2763 Housewife Wind (树链剖分)

热度:73   发布时间:2023-12-17 03:34:46.0

题意:

给你n个点的树,每个边有个权值,开始处于点m,现在有q次操作,每次单点修改或者将点移动到u,对于每次移动输出经过路程的权值

思路:

树链剖分的基础题,不同点在于每次记得更新点时候把点的更新即可

错误及反思:

C++跑了1700,G++跑了2200,hdu上faq说过G++处理大量输入的时候会

代码:

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define fi first
#define se second
const int N =100010;
int segtree[N*4];
int top[N],si[N],fa[N],fi[N],son[N],depth[N],id[N],val[N];
pair<int,int> e[N*2],b[N];
int m,tot=0,tid=0,n,q;
void addedge(int x,int y)
{e[tot].fi=y;e[tot].se=fi[x];fi[x]=tot++;e[tot].fi=x;e[tot].se=fi[y];fi[y]=tot++;
}
void dfs1(int now,int bef,int dep)
{fa[now]=bef;depth[now]=dep;si[now]=1;for(int i=fi[now];i!=-1;i=e[i].se)if(e[i].fi!=bef){dfs1(e[i].fi,now,dep+1);si[now]+=si[e[i].fi];if(son[now]==-1) son[now]=e[i].fi;else son[now]=si[e[i].fi]>si[son[now]]?e[i].fi:son[now];}
}
void dfs2(int now,int tp)
{top[now]=tp;id[now]=tid++;if(son[now]!=-1) dfs2(son[now],tp);for(int i=fi[now];i!=-1;i=e[i].se)if(e[i].fi!=fa[now]&&e[i].fi!=son[now])dfs2(e[i].fi,e[i].fi);
}
void change(int pos,int v,int l,int r,int rt)
{if(pos==l&&l==r){segtree[rt]=v;return;}int m=(l+r)/2;if(m>=pos) change(pos,v,lson);else change(pos,v,rson);segtree[rt]=segtree[rt<<1]+segtree[rt<<1|1];
}
int query(int L,int R,int l,int r,int rt)
{if(L<=l&&R>=r)return segtree[rt];int m=(l+r)/2;int ans=0;if(m>=L) ans+=query(L,R,lson);if(m<R) ans+=query(L,R,rson);return ans;
}
int modify(int L,int R)
{int f1=top[L],f2=top[R];int ans=0;while(f1!=f2){if(depth[f1]<depth[f2]){swap(f1,f2);swap(L,R);}ans+=query(id[f1],id[L],0,tid-1,1);L=fa[f1];f1=top[L];}if(L==R) return ans;if(depth[L]>depth[R]) swap(L,R);ans+=query(id[son[L]],id[R],0,tid-1,1);return ans;
}
int main()
{memset(first,-1,sizeof(first));memset(son,-1,sizeof(son));scanf("%d%d%d",&n,&q,&m);for(int i=0;i<n-1;i++){scanf("%d%d%d",&b[i].fi,&b[i].se,&val[i]);addedge(b[i].fi,b[i].se);}dfs1(1,1,1);dfs2(1,1);for(int i=0;i<n-1;i++){if(depth[b[i].fi]<depth[b[i].se])swap(b[i].fi,b[i].se);change(id[b[i].fi],val[i],0,tid-1,1);}while(q--){int ta,tb,tc;scanf("%d",&ta);if(ta){scanf("%d%d",&tb,&tc);change(id[b[tb-1].fi],tc,0,tid-1,1);}else{scanf("%d",&tb);printf("%d\n",modify(m,tb));m=tb;}}
}