1. 程式人生 > >Vasya and a Tree CodeForces

Vasya and a Tree CodeForces

m次修改 v k x代表將v節點之下k層的節點全都加x 最後統一查詢 即輸出每個節點的值

因為修改過程無查詢 考慮離線做法 因為是按層更新 一開始考慮維護bfs序 但k不是一個小常數 並且也無法對層數差分 只能換思路

很多資料結構問題中 區間更新點查詢和區間查詢點更新是可以相互轉換的 在這裡我就可以對每一個點 查詢會對它造成影響的所有更新的權值和 對於節點v來說 有影響的只是v到1這條鏈上的更新 並且層數還要大於等於v的層數 這樣就可以先樹剖出個dfs序 然後靜態主席樹搞個字首和 在1到v這條鏈上差分即可

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
typedef long long ll;
const int maxn=3e5+10;

struct node0
{
    ll val;
    int k;
};

struct node1
{
    int v;
    int next;
};

struct node2
{
    int l;
    int r;
    ll val;
};

vector <node0> pre[maxn];
node1 edge[2*maxn];
node2 tree[40*maxn];
ll ans[maxn];
int first[maxn],fa[maxn],deep[maxn],sum[maxn],son[maxn],top[maxn],mp1[maxn],mp2[maxn],root[maxn];
int n,q,num;

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
    return;
}

void dfsI(int cur)
{
    int i,v;
    sum[cur]=1,son[cur]=-1;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa[cur]){
            fa[v]=cur,deep[v]=deep[cur]+1;
            dfsI(v);
            sum[cur]+=sum[v];
            if(son[cur]==-1||sum[son[cur]]<sum[v]){
                son[cur]=v;
            }
        }
    }
}

void dfsII(int cur,int tp)
{
    int i,v;
    num++;
    top[cur]=tp,mp1[cur]=num,mp2[num]=cur;
    if(son[cur]==-1) return;
    dfsII(son[cur],tp);
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa[cur]&&v!=son[cur]){
            dfsII(v,v);
        }
    }
}

int build(int l,int r)
{
    int cur,m;
    cur=num++;
    tree[cur].l=0,tree[cur].r=0,tree[cur].val=0;
    if(l==r) return cur;
    m=(l+r)/2;
    tree[cur].l=build(l,m);
    tree[cur].r=build(m+1,r);
    return cur;
}

int update(int rot,int tar,ll val,int l,int r)
{
    int cur,m;
    cur=num++;
    tree[cur]=tree[rot];
    tree[cur].val+=val;
    if(l==r) return cur;
    m=(l+r)/2;
    if(tar<=m) tree[cur].l=update(tree[rot].l,tar,val,l,m);
    else tree[cur].r=update(tree[rot].r,tar,val,m+1,r);
    return cur;
}

ll query(int lrot,int rrot,int pl,int pr,int l,int r)
{
    ll res;
    int m;
    if(pl<=l&&r<=pr) return tree[rrot].val-tree[lrot].val;
    res=0,m=(l+r)/2;
    if(pl<=m) res+=query(tree[lrot].l,tree[rrot].l,pl,pr,l,m);
    if(pr>m) res+=query(tree[lrot].r,tree[rrot].r,pl,pr,m+1,r);
    return res;
}

ll solve(int u,int v,int tar)
{
    ll res;
    //printf("***%d***\n",tar);
    res=0;
    while(top[u]!=top[v]){
        if(deep[top[u]]<deep[top[v]]) swap(u,v);
        //printf("*%d %d*\n",top[u],u);
        res+=query(root[mp1[top[u]]-1],root[mp1[u]],deep[tar],n,1,n);
        u=fa[top[u]];
    }
    if(deep[u]<deep[v]) swap(u,v);
    //printf("*%d %d*\n",v,u);
    res+=query(root[mp1[v]-1],root[mp1[u]],deep[tar],n,1,n);
    return res;
}

void show(int rot,int l,int r)
{
    int m;
    printf("*%lld %d %d*\n",tree[rot].val,l,r);
    if(l==r) return;
    m=(l+r)/2;
    show(tree[rot].l,l,m);
    show(tree[rot].r,m+1,r);
}

int main()
{
    node0 tmp;
    int i,j,u,v,flag;
    scanf("%d",&n);
    memset(first,-1,sizeof(first));
    num=0;
    for(i=1;i<=n-1;i++){
        scanf("%d%d",&u,&v);
        addedge(u,v);
        addedge(v,u);
    }
    fa[1]=0,deep[1]=1;
    dfsI(1);
    num=0;
    dfsII(1,1);
    scanf("%d",&q);
    while(q--){
        scanf("%d%d%lld",&v,&tmp.k,&tmp.val);
        pre[mp1[v]].pb(tmp);
    }
    num=0;
    root[0]=build(1,n);
    for(i=1;i<=n;i++){
        flag=0;
        //printf("***%d %d***\n",i,deep[i]);
        for(j=0;j<pre[i].size();j++){
            //printf("*%d %lld*\n",pre[i][j].k,pre[i][j].val);
            if(flag==0){
                root[i]=update(root[i-1],min(n,deep[mp2[i]]+pre[i][j].k),pre[i][j].val,1,n);
                flag=1;
            }
            else{
                root[i]=update(root[i],min(n,deep[mp2[i]]+pre[i][j].k),pre[i][j].val,1,n);
            }
        }
        if(flag==0) root[i]=root[i-1];//
    }
    /*
    for(i=1;i<=n;i++){
        printf("***%d***\n",i);
        show(root[i],1,n);
    }
    */
    for(i=1;i<=n;i++){
        ans[i]=solve(1,i,i);
    }
    for(i=1;i<=n;i++) printf("%lld ",ans[i]);
    printf("\n");
    return 0;
}