51nod1462 樹據結構(樹鏈剖分+線段樹)

  這題很久以前就被學長安利了...一直沒寫珍藏在收藏夾一個鮮爲人知的角落233ios

  這題怎麼作...咱們來數形結合,橫座標爲$t_i$被加的次數(可看做時間$t$),縱座標爲$v_i$,那麼$t_i$實際上就是階梯圖形的面積。ide

  上圖是父親節點和子節點合併後的樣子,陰影部分爲答案即$t_i$,顯然能夠經過$deltat*deltav-deltatv$來獲得。spa

  信息下傳的時候兒子的$deltatv$加上父親的$deltatv$後還要加上一個長方體的面積即$deltat_{son}*deltav_{fa}$,而後兒子的$deltat$和$deltav$都加上父親的就好了。3d

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=200010, inf=1e9;
struct poi{int too, pre;}e[maxn<<1];
struct tjm{ll deltat, deltav, deltatv;}tree[maxn<<2];
int n, m, tott, tot, ty, x, y, z;
int fa[maxn], d[maxn], size[maxn], top[maxn], w[maxn], last[maxn], son[maxn];
char buf[8000000], *ptr=buf-1;
inline void read(int &k)
{
    int f=1; k=0; char c=*++ptr;
    while(c<'0' || c>'9') c=='-'&&(f=-1), c=*++ptr;
    while(c<='9' && c>='0') k=k*10+c-'0', c=*++ptr;
    k*=f;    
} 
inline void add(int x, int y){e[++tot]=(poi){y, last[x]}; last[x]=tot;}
void dfs1(int x)
{
    size[x]=1;
    for(int i=last[x], too;i;i=e[i].pre)
    {
        d[too=e[i].too]=d[x]+1;
        dfs1(too);
        size[x]+=size[too];
        if(size[too]>size[son[x]]) son[x]=too;
    }
}
void dfs2(int x, int tp)
{
    top[x]=tp; w[x]=++tott; 
    if(son[x]) dfs2(son[x], tp);
    for(int i=last[x], too;i;i=e[i].pre)
    if((too=e[i].too)!=son[x]) dfs2(too, too);
}
inline void addone(int x, ll deltat, ll deltav, ll deltatv)
{
    tree[x].deltatv+=deltatv+deltav*tree[x].deltat;
    tree[x].deltat+=deltat;
    tree[x].deltav+=deltav;
}
inline void down(int x)
{
    addone(x<<1, tree[x].deltat, tree[x].deltav, tree[x].deltatv);
    addone(x<<1|1, tree[x].deltat, tree[x].deltav, tree[x].deltatv);
    tree[x].deltat=tree[x].deltav=tree[x].deltatv=0;
}
void update(int x, int l, int r, int cl, int cr, int delta, int ty)
{
    if(cl<=l && r<=cr)
    {
        if(ty) tree[x].deltat+=delta;
        else tree[x].deltatv+=tree[x].deltat*delta, tree[x].deltav+=delta;
        return;
    }
    down(x);
    int mid=(l+r)>>1;
    if(cl<=mid) update(x<<1, l, mid, cl, cr, delta, ty);
    if(cr>mid) update(x<<1|1, mid+1, r, cl, cr, delta, ty);
}
ll query(int x, int l, int r, int cx)
{
    if(l==r) return tree[x].deltat*tree[x].deltav-tree[x].deltatv;
    down(x);
    int mid=(l+r)>>1;
    if(cx<=mid) return query(x<<1, l, mid, cx);
    else return query(x<<1|1, mid+1, r, cx);
}
inline void solve(int x, int y, int delta, int ty)
{
    int f1=top[x], f2=top[y];
    while(f1!=f2)
    {
        if(d[f1]<d[f2]) swap(x, y), swap(f1, f2);
        update(1, 1, n, w[f1], w[x], delta, ty);
        x=fa[f1]; f1=top[x];
    }
    if(d[x]<d[y]) swap(x, y);
    update(1, 1, n, w[y], w[x], delta, ty);
}
int main()
{
    fread(buf, 1, sizeof(buf), stdin); read(n);
    for(int i=2;i<=n;i++) read(fa[i]), add(fa[i], i);
    dfs1(1); dfs2(1, 1); read(m);
    for(int i=1;i<=m;i++) read(ty), read(x), read(y), solve(1, x, y, ty-1);
    for(int i=1;i<=n;i++) printf("%lld\n", query(1, 1, n, w[i]));
}
View Code

  若是這題要求區間和呢?由於一個區間內可能有不少個階梯形,可是上傳的時候都是接上一樣的一段父親的階梯,因此這個貢獻仍是能夠計算的,只須要多記錄一下區間內的$sumv,sumt$就很好求了。code

相關文章
相關標籤/搜索