Query on a tree SPOJ - QTREE(树链剖分)

传送门
题意:给定一棵树,告诉了每条边的权值,然后给出两种操作:
(1)把第i条边的权值改为val
(2)询问a,b路径上权值最大的边
假期之后的第一篇题解,只学习了一种新的数据结构,树链剖分,此题为入门题,这个我算是学习的最晚的一个人了。这个有种调用了线段的API的感觉

#include<cstdio>
#include<cstring>
#include<algorithm>
#include <iostream>
#include<cmath>
#include <string>
#define N 10010
using namespace std;
const int INF=1<<30;
int n,total;
int num[N],siz[N],top[N],son[N];
int dep[N],tid[N],fa[N];
//int rankk[N];
int head[N],to[2*N],nex[2*N],edge;
struct Edge
{
    int u,v,c;
} E[N];
void init()
{
    memset(head,-1,sizeof(head));
    memset(son,-1,sizeof(son));
    total=0;
    edge=0;
}
void addEdge(int u,int v,int c)
{
    to[edge]=v,nex[edge]=head[u],head[u]=edge++;
    to[edge]=u,nex[edge]=head[v],head[v]=edge++;
}
void dfs1(int u,int last,int d)
{
    dep[u]=d;
    fa[u]=last;
    siz[u]=1;
    for(int i=head[u];~i;i=nex[i])
    {
        int v=to[i];
        if(v==last)
            continue;
        dfs1(v,u,d+1);
        siz[u]+=siz[v];
        if(son[u]==-1||siz[v]>siz[son[u]])
            son[u]=v;
    }
}
void dfs2(int u, int tp)
{
    top[u]=tp;
    tid[u]=++total;
    //rankk[total]=u;
    if(son[u]==-1)
        return ;
    dfs2(son[u],tp);
    for(int i=head[u];~i;i=nex[i])
    {
        int v=to[i];
        if(v==son[u]||v==fa[u])
            continue;
        dfs2(v,v);
    }
}
int Max[N<<2];
void pushUp(int rt)
{
    Max[rt]=max(Max[rt<<1],Max[rt<<1|1]);
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        Max[rt]=num[l];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushUp(rt);
}
void change(int l,int r,int rt,int p,int val)
{
    if(l==r)
    {
        Max[rt]=val;
        return;
    }
    int mid=(l+r)>>1;
    if(p<=mid)
        change(l,mid,rt<<1,p,val);
    else
        change(mid+1,r,rt<<1|1,p,val);
    pushUp(rt);
}
int ask(int l,int r,int rt,int L,int R)
{
    if(L<=l&&r<=R)
        return Max[rt];
    int mid=(l+r)>>1;
    int ret=-INF;
    if(L<=mid)ret=max(ret,ask(l,mid,rt<<1,L,R));
    if(R>mid)ret=max(ret,ask(mid+1,r,rt<<1|1,L,R));
    return ret;
}
void change(int x,int val)
{
    if(dep[E[x].u]>dep[E[x].v])
        change(2,n,1,tid[E[x].u],val);
    else
        change(2,n,1,tid[E[x].v],val);
}
int ask(int x,int y)
{
    int ans=-INF;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        ans=max(ans,ask(2,n,1,tid[top[x]],tid[x]));
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    if(x!=y)
        ans=max(ans,ask(2,n,1,tid[x]+1,tid[y]));
    return ans;
}
int main()
{
    int t;
    char op[20];
    int a,b,c;
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d",&n);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            E[i].u=a,E[i].v=b,E[i].c=c;
            addEdge(a,b,c);
        }
        dfs1(1,1,1);
        dfs2(1,1);
        for(int i=1;i<n;i++)
            if(dep[E[i].u]>dep[E[i].v])
                num[tid[E[i].u]]=E[i].c;
            else
                num[tid[E[i].v]]=E[i].c;
        build(2,n,1);
        for(;;)
        {
            scanf("%s",op);
            if(op[0]=='D')
                break;
            scanf("%d%d",&a,&b);
            if(op[0]=='Q')
                printf("%d\n",ask(a,b));
            else
                change(a,b);
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值