[bzoj 3252]攻略

众所周知，桂木桂马是攻略之神，开启攻略之神模式后，他可以同时攻略k部游戏。今天他得到了一款新游戏《XX半岛》，这款游戏有n个场景(scene)，某些场景可以通过不同的选择支到达其他场景。所有场景和选择支构成树状结构：开始游戏时在根节点（共通线），叶子节点为结局。每个场景有一个价值，现在桂马开启攻略之神模式，同时攻略k次该游戏，问他观赏到的场景的价值和最大是多少（同一场景观看多次是不能重复得到价值的）
“为什么你还没玩就知道每个场景的价值呢？”
“我已经看到结局了。”

这道题一开始看到之后xjb乱想，以为是道大难题。但是%了一下star_feel大佬之后，发现其实想太复杂了，关键还是对于树之类的做得太少了。我们一开始用贪心的思想，发现其实k次，每次都是取最长链，也就是到根节点的价值最大的链，看到这个就应该想到数据结构（线段树）来维护每个点到跟根节点的价值和（我真是太菜了）。那取了之后怎么修改呢，跑一次dfs序，让子树编号连续，为什么，因为我们发现当一个点的权值改变时，只会对它所在的子树有影响。
那之后就只剩下了一点，也是比较重要的点，要进行一点小优化，因为每次修改要跳整一条链，太慢了。那我们给每个修改过的点打上一个标记，再次询问到它时，就不管它，它的祖先也不管，为什么，因为我们是从下往上跳，如果它已经被访问过，那它的祖先也被访问过（每个点只有一个父亲）。
（我才不会告诉你我之所以决心做这题是因为我喜欢桂木桂马君呢）

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
using namespace std;
struct node
{
    int x,y,next;
}a[210000];int len,last[210000];
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
int id,bl[210000],br[210000],ys[210000],fa[210000];
long long s[210000],sum[210000];
void dfs(int x)
{
    bl[x]=++id;ys[id]=x;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa[x])
        {
            sum[y]=sum[x]+s[y];
            fa[y]=x;dfs(y);
        }
    }
    br[x]=id;
}
struct trnode
{
    int l,r,lc,rc,d;
    long long c,del;
}tr[410000];int trlen;
void bt(int l,int r)
{
    trlen++;int now=trlen;
    tr[now].l=l;tr[now].r=r;
    tr[now].lc=tr[now].rc=-1;
    tr[now].c=tr[now].del=0;
    if(l==r)tr[now].c=sum[ys[l]],tr[now].d=ys[l];
    else
    {
        int mid=(l+r)/2;
        tr[now].lc=trlen+1;bt(l,mid);
        tr[now].rc=trlen+1;bt(mid+1,r);
        tr[now].c=max(tr[tr[now].lc].c,tr[tr[now].rc].c);
        if(tr[now].c==tr[tr[now].lc].c)tr[now].d=tr[tr[now].lc].d;
        else tr[now].d=tr[tr[now].rc].d;
    }
}
void update(int now)
{
    int lc=tr[now].lc,rc=tr[now].rc;
    tr[lc].c-=tr[now].del;tr[lc].del+=tr[now].del;
    tr[rc].c-=tr[now].del;tr[rc].del+=tr[now].del;
    tr[now].del=0;
}
void change(int now,int l,int r,int k)
{
    if(tr[now].l==l && tr[now].r==r){tr[now].c-=k;tr[now].del+=k;return ;}
    if(tr[now].del!=0)update(now);
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(r<=mid)change(lc,l,r,k);
    else if(mid+1<=l)change(rc,l,r,k);
    else change(lc,l,mid,k),change(rc,mid+1,r,k);
    tr[now].c=max(tr[lc].c,tr[rc].c);
    if(tr[now].c==tr[lc].c)tr[now].d=tr[lc].d;
    if(tr[now].c==tr[rc].c)tr[now].d=tr[rc].d;
}
bool v[210000];
void solve(int x)
{
    while(x!=0)
    {
        if(v[x]==false)return ;
        change(1,bl[x],br[x],s[x]);
        v[x]=false;
        x=fa[x];
    }
}
int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)scanf("%lld",&s[i]);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);
    }
    id=0;sum[1]=s[1];fa[1]=0;dfs(1);
    trlen=0;bt(1,id);
    memset(v,true,sizeof(v));
    long long ans=0;
    while(k--)
    {
        if(tr[1].c==0)break;
        ans+=tr[1].c;
        int wy=tr[1].d;
        solve(wy);
    }
    printf("%lld\n",ans);
    return 0;
}