【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

题解
dfs序+线段树。
每次贪心地取叶子结点到根路径和最大的。然后利用dfs序+线段树修改。
（数组开小wa了好几遍。。）
代码

#include<bits/stdc++.h>
#define ll long long
#define N 200005
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int tot,tim,n,k;
ll ans,lazy[4*N],mx[N],val[N];
int ret[2*N],Next[2*N],Head[N],fa[N];
int l[N],r[N],a[N],p[N];
struct node{ll mx,id;}t[4*N];
bool flag[N];
inline void ins(int u,int v)
{
    ret[++tot]=v;
    Next[tot]=Head[u];Head[u]=tot;
}
void dfs(int u)
{
    l[u]=++tim;mx[tim]=val[u];p[tim]=u;
    for (int i=Head[u];i;i=Next[i])
    {
        int v=ret[i];
        if (v!=fa[u])
        {
            fa[v]=u;val[v]+=val[u];
            dfs(v);
        }
    }
    r[u]=tim;
}
bool operator<(node a,node b)
{
    return a.mx<b.mx;
}
inline void update(int k)
{
    t[k]=max(t[k<<1],t[k<<1|1]);
}
inline void pushdown(int k,int l,int r)
{
    if (l==r) return;
    t[k<<1].mx+=lazy[k];
    t[k<<1|1].mx+=lazy[k];
    lazy[k<<1]+=lazy[k];lazy[k<<1|1]+=lazy[k]; 
    lazy[k]=0;
}
void build(int k,int l,int r)
{
    if (l==r)
    {
        t[k].mx=mx[l];
        t[k].id=p[l];
        return;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);build(k<<1|1,mid+1,r);
    update(k);
}
void modify(int k,int l,int r,int x,int y,int z)
{
    //cout<<k<<" "<<l<<" "<<r<<" "<<x<<" "<<y<<endl;
    if (lazy[k]) pushdown(k,l,r);
    if (l==x&&r==y)
    {
        lazy[k]+=z;
        t[k].mx+=z;
        return;
    }
    int mid=(l+r)>>1;
    if (y<=mid) modify(k<<1,l,mid,x,y,z);
    else if (x>mid) modify(k<<1|1,mid+1,r,x,y,z);
    else modify(k<<1,l,mid,x,mid,z),modify(k<<1|1,mid+1,r,mid+1,y,z);
    update(k);
}
void del(int u)
{
    if (flag[u]) return;
    flag[u]=1;
    modify(1,1,n,l[u],r[u],-a[u]);
    del(fa[u]);
}
int main()
{
    n=read();k=read();
    for (int i=1;i<=n;i++) val[i]=a[i]=read();
    for (int i=1;i<n;i++)
    {
        int u=read(),v=read();
        ins(u,v);ins(v,u);
    }
    dfs(1);
    build(1,1,n);flag[0]=1;
    for (int i=1;i<=k;i++)
    {
        if (flag[t[1].id]) break;
        //cout<<t[1].mx<<" "<<t[1].id<<endl;
        ans+=t[1].mx;
        del(t[1].id);
    }
    printf("%lld",ans);
    return 0;
}