[BZOJ4034][HAOI2015]T2 树链剖分+dfs序

常规题,比较特殊的操作是修改以x为根的子树的所有节点

我们知道,这颗子树的dfs序一定是连续的,所以我们只需要对每个节点记录一下,最大的dfs序即可

因为树剖要dfs,所以我们直接在第二次dfs时记录,但需要注意的是,因为我们最先对重链的节点编号,所以dfs时也必须先dfs重链

我采取的办法是直接加边,因为边表保证了最后加的边最先扩展,或者也可以直接记录一下重边连的点

(因为longlongWA了一次,树剖到此为止,不做了,明天再做一道吧)

/**************************************************************
    Problem: 4034
    User: cabinfever
    Language: C++
    Result: Accepted
    Time:2332 ms
    Memory:17332 kb
****************************************************************/
 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
 
using namespace std;
 
const int maxn = 100010;
 
struct edge
{
    int v,next;
}e[maxn * 3];
int h[maxn];
int tot = 0, num = 0, cnt = 0;
struct node
{
    int l,r;
    long long x,mark;
}t[maxn << 1];
long long d;
long long a[maxn];
bool vis[maxn];
int n,m,u,v,x,y;
int fa[maxn];
int dep[maxn];
int son[maxn];
int pos[maxn];
int bot[maxn];
int ppos[maxn];
int top[maxn];
int size[maxn];
 
void build_edge(int u,int v)
{
    num++;
    e[num].v = v;
    e[num].next = h[u];
    h[u] = num;
}
 
void dfs1(int x)
{
    dep[x] = dep[fa[x]] + 1;
    vis[x] = true;
    size[x] = 1;
    son[x] = 0;
    for(int i = h[x]; i; i = e[i].next)
    {
        if(vis[e[i].v])
            continue;
        fa[e[i].v] = x;
        dfs1(e[i].v);
        if(size[e[i].v] > size[son[x]])
            son[x] = e[i].v;
        size[x] += size[e[i].v];
    }
}
 
void dfs2(int x)
{
    vis[x] = true;
    if(son[fa[x]] == x)
        top[x] = top[fa[x]];
    else
    {
        top[x] = x;
        for(int i = x; i; i = son[i])
        {
            if(son[i])
                build_edge(i,son[i]);
            pos[i] = ++cnt;
            ppos[cnt] = i;
        }
    }
    for(int i = h[x]; i; i = e[i].next)
        if(!vis[e[i].v])
            dfs2(e[i].v);
    bot[x] = cnt;
}
 
void build(int p,int l,int r)
{
    int mid = l + r >> 1;
    if(l == r)
    {
        t[p].x = a[ppos[mid]];
        return;
    }
    t[p].l = ++tot;
    t[p].r = ++tot;
    build(t[p].l,l,mid);
    build(t[p].r,mid+1,r);
    t[p].x = t[t[p].l].x + t[t[p].r].x;
}
 
void change(int p,int l,int r,int L,int R)
{
    int mid = l + r >> 1;
    if(l == L && r == R)
    {
        t[p].x += (r - l + 1) * d;
        t[p].mark += d;
        return;
    }
    if(t[p].mark)
    {
        t[t[p].l].x += (mid - l + 1) * t[p].mark;
        t[t[p].r].x += (r - mid) * t[p].mark;
        t[t[p].l].mark += t[p].mark;
        t[t[p].r].mark += t[p].mark;
        t[p].mark = 0;
    }
    if(R <= mid)
        change(t[p].l,l,mid,L,R);
    else if(L > mid)
        change(t[p].r,mid+1,r,L,R);
    else
    {
        change(t[p].l,l,mid,L,mid);
        change(t[p].r,mid+1,r,mid+1,R);
    }
    t[p].x = t[t[p].l].x + t[t[p].r].x;
}
 
long long getans(int p,int l,int r,int L,int R)
{
    int mid = l + r >> 1;
    if(l == L && r == R)
    {
        return t[p].x;
    }
    if(t[p].mark)
    {
        t[t[p].l].x += (mid - l + 1) * t[p].mark;
        t[t[p].r].x += (r - mid) * t[p].mark;
        t[t[p].l].mark += t[p].mark;
        t[t[p].r].mark += t[p].mark;
        t[p].mark = 0;
    }
    if(R <= mid)
        return getans(t[p].l,l,mid,L,R);
    else if(L > mid)
        return getans(t[p].r,mid+1,r,L,R);
    return getans(t[p].l,l,mid,L,mid) + getans(t[p].r,mid+1,r,mid+1,R);
}
 
void updata_node(int x)
{
    change(0,1,n,pos[x],pos[x]);
}
 
void updata_root(int x)
{
    change(0,1,n,pos[x],bot[x]);
}
 
long long query(int x)
{
    long long ans = 0;
    int fx = top[x];
    while(fx != 1)
    {
        ans += getans(0,1,n,pos[fx],pos[x]);
        x = fa[fx];
        fx = top[x];
    }
    ans += getans(0,1,n,pos[fx],pos[x]);
    return ans;
}
 
int main()
{
    scanf("%d%d",&n,&m);
    for(int i = 1; i <= n; i++)
        scanf("%lld",&a[i]);
    for(int i = 1; i < n; i++)
    {
        scanf("%d%d",&u,&v);
        build_edge(u,v);
        build_edge(v,u);
    }
    memset(vis,0,sizeof(vis));
    dfs1(1);
    memset(vis,0,sizeof(vis));
    dfs2(1);
    build(0,1,n);
    int c,x;
    while(m--)
    {
        scanf("%d",&c);
        if(c == 1)
        {
            scanf("%d%lld",&x,&d);
            updata_node(x);
        }
        if(c == 2)
        {
            scanf("%d%lld",&x,&d);
            updata_root(x);
        }
        if(c == 3)
        {
            scanf("%d",&x);
            printf("%lld\n",query(x));
        }
    }
    return 0;
}

你可能感兴趣的:(bzoj,数据结构——树链剖分)