LibreOJ #6208. 树上询问

　　　　　　　　　　　　　　　　　　　　　　　　内存限制：512 MiB 时间限制：500 ms 标准输入输出

　　　　　　　　　　　　　　　　　　　　　　　　　　　　题目类型：传统 评测方式：文本比较

　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　上传者： 匿名

树链剖分+线段树

屠龙宝刀点击就送

#include <vector>
#include <cstdio>
#define N 100005
using namespace std;
vector<int>G[N];
struct Segment
{
    int l,r,mid,t,ki,k,flagk,flagt;
    Segment * ch[];
    Segment ()
    {
        t=ki=k=flagk=flagt=;
        ch[]=ch[]=NULL;
    }
}*root=new Segment;
int n,m,tim,top[N],siz[N],dep[N],fa[N],belong[N];
void dfs1(int x)
{
    siz[x]=;
    dep[x]=dep[fa[x]]+;
    for(int i=;i<G[x].size();++i)
    {
        int v=G[x][i];
        if(fa[x]==v) continue;
        fa[v]=x;
        dfs1(v);
        siz[x]+=siz[v];
    }
}
void dfs2(int x)
{
    if(!top[x]) top[x]=x;
    int p=;
    belong[x]=++tim;
    for(int i=;i<G[x].size();++i)
    {
        int v=G[x][i];
        if(fa[x]!=v&&siz[p]<siz[v]) p=v;
    }
    if(p) top[p]=top[x],dfs2(p);
    for(int i=;i<G[x].size();++i)
    {
        int v=G[x][i];
        if(fa[x]!=v&&v!=p) dfs2(v);
    }
}
void build(Segment *&k,int l,int r)
{
    k=new Segment;
    k->l=l;k->r=r;
    if(l==r) return;
    k->mid=(l+r)>>;
    build(k->ch[],l,k->mid);
    build(k->ch[],k->mid+,r);
}
void pushdown(Segment *&k)
{
    k->ch[]->t+=k->t+k->flagt*k->ch[]->k;
    k->ch[]->t+=k->t+k->flagt*k->ch[]->k;
    k->ch[]->ki+=k->flagk;
    k->ch[]->ki+=k->flagk;
    k->ch[]->flagk+=k->flagk;
    k->ch[]->flagk+=k->flagk;
    k->ch[]->flagt+=k->flagt;
    k->ch[]->flagt+=k->flagt;
    k->flagk=;
    k->flagt=;
    k->t=;
}
void plusk(Segment *&k,int l,int r,int d)
{
    if(k->l==l&&k->r==r) {k->ki+=d;k->k+=d;k->flagk+=d;return;}
    pushdown(k);
    if(l>k->mid) plusk(k->ch[],l,r,d);
    else if(r<=k->mid) plusk(k->ch[],l,r,d);
    else plusk(k->ch[],l,k->mid,d),plusk(k->ch[],k->mid+,r,d);
}
void plust(Segment *&k,int l,int r,int d)
{
    if(k->l==l&&k->r==r) {k->t+=d*k->ki;k->flagt+=d;return;}
    pushdown(k);
    if(l>k->mid) plust(k->ch[],l,r,d);
    else if(r<=k->mid) plust(k->ch[],l,r,d);
    else plust(k->ch[],l,k->mid,d),plust(k->ch[],k->mid+,r,d);
}
int query(Segment *&k,int t)
{
    if(k->l==k->r) return k->t;
    pushdown(k);
    if(t<=k->mid) return query(k->ch[],t);
    else return query(k->ch[],t);
}
void solvek(int x,int y,int d)
{
    for(;top[x]!=top[y];x=fa[top[x]])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        plusk(root,belong[top[x]],belong[x],d);
    }
    if(dep[x]<dep[y]) swap(x,y);
    plusk(root,belong[y],belong[x],d);
}
void solvet(int x,int y,int d)
{
    for(;top[x]!=top[y];x=fa[top[x]])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        plust(root,belong[top[x]],belong[x],d);
    }
    if(dep[x]<dep[y]) swap(x,y);
    plust(root,belong[y],belong[x],d);
}
int Main()
{
    scanf("%d",&n);
    for(int u,v,i=;i<n;++i)
    {
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs1();dfs2();
    build(root,,n);
    scanf("%d",&m);
    for(int opt,x,d;m--;)
    {
        scanf("%d%d",&opt,&x);
        if(opt==) {scanf("%d",&d);if(d) solvek(,x,d);}
        else if(opt==) {scanf("%d",&d);if(d) solvet(,x,d);}
        else printf("%d\n",query(root,belong[x]));
    }
    return ;
}
int sb=Main();
int main(int argc,char *argv[]){;}