Aragorn's Story HDU - 3966 + Query on a tree SPOJ - QTREE

树链剖分模板题

点击打开链接 基于点权 链修改单点查询

点击打开链接 基于边权 单点修改链查询

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e4+10;

struct node
{
    int v,next;
};

node edge[2*maxn];
int tree[4*maxn];
int first[maxn],val[maxn],sum[maxn],son[maxn],deep[maxn],fa[maxn],mp1[maxn],mp2[maxn],top[maxn];
int n,m,q,num,ans;

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
}

void dfsI(int cur)
{
    int i,v;
    sum[cur]=1,son[cur]=-1;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa[cur]){
            deep[v]=deep[cur]+1,fa[v]=cur;
            dfsI(v);
            sum[cur]+=sum[v];
            if(son[cur]==-1||sum[son[cur]]<sum[v]){
                son[cur]=v;
            }
        }
    }
}

void dfsII(int cur,int tp)
{
    int i,v;
    num++;
    mp1[cur]=num,mp2[num]=cur,top[cur]=tp;
    if(son[cur]==-1) return;
    dfsII(son[cur],tp);
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa[cur]&&v!=son[cur]){
            dfsII(v,v);
        }
    }
}

void build(int l,int r,int cur)
{
    int m;
    tree[cur]=0;
    if(l==r){
        tree[cur]=val[mp2[l]];
        return;
    }
    m=(l+r)/2;
    build(l,m,2*cur);
    build(m+1,r,2*cur+1);
}

void updateII(int pl,int pr,int val,int l,int r,int cur)
{
    int m;
    if(pl<=l&&r<=pr){
        tree[cur]+=val;
        return;
    }
    m=(l+r)/2;
    if(pl<=m) updateII(pl,pr,val,l,m,2*cur);
    if(pr>m) updateII(pl,pr,val,m+1,r,2*cur+1);
}

void updateI(int u,int v,int w)
{
    while(top[u]!=top[v]){
        if(deep[top[u]]<deep[top[v]]) swap(u,v);
        updateII(mp1[top[u]],mp1[u],w,1,n,1);
        u=fa[top[u]];
    }
    if(deep[u]<deep[v]) swap(u,v);
    updateII(mp1[v],mp1[u],w,1,n,1);
}

void query(int tar,int l,int r,int cur)
{
    int m;
    ans+=tree[cur];
    if(l==r) return;
    m=(l+r)/2;
    if(tar<=m) query(tar,l,m,2*cur);
    else query(tar,m+1,r,2*cur+1);
}

int main()
{
    int i,u,v,w;
    char op[10];
    while(scanf("%d%d%d",&n,&m,&q)!=EOF){
        for(i=1;i<=n;i++) scanf("%d",&val[i]);
        for(i=1;i<=n;i++) first[i]=-1;
        num=0;
        while(m--){
            scanf("%d%d",&u,&v);
            addedge(u,v),addedge(v,u);
        }
        deep[1]=1,fa[1]=0;
        dfsI(1);
        num=0;
        dfsII(1,1);
        build(1,n,1);
        while(q--){
            scanf("%s",op);
            if(op[0]=='I'){
                scanf("%d%d%d",&u,&v,&w);
                updateI(u,v,w);
            }
            else if(op[0]=='D'){
                scanf("%d%d%d",&u,&v,&w);
                updateI(u,v,-w);
            }
            else{
                scanf("%d",&u);
                ans=0;
                query(mp1[u],1,n,1);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}

 

#include <bits/stdc++.h>
using namespace std;
#define N 0x3f3f3f3f

struct node1
{
    int u;
    int v;
    int w;
};

struct node2
{
    int v;
    int next;
};

struct node3
{
    int l;
    int r;
    int val;
};

node1 pre[10010];
node2 edge[20010];
node3 tree[40010];
int val[10010],first[10010],fa[10010],deep[10010],sum[10010],son[10010],top[10010],mp1[10010],mp2[10010];
int n,num;

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
}

void dfsI(int cur)
{
    int i,v;
    sum[cur]=1,son[cur]=-1;
    for(i=first[cur];i!=-1;i=edge[i].next)
    {
        v=edge[i].v;
        if(v!=fa[cur])
        {
            fa[v]=cur,deep[v]=deep[cur]+1;
            dfsI(v);
            sum[cur]+=sum[v];
            if(son[cur]==-1||sum[son[cur]]<sum[v])
            {
                son[cur]=v;
            }
        }
    }
}

void dfsII(int cur,int tp)
{
    int i,v;
    num++;
    top[cur]=tp,mp1[cur]=num,mp2[num]=cur;
    if(son[cur]==-1) return;
    dfsII(son[cur],tp);
    for(i=first[cur];i!=-1;i=edge[i].next)
    {
        v=edge[i].v;
        if(v!=fa[cur]&&v!=son[cur])
        {
            dfsII(v,v);
        }
    }
}

void pushup(int cur)
{
    tree[cur].val=max(tree[2*cur].val,tree[2*cur+1].val);
}

void build(int l,int r,int cur)
{
    int m;
    tree[cur].l=l;
    tree[cur].r=r;
    tree[cur].val=0;
    if(l==r)
    {
        tree[cur].val=val[mp2[l]];
        return;
    }
    m=(l+r)/2;
    build(l,m,2*cur);
    build(m+1,r,2*cur+1);
    pushup(cur);
}

void update(int tar,int val,int cur)
{
    if(tree[cur].l==tree[cur].r)
    {
        tree[cur].val=val;
        return;
    }
    if(tar<=tree[2*cur].r) update(tar,val,2*cur);
    else update(tar,val,2*cur+1);
    pushup(cur);
}

int queryII(int pl,int pr,int cur)
{
    int res;
    if(pl<=tree[cur].l&&tree[cur].r<=pr)
    {
        return tree[cur].val;
    }
    res=-N;
    if(pl<=tree[2*cur].r) res=max(res,queryII(pl,pr,2*cur));
    if(pr>=tree[2*cur+1].l) res=max(res,queryII(pl,pr,2*cur+1));
    return res;
}

int queryI(int u,int v)
{
    int res;
    res=-N;
    while(top[u]!=top[v])
    {
        if(deep[top[u]]<deep[top[v]]) swap(u,v);
        res=max(res,queryII(mp1[top[u]],mp1[u],1));
        u=fa[top[u]];
    }
    if(u==v) return res;
    if(deep[u]<deep[v]) swap(u,v);
    res=max(res,queryII(mp1[son[v]],mp1[u],1));
    return res;
}

int main()
{
    int t,i,u,v,w;
    char op[10];
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        memset(first,-1,sizeof(first));
        num=0;
        for(i=1;i<=n-1;i++)
        {
            scanf("%d%d%d",&pre[i].u,&pre[i].v,&pre[i].w);
            addedge(pre[i].u,pre[i].v);
            addedge(pre[i].v,pre[i].u);
        }
        fa[1]=-1,deep[1]=1;
        dfsI(1);
        num=0;
        dfsII(1,1);
        val[1]=-N;
        for(i=1;i<=n-1;i++)
        {
            if(deep[pre[i].u]<deep[pre[i].v]) swap(pre[i].u,pre[i].v);
            val[pre[i].u]=pre[i].w;
        }
        build(1,n,1);
        while(1)
        {
            scanf("%s",op);
            if(op[0]=='C')
            {
                scanf("%d%d",&i,&w);
                update(mp1[pre[i].u],w,1);
            }
            else if(op[0]=='Q')
            {
                scanf("%d%d",&u,&v);
                printf("%d\n",queryI(u,v));
            }
            else break;
        }
    }
    return 0;
}