码迷,mamicode.com
首页 > 其他好文 > 详细

fzu2028

时间:2015-05-29 21:35:34      阅读:106      评论:0      收藏:0      [点我收藏+]

标签:

技术分享
//Accepted    7324 KB    203 ms
/*
    source:fzu2028
    time  :2015.5.29
    by    :songt
  */
/*题解:
树链剖分
单点更新,求路径和
  */
#include <cstdio>
#include <cstring>

const int imax_n = 50005;

int max(int a,int b)
{
    return a>b?a:b;
}

void swap(int &a,int &b)
{
    int t=a;
    a=b;
    b=t;
}

struct Edge
{
    int u,v;
    Edge(){}
    Edge(int u,int v):u(u),v(v){}
}
edge[2*imax_n];
int head[imax_n];
int next[2*imax_n];
int tot;
void addEdge(int u,int v)
{
    edge[tot]=Edge(u,v);
    next[tot]=head[u];
    head[u]=tot++;
}
int fa[imax_n],deep[imax_n],num[imax_n],son[imax_n];
int p[imax_n],fp[imax_n],top[imax_n];
int pos;
void init()
{
    memset(head,-1,sizeof(head));
    memset(next,-1,sizeof(next));
    tot=0;
    memset(son,-1,sizeof(son));
    pos=0;
}
void dfs1(int u,int pre,int depth)
{
    deep[u]=depth;
    fa[u]=pre;
    num[u]=1;
    for (int i=head[u];i+1;i=next[i])
    {
        int v=edge[i].v;
        if (v!=pre)
        {
            dfs1(v,u,depth+1);
            num[u]+=num[v];
            if (son[u]==-1 || num[son[u]]<num[v])
                son[u]=v;
        }
    }
}

void dfs2(int u,int sp)
{
    p[u]=pos++;
    fp[p[u]]=u;
    top[u]=sp;
    if (son[u]==-1) return ;
    dfs2(son[u],sp);
    for (int i=head[u];i+1;i=next[i])
    {
        int v=edge[i].v;
        if (v!=fa[u] && v!=son[u])
        {
            dfs2(v,v);
        }
    }
}
struct Tree
{
    int l,r;
    long long sum;
    int tmax;
}f[imax_n*3];
void build(int t,int l,int r)
{
    f[t].l=l;
    f[t].r=r;
    f[t].sum=0;
    f[t].tmax=0;
    if (l==r)
    {
        return ;
    }
    int mid=(f[t].l+f[t].r)>>1;
    build(2*t,l,mid);
    build(2*t+1,mid+1,r);
}
void update(int t,int k,int value)
{
    if (f[t].l==k && f[t].r==k)
    {
        f[t].sum=value;
        f[t].tmax=value;
        return ;
    }
    int mid=(f[t].l+f[t].r)>>1;
    if (k<=mid) update(2*t,k,value);
    else update(2*t+1,k,value);
    f[t].sum=f[2*t].sum+f[2*t+1].sum;
    f[t].tmax=max(f[2*t].tmax,f[2*t+1].tmax);
}
long long query(int t,int l,int r,int &tmax)
{
    if (f[t].l==l && f[t].r==r)
    {
        tmax=f[t].tmax;
        return f[t].sum;
    }
    int mid=(f[t].l+f[t].r)>>1;
    if (r<=mid) return query(2*t,l,r,tmax);
    else
    {
        if (l>mid) return query(2*t+1,l,r,tmax);
        else
        {
            int tmax1,tmax2;
            long long sum1,sum2;
            sum1=query(2*t,l,mid,tmax1);
            sum2=query(2*t+1,mid+1,r,tmax2);
            tmax=max(tmax1,tmax2);
            return sum1+sum2;
        }
    }
}

long long find(int u,int v,int tmax)
{
    int f1=top[u],f2=top[v];
    long long sum=0;
    int tmp=0;
    int t;
    while (f1!=f2)
    {
        if (deep[f1]<deep[f2])
        {
            swap(f1,f2);
            swap(u,v);
        }
        sum+=query(1,p[f1],p[u],t);
        tmp=max(tmp,t);
        u=fa[f1];
        f1=top[u];
    }
    if (u==v) return sum;
    if (deep[u]>deep[v]) swap(u,v);
    sum+=query(1,p[son[u]],p[v],t);
    tmp=max(tmp,t);
    tmax=tmp;
    return sum;
}

int e[imax_n][3];
int n,m;

int main()
{
    while (scanf("%d%d",&n,&m)==2)
    {
        init();
        for (int i=0;i<n-1;i++)
        {
            scanf("%d%d%d",&e[i][0],&e[i][1],&e[i][2]);
            addEdge(e[i][0],e[i][1]);
            addEdge(e[i][1],e[i][0]);
        }
        dfs1(1,0,0);
        dfs2(1,1);
        //printf("after dfs\n");
        build(1,0,pos-1);
        //printf("after build\n");
        for (int i=0;i<n-1;i++)
        {
            if (deep[e[i][0]]<deep[e[i][1]])
                swap(e[i][0],e[i][1]);
            update(1,p[e[i][0]],e[i][2]);
        }
        //printf("after insert\n");
        int kind;
        int u,v,c;
        int tmax;
        for (int i=0;i<m;i++)
        {
            scanf("%d%d%d",&kind,&u,&v);
            if (kind==0)
            {
                update(1,p[e[u-1][0]],v);
            }
            else
            {
                printf("%lld\n",find(u,v,tmax));
            }
        }
    }
    return 0;
}
View Code

 

fzu2028

标签:

原文地址:http://www.cnblogs.com/djingjing/p/4539381.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!