POJ 2763 树链剖分

点击打开链接

题意：给一个树，然后树上的边的边权，然后两个操作，一个是询问u到v的路上权值和，一个是将第几条边的权值修改

思路：与SPOJ 375 的那道题目很像，都是边上的权值，然后维护一个线段树进行修改和求和就行了

#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=100010;
int fa[maxn],siz[maxn],son[maxn],w[maxn],p[maxn],dep[maxn],fp[maxn],head[maxn];
//fa为父节点，siz为子节点中siz最大的，dep为深度，son为重儿子，w表示在线段树中的位置
int num[maxn<<2];
int tree_id,n,kkk=0;
struct node{
    int to,next;
}EE[maxn*10];
void add_edge(int u,int v){
    EE[kkk].to=v;EE[kkk].next=head[u];head[u]=kkk++;
}
void dfs1(int u,int ff,int deep){
    son[u]=0;fa[u]=ff;siz[u]=1;dep[u]=deep;
    for(int i=head[u];i!=-1;i=EE[i].next){
        int v=EE[i].to;
        if(v==ff) continue;
        dfs1(v,u,deep+1);
        siz[u]+=siz[v];
        if(siz[v]>siz[son[u]]) son[u]=v;
    }
}
void dfs2(int u,int ff){
    w[u]=++tree_id;p[u]=ff;
    if(son[u]) dfs2(son[u],ff);
    else return ;
    for(int i=head[u];i!=-1;i=EE[i].next){
        int v=EE[i].to;
        if(v!=fa[u]&&v!=son[u]) dfs2(v,v);
    }
}
void update(int pos,int val,int le,int ri,int node){
    if(le==ri){
        num[node]=val;
        return ;
    }
    int t=(le+ri)>>1;
    if(pos<=t) update(pos,val,le,t,node<<1);
    else update(pos,val,t+1,ri,node<<1|1);
    num[node]=num[node<<1]+num[node<<1|1];
}
int query(int l,int r,int le,int ri,int node){
    if(l<=le&&ri<=r) return num[node];
    int t=(le+ri)>>1,ans=0;
    if(l<=t) ans+=query(l,r,le,t,node<<1);
    if(r>t) ans+=query(l,r,t+1,ri,node<<1|1);
    return ans;
}
int getsum(int u,int v){
    int f1=p[u],f2=p[v],tmp=0;
    while(f1!=f2){
        if(dep[f1]dep[v]) swap(u,v);
    return tmp+=query(w[son[u]],w[v],1,n,1);
}
int U[maxn],V[maxn],C[maxn];
int main(){
    int u,v,q,s,str;
    while(scanf("%d%d%d",&n,&q,&s)!=-1){
        memset(head,-1,sizeof(head));
        memset(son,0,sizeof(son));tree_id=0;kkk=0;
        for(int i=0;idep[V[i]]) swap(U[i],V[i]);
            update(w[V[i]],C[i],1,n,1);
        }
        while(q--){
            scanf("%d",&str);
            if(str==1) scanf("%d%d",&u,&v);
            else scanf("%d",&u);
            if(str==1) update(w[V[u-1]],v,1,n,1);
            else{
                printf("%d\n",getsum(s,u));
                s=u;
            }
        }
    }
    return 0;
}