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 Input51 22 31 42 551 4 52 1 51 1 32 5 32 4 3样例输出 Sample Output346数据范围及提示 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;}