【BZOJ3252】攻略

Description
题目简述：树版[k取方格数]

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

输出一个整数表示答案
Sample Input
5 2
4 3 2 1 1
1 2
1 5
2 3
2 4
Sample Output
10
HINT

对于100%的数据，n<=200000，1<=场景价值<=2^31-1

Source

dfs序+线段树

显然每次选权值和最大的边
Source里写了标算是什么..
其实黄学长那种不用dfs序线段树的堆的解决方法也很好只是我没写过手写堆也不会用pbds的优先队列所以没去写

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define LL long long
#define MAXN 200010
#define GET (ch>='0'&&ch<='9')
#define lchild rt<<1,l,mid
#define rchild rt<<1|1,mid+1,r
#define ln rt<<1
#define rn rt<<1|1
using namespace std;
int n,k,top,Top;
int in[MAXN<<1],out[MAXN<<1],rmp[MAXN<<1],fa[MAXN],val[MAXN];
bool vis[MAXN<<1];
LL ans;
struct edge
{
    int to;
    edge *next;
}e[MAXN<<1],*prev[MAXN];
struct seg
{
    int l,r,pos;
    LL maxn,add;
}tree[MAXN<<3];
void In(int &x)
{
    x=0;char ch=getchar();int flag=1;
    while (!GET)    flag=ch=='-'?-1:1,ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();x*=flag;
}
void push_up(int rt)
{
    if (tree[ln].maxn<tree[rn].maxn)    tree[rt].pos=tree[rn].pos,tree[rt].maxn=tree[rn].maxn;
    else    tree[rt].pos=tree[ln].pos,tree[rt].maxn=tree[ln].maxn;
}
void push_down(int rt)
{
    if (tree[rt].add!=0&&tree[rt].l!=tree[rt].r)
    {
        tree[ln].maxn+=tree[rt].add;tree[ln].add+=tree[rt].add;
        tree[rn].maxn+=tree[rt].add;tree[rn].add+=tree[rt].add;tree[rt].add=0;
    }
}
void build(int rt=1,int l=1,int r=(n<<1))
{
    tree[rt].l=l;tree[rt].r=r;
    if (l==r)   {tree[rt].pos=l;return;}
    int mid=l+r>>1;build(lchild);build(rchild);push_up(rt);
}
void modify(int rt,int l,int r,int delta)
{
    int L=tree[rt].l,R=tree[rt].r,mid=L+R>>1;
    if (l<=L&&r>=R) {tree[rt].add+=delta;tree[rt].maxn+=delta;return;}push_down(rt);
    if (r<=mid) modify(ln,l,r,delta);
    else    if (l>mid)  modify(rn,l,r,delta);
    else    modify(ln,l,mid,delta),modify(rn,mid+1,r,delta);
    push_up(rt);
}
void insert(int u,int v)
{
    e[++top].to=v;e[top].next=prev[u];prev[u]=&e[top];
}
void dfs(int x)
{
    in[x]=++Top;rmp[Top]=x;
    for (edge *i=prev[x];i;i=i->next)
        if (i->to!=fa[x])   fa[i->to]=x,dfs(i->to);
    out[x]=++Top;
}
int main()
{
    In(n);In(k);int u,v;
    for (int i=1;i<=n;i++)  In(val[i]);
    for (int i=1;i<n;i++) In(u),In(v),insert(u,v);
    dfs(1);build();
    for (int i=1;i<=n;i++)  modify(1,in[i],n<<1,val[i]);
    for (int i=1;i<=n;i++)  modify(1,out[i],n<<1,-val[i]);
    while (k--)
    {
        int x=rmp[tree[1].pos];
        while (x&&!vis[x])  ans+=val[x],modify(1,in[x],n<<1,-val[x]),modify(1,out[x],n<<1,val[x]),vis[x]=1,x=fa[x];
    }
    cout<<ans<<endl;
}