众所周知,桂木桂马是攻略之神,开启攻略之神模式后,他可以同时攻略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;
}