【SPOJ Query on a tree 】 (树链剖分)

http://acm.hust.edu.cn/vjudge/problem/13013

题意：

有一棵N个节点的树（1<=N<=10000），N-1条边，边的编号为1~N-1，每条边有一个权值，要求模拟两种操作：

1：QUERY x y 求节点x和节点y之间的路径中权值最大的边。

2：CHANGE p k修改第p条边的权值为k。

 

【分析】

　　树链剖分裸题。。

　　【表示我一开始怎么TLE，后来怎么AC的并不知道。。

 

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<set>
using namespace std;
#define Maxn 10010
#define INF 0x7fffffff

int mymax(int x,int y) {return x>y?x:y;}

struct node
{
    int x,y,c,next;
}t[2*Maxn];int len;

int first[Maxn];
void ins(int x,int y,int c)
{
    t[++len].x=x;t[len].y=y;t[len].c=c;
    t[len].next=first[x];first[x]=len;
}

char s[110];

int son[Maxn],dfn[Maxn],top[Maxn],cnt;
int sm[Maxn],fa[Maxn],id[Maxn],dep[Maxn];
void dfs1(int x,int f)
{
    dep[x]=dep[f]+1;
    sm[x]=1;son[x]=0;fa[x]=f;
    for(int i=first[x];i;i=t[i].next) if(t[i].y!=f)
    {
        int y=t[i].y;
        dfs1(y,x);
        sm[x]+=sm[y];
        if(son[x]==0||sm[son[x]]<sm[y]) son[x]=y;
    }
}

void dfs2(int x,int tp)
{
    dfn[x]=++cnt;top[x]=tp;
    if(son[x]) dfs2(son[x],tp);
    for(int i=first[x];i;i=t[i].next) if(t[i].y!=fa[x]&&t[i].y!=son[x])
     dfs2(t[i].y,t[i].y);
}

struct nnode
{
    int l,r,lc,rc,mx;
}tr[2*Maxn];int tot;

int build(int l,int r)
{
    int x=++tot;
    tr[x].l=l;tr[x].r=r;
    int mid=(l+r)>>1;
    if(l!=r)
    {
        tr[x].lc=build(l,mid);
        tr[x].rc=build(mid+1,r);
    }
    tr[x].mx=-INF;
    return x;
}

void change(int x,int y,int z)
{
    if(tr[x].l==tr[x].r)
    {
        tr[x].mx=z;
        return;
    }
    int mid=(tr[x].l+tr[x].r)>>1;
    if(y<=mid) change(tr[x].lc,y,z);
    else change(tr[x].rc,y,z);
    tr[x].mx=mymax(tr[tr[x].lc].mx,tr[tr[x].rc].mx);
}

int query(int x,int l,int r)
{
    if(tr[x].l==l&&tr[x].r==r) return tr[x].mx;
    int mid=(tr[x].l+tr[x].r)>>1;
    if(r<=mid) return query(tr[x].lc,l,r);
    else if(l>mid) return query(tr[x].rc,l,r);
    return mymax(query(tr[x].lc,l,mid),query(tr[x].rc,mid+1,r));
}

int get_ans(int x,int y)
{
    int mx=0;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        mx=mymax(mx,query(1,dfn[top[x]],dfn[x]));
        x=fa[top[x]];
    }
    if(x==y) return mx;
    if(dep[x]>dep[y]) swap(x,y);
    mx=mymax(mx,query(1,dfn[x]+1,dfn[y]));
    return mx;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        len=0;
        memset(first,0,sizeof(first));
        for(int i=1;i<n;i++)
        {
            int x,y,c;
            scanf("%d%d%d",&x,&y,&c);
            ins(x,y,c);ins(y,x,c);
        }
        dep[0]=0;dfs1(1,0);
        cnt=0;dfs2(1,1);
        tot=0;build(1,n);
        for(int i=1;i<=len;i+=2)
        {
            if(fa[t[i].x]==t[i].y) id[(i+1)/2]=t[i].x;
            else id[(i+1)/2]=t[i].y;
            change(1,dfn[id[(i+1)/2]],t[i].c);
        }
        while(1)
        {
            scanf("%s",s);
            if(s[0]=='D') break;
            int x,y;
            scanf("%d%d",&x,&y);
            if(s[0]=='C')
            {
                change(1,dfn[id[x]],y);
            }
            else
            {
                printf("%d
",get_ans(x,y));
            }
        }
    }
    return 0;
}

2017-01-21 11:45:00