CODEVS 4633 [Mz]树链剖分练习

题目描述 Description
给定一棵结点数为n的树，初始点权均为0，有依次q个操作，每次操作有三个参数a,b,c，当a=1时，表示给b号结点到c号结点路径上的所有点（包括b,c，下同）权值都增加1，当a=2时，表示询问b号结点到c号结点路径上的所有点权值之和。

输入描述 Input Description
第一行，一个正整数n。
接下来n-1行，每行一对正整数x,y，表示x号结点和y号结点之间有一条边。
第n+1行，一个正整数q。
最后q行，每行一组正整数a,b,c，表示操作的三个参数。b和c可能相等。
保证数据都是合法的。

输出描述 Output Description
若干行，每行一个非负整数表示答案。

数据范围及提示 Data Size & Hint
共有10个测试点，对于第i个测试点，当1<=i<=4时，n=q=10^i，当5<=i<=10时，n=q=10000*i。

以后再也不作死学省选内容了…..（除了线段树）

这个题直接带点权的，所以我们不需要把边权下放到点权了23333舒服。

对于修改操作….和询问区间和操作，其实两者长的很像，嗯嗯。

对于一条重链，它上面的元素是连续的，直接对xds[x]和xds[top[x]]进行操作即可。

还有，点不是边….点的话他在线段树中的位置就是他自己的位置辣。

直接上代码吧..

/*题目描述 Description
给定一棵结点数为n的树，初始点权均为0，有依次q个操作，每次操作有
三个参数a,b,c，当a=1时，表示给b号结点到c号结点路径上的所有点
（包括b,c，下同）权值都增加1，当a=2时，表示询问b号结点到c号
结点路径上的所有点权值之和。
输入描述 Input Description
第一行，一个正整数n。
接下来n-1行，每行一对正整数x,y，表示x号结点和y号结点之间有一条边。
第n+1行，一个正整数q。
最后q行，每行一组正整数a,b,c，表示操作的三个参数。b和c可能相等。
保证数据都是合法的。
输出描述 Output Description
若干行，每行一个非负整数表示答案。
样例输入 Sample Input
5
1 2
2 3
1 4
2 5
5
1 4 5
2 1 5
1 1 3
2 5 3
2 4 3
样例输出 Sample Output
3
4
6
数据范围及提示 Data Size & Hint
共有10个测试点，对于第i个测试点，当1<=i<=4时，n=q=10^i，当5<=i<=10时，n=q=10000*i。*/
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<queue>
#define hah puts("hah")
using namespace std;
const int maxn=150000*2+50;
int n;
struct Edge
{
    int f;
    int to;
    int next;
}edge[maxn];
int head[maxn];
int tot;
void add1(int f,int t)
{
    edge[++tot]=(Edge){f,t,head[f]};
    head[f]=tot;
} 
int siz[maxn],deep[maxn],son[maxn],fa[maxn];
void dfs1(int f,int t)
{
    fa[t]=f;
    deep[t]=deep[f]+1;
    siz[t]=1;
    for(int i=head[t];i;i=edge[i].next)
    {
        Edge e=edge[i];
        if(e.to==f)
            continue;
        dfs1(t,e.to);
        siz[t]+=siz[e.to];
        if(siz[e.to]>siz[son[t]])
            son[t]=e.to;
    }
}
int top[maxn],xds[maxn],intr[maxn];
int xdscnt;
void dfs2(int u,int topu)
{
    top[u]=topu;
    xds[u]=++xdscnt;
    intr[xdscnt]=u;
    if(!son[u])
        return;
    dfs2(son[u],topu);
    for(int i=head[u];i;i=edge[i].next)
    {
        Edge e=edge[i];
        if(e.to==son[u]||e.to==fa[u])
            continue;
        dfs2(e.to,e.to);
    }
}
struct TREE
{
    int l,r;
    int add,sum;
}t[maxn<<3];
void up(int p)
{
    t[p].sum=(t[p<<1].sum+t[p<<1|1].sum);
}
void buff(int p)
{
    if(t[p].add)
    {
        t[p<<1].sum+=(t[p<<1].r-t[p<<1].l+1)*t[p].add;
        t[p<<1|1].sum+=(t[p<<1|1].r-t[p<<1|1].l+1)*t[p].add;
        t[p<<1].add+=t[p].add;
        t[p<<1|1].add+=t[p].add;
        t[p].add=0;
    }
}
void expand(int p,int l,int r)
{
    t[p].l=l;
    t[p].r=r;
    if(l==r)
    {
        t[p].sum=0;
        return;
    }
    int mid=(l+r)>>1;
    expand(p<<1,l,mid);
    expand(p<<1|1,mid+1,r);
    up(p);
}
void add(int p,int l,int r)
{
    if(l<=t[p].l&&t[p].r<=r)
    {
        t[p].sum+=(t[p].r-t[p].l+1)*1;
        t[p].add+=1;
        return;
    }
    int mid=(t[p].l+t[p].r)>>1;
    if(l<=mid)
        add(p<<1,l,r);
    if(mid+1<=r)
        add(p<<1|1,l,r);
    buff(p);
    up(p); 
}
void add_hah(int x,int y)
{

    int fax=top[x];
    int fay=top[y];
    while(fax!=fay)
    {
        if(deep[fax]<deep[fay])
            swap(fax,fay),swap(x,y);
        add(1,xds[fax],xds[x]);
        x=fa[fax];
        fax=top[x];
    }
    if(deep[x]>deep[y])
    swap(x,y);
        add(1,xds[x],xds[y]);
    return;
}
int ask_sum(int p,int l,int r)
{
    buff(p);
    if(l<=t[p].l&&t[p].r<=r)
    {
        return t[p].sum;
    }   
    int ans=0;
    int mid=(t[p].l+t[p].r)>>1;
    if(l<=mid)
        ans+=ask_sum(p<<1,l,r);
    if(mid+1<=r)
        ans+=ask_sum(p<<1|1,l,r);
    return ans;
}
int find_sum(int x,int y)
{
    int ans=0;
    int fax=top[x],fay=top[y];
    while(fax!=fay)
    {
        if(deep[fax]<deep[fay])
            swap(fax,fay),swap(x,y);
        ans+=ask_sum(1,xds[fax],xds[x]);
        x=fa[fax];
        fax=top[x];
    }
    if(deep[x]>deep[y])
    swap(x,y);

        ans+=ask_sum(1,xds[x],xds[y]);
    return ans;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add1(a,b);
        add1(b,a);
    }
    dfs1(100200,1);
    dfs2(1,1);

    expand(1,1,n);
    int q;
    scanf("%d",&q);
    while(q--)
    {
        int a,b,c;

        scanf("%d%d%d",&a,&b,&c);
        if(a==1)
            add_hah(b,c);
        else    
            printf("%d\n",find_sum(b,c));
    }
    return 0;
}