[树链剖分]POJ2763 Housewife Wind

 

原题链接:

https://vjudge.net/problem/POJ-2763

题意:

维护一个树，权值在边上，支持单点更新和区间查询操作。

 

题解:

树链剖分的模板题，只不过把点权换成了边权，因为树每个点头上只有一条边，所以可以把边权映射到它的深度较深的那个点上，然后就套上模板就好了。

总结：

虽然是个模板提，但是还是wa到怀疑人生。

首先我开始是用mp[mk(u,v)]哈希的边，但是tle了。

后来用边的序号来就快了很多。

然后就是刚开始的时候以为应该在每次查询的时候最后如果x==y的话都减去深度最小的那个点，保证查询的是边权，但是无限wa。怀疑了两天人生之后发现查询每次跳top的时候并不是跳到了top，而是跳到了top的父亲，所以如果x==y的话再减去深度最小的点的话就多余了。。

 

代码:

#define push_back pb
#define make_pair mk
#define rd read()
#define mem(a,b) memset(a,b,sizeof(a))
#define bug printf("*********\n");
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define ls tr[u].ch[0]
#define rs tr[u].ch[1]
#define FIN freopen(D://code//in.txt", "r", stdin);
#define debug(x) cout<<"["<<x<<"]" <<endl;
#define IO ios::sync_with_stdio(flse),cin.tie(0);
//#pragma comment(linker,"/STACk:1024000000,1024000000")
//#include<bits/stdc++.h>
#include<time.h>
#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<cmath>
#include<map>
#include<algorithm>
#include<string>
#include<string.h>
#include<set>
#include<queue>
#include<stack>
#include<functional>
using std::pair;


typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
//const double PI=acos(-1);


const int maxn = 2e5 + 10;
const int maxm = 1e6 + 10;
const int mod = 1000000007;
const int inf = 0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;
const double dinf=1e20;
const double eps=1e-8;
using namespace std;

ll read() {
    ll X = 0, p = 1; char c = getchar();
    for(; c > '9' || c < '0'; c = getchar()) if(c == '-') p = -1;
    for(; c >= '0' && c <= '9'; c = getchar()) X = X * 10 + c - '0';
    return X * p;
}
//*********************************************************************
struct edge{int to,next,id;}e[maxn<<1];
int head[maxn];
int num;
void addedge(int u,int v)
{
    e[num].to=v;
    e[num].next=head[u];
    e[num].id=num;
    head[u]=num++;
    e[num].to=u;
    e[num].next=head[v];
    e[num].id=num;
    head[v]=num++;
}
int cnt,tot;
int n,m,q;
int w[maxn<<1];
int ww[maxn<<1];
int a[maxn],b[maxn];
int dep[maxn],f[maxn],son[maxn],sz[maxn],top[maxn],id[maxn];
int sum[maxn<<2];
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}


void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=a[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}

void update(int p,int x,int l,int r,int rt)
{
    if(p==l&&p==r)
    {
        sum[rt]=x;
        return ;
    }
    int mid=(l+r)>>1;
    //pushdown(rt,r-l+1);
    if(p<=mid) update(p,x,lson);
    if(p>mid) update(p,x,rson);
    pushup(rt);
}

int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return sum[rt];
    }
    int mid=(l+r)>>1;
    int ans=0;
    if(L<=mid) ans+=query(L,R,lson);
    if(R>mid) ans+=query(L,R,rson);
    return ans;
}

void dfs1(int u,int fa,int depth)
{
    dep[u]=depth;
    f[u]=fa;
    sz[u]=1;
    int maxson=-1;
    for(int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].to;
        if(v==fa) continue;
        int idd=w[e[i].id]?e[i].id:e[i^1].id;
        b[v]=w[idd];
        ww[idd]=v;
        ww[idd^1]=v;
        dfs1(v,u,depth+1);
        sz[u]+=sz[v];
        if(sz[v]>maxson) maxson=sz[v],son[u]=v;
    }
}

void dfs2(int u,int t)
{
    id[u]=++tot;
    a[tot]=b[u];
    top[u]=t;
    if(!son[u]) return;
    dfs2(son[u],t);
    for(int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].to;
        if(!id[v])
            dfs2(v,v);
    }
}
void querypath(int x,int y)
{
    if(x==y)
    {
        printf("0\n");
        return ;
    }
    int ans=0;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        ans+=query(id[top[x]],id[x],1,n,1);
        x=f[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    if(x!=y)
    ans+=query(id[son[x]],id[y],1,n,1);
    printf("%d\n",ans);
}
void init()
{
    tot=cnt=0;
    mem(sum,0);
    mem(head,-1);
    mem(son,0);
    mem(id,0);
    mem(w,0);
    mem(ww,0);
    mem(b,0);
    num=0;
}
int main()
{
scanf("%d%d%d",&n,&q,&m);

        int u,v;
        init();
        for(int i=0;i<n-1;i++)
        {
            scanf("%d%d%d",&u,&v,&w[2*i]);
            addedge(u,v);
        }
        dfs1(1,0,1);
        dfs2(1,1);
        build(1,n,1);
        int op,x,y;
       while(q--)
       {
            scanf("%d%d",&op,&x);
            if(op==0)
            {
                querypath(m,x);
                m=x;
            }
            else
            {
                scanf("%d",&y);
                x--;
                update(id[ww[x*2]],y,1,n,1);
            }
       }

}