bzoj 3252 攻略 长链剖分思想+贪心

攻略

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 889  Solved: 423
[Submit][Status][Discuss]

Description

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

Input

第一行两个正整数n,k
第二行n个正整数,表示每个场景的价值
以下n-1行,每行2个整数a,b,表示a场景有个选择支通向b场景(即a是b的父亲)
保证场景1为根节点
n<=200000,1<=场景价值<=2^31-1

Output

输出一个整数表示答案

Sample Input

5 2
4 3 2 1 1
1 2
1 5
2 3
2 4

Sample Output

10
 
其实有种比较好的算法,就是长链剖分放入队列,排个序就可以了。
然而配对堆实现了一下,比较麻烦,删除的东西比较多,但是复杂度是一的。
 
 #include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<queue>
#include<ext/pb_ds/priority_queue.hpp> #define ll long long
#define pa pair<ll,int>
#define N 200007
using namespace std;
inline int read()
{
int x=,f=;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-;ch=getchar();}
while(isdigit(ch)){x=(x<<)+(x<<)+ch-'';ch=getchar();}
return x*f;
} int n,K;
int val[N],last[N];
ll mx[N],ans;
int cnt,hed[N],rea[N],nxt[N];
__gnu_pbds::priority_queue<pa >::point_iterator id[N];
__gnu_pbds::priority_queue<pa >q; void add(int u,int v)
{
nxt[++cnt]=hed[u];
hed[u]=cnt;
rea[cnt]=v;
}
void dp(int x)
{
for(int i=hed[x];~i;i=nxt[i])
{
int v=rea[i];
dp(v);
mx[x]=max(mx[x],mx[v]);
}
mx[x]+=val[x];
id[x]=q.push(make_pair(mx[x],x));
}
void del(int x)
{
q.erase(id[x]);
for(int i=hed[x];~i;i=nxt[i])
{
int v=rea[i];
if(mx[v]==mx[x]-val[x])
{
del(v);
break;
}
}
}
int main()
{
memset(hed,-,sizeof(hed));
n=read(),K=read();
for(int i=;i<=n;i++)
val[i]=read();
for(int i=;i<n;i++)
{
int u=read(),v=read();
add(u,v);
}
dp();
for(int i=;i<=K&&!q.empty();i++)
{
int x=q.top().second;
ans+=mx[x];
del(x);
}
printf("%lld\n",ans);
}
上一篇:word----遇到问题-----word中插入的图片无法左对齐----格式按钮为灰色


下一篇:Head First 设计模式之工厂模式(Factory Pattern)