hdu-4836-The Query on the Tree（线段树+LCA）

题目链接

思路：对于每次询问，主要是看x和root的关系，求出root和xlca

root=x ，ans为总的和

lca=x  那么ans=总的和-（root到x这条链上父节点为x的那个点的子树和）

否则，ans就是x的子树和

求子树和和修改直接线段树维护。节点的编号和lca我是用的树链剖分求的

AC代码：

#include
using namespace std;
const int maxn=1e4+10;
int val[maxn];
struct E
{
    int next,to;
}edge[maxn*2];
int head[maxn],deep[maxn],top[maxn],fa[maxn],son[maxn],sz[maxn],id[maxn],fid[maxn],tot,num;
void add_edge(int u,int v)
{
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void dfs1(int u,int f,int dep)
{
    fa[u]=f;
    son[u]=0;
    sz[u]=1;
    deep[u]=dep;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int ff=edge[i].to;
        if(ff==f) continue;
        dfs1(ff,u,dep+1);
        sz[u]+=sz[ff];
        if(sz[ff]>sz[son[u]]) son[u]=ff;
    }
}
void dfs2(int u,int tp)
{
    id[u]=++num;
    top[u]=tp;
    if(son[u]) dfs2(son[u],tp);
    for(int i=head[u];i!=-1;i=edge[i].next){
        int ff=edge[i].to;
        if(ff==fa[u]||ff==son[u]) continue;
        dfs2(ff,ff);
    }
}
void init()
{
    num=tot=0;
    memset(head,-1,sizeof(head));
}

#define ll t[rt].l
#define rr t[rt].r
#define ls rt<<1
#define rs rt<<1|1

struct node
{
    int l,r;
    int val;
}t[maxn<<2];
void pushup(int rt)
{
    t[rt].val=t[ls].val+t[rs].val;
}
void build(int rt,int l,int r)
{
    t[rt].l=l;
    t[rt].r=r;
    if(l==r){
        t[rt].val=val[l];
        return  ;
    }
    int mid=(l+r)>>1;
    build(ls,l,mid);
    build(rs,mid+1,r);
    pushup(rt);
}
void update(int rt,int pos,int v)
{
    if(ll==rr){
        t[rt].val=v;
        return;
    }
    int mid=(ll+rr)>>1;
    if(pos<=mid) update(ls,pos,v);
    else update(rs,pos,v);
    pushup(rt);
}
int query(int rt,int l,int r)
{
    if(l<=ll&&rr<=r) return t[rt].val;
    int ans=0;
    int mid=(ll+rr)>>1;
    if(l<=mid) ans+=query(ls,l,r);
    if(r>mid) ans+=query(rs,l,r);
    return ans;
}

int Lca(int u,int v)
{
    while(top[u]!=top[v]){
        if(deep[top[u]]deep[v]) swap(u,v);
    if(u==v) return id[u]+1;
    return id[v]-(deep[v]-deep[u])+1;
}

int main()
{
    int T;
    scanf("%d",&T);
    int cas=1;
    while(T--){
        init();
        int n;
        scanf("%d",&n);
        int u,v;
        for(int i=1;i