省选模拟赛Day4T2 攻略

Problem Problem Problem Problem Problem Problem Problem Problem 2 攻略 (game)(game)(game)(game)(game)(game)
【题目描述】 【题目描述】
一款游戏有 n个场景 (scene) (scene) ，某些场 景可以通过不同的选择支到达其他景。所有场和选择支构成树状结：开始游戏时在根节点 景。所有场和选择支构成树状结：开始游戏时在根节点 景。所有场和选择支构成树状结：开始游戏时在根节点 景。所有场和选择支构成树状结：开始游戏时在根节点 景。所有场和选择支构成树状结：开始游戏时在根节点 景。所有场和选择支构成树状结：开始游戏时在根节点 景。所有场和选择支构成树状结：开始游戏时在根节点 景。所有场和选择支构成树状结：开始游戏时在根节点 ，叶子节点为结局。 ，叶子节点为结局。 ，叶子节点为结局。 每个场景有一价值。名玩家 每个场景有一价值。名玩家 攻略 k次该游戏，问 次该游戏，问 该玩家能 观赏到的场景价 值 和最大是多少（同一场景观看次不能重复得到价）？
【输入格式】 【输入格式】
第一行两个正整数 n,k 。
第二行 n个正整数，表示每场景的价值 。
以下 n-1行,每行 2个整数 个整数 a,b ，表示 a场景有个选择支通向 场景有个选择支通向 b场景（即 场景（即 a是 b的父亲） 。
保证场景 1为根节点 。
【输出格式】 【输出格式】
输出一个整数表示答案 。
【样例输入】 【样例输入】
5 2
4 3 2 1
1 2
1 5
2 3
2 4
【样例输出】 【样例输出】
10
【数据范围】 【数据范围】
对于 20% 的数据， n<=1000 ，k<=2
对于 40% 的数据， n<=1000
对于 70% 的数据， n<=100000

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

我的方法是每次贪心的找最长链。找完后删除，将新产生的链塞进set继续找。。。。。。

感性理解一下，具体看代码：

ACcode:

#include<cstdio>
#include<algorithm>
#include<cctype>
#include<cstring>
#include<set>
#define LL long long
#define maxn 400005
using namespace std;

int n,k;
int info[maxn],to[maxn*2],Prev[maxn*2],cnt_e;
LL val[maxn],dp[maxn];

void Node(int u,int v)
{
	cnt_e++;
	Prev[cnt_e]=info[u],info[u]=cnt_e,to[cnt_e]=v;
}

int nxt[maxn],dep[maxn];
void dfs(int now,int ff)
{
	dep[now]=dep[ff]+1;
	dp[now]=val[now];
	for(int i=info[now];i;i=Prev[i])
		if(to[i]!=ff)
		{
			dfs(to[i],now);
			if(dp[to[i]]+val[now]>=dp[now])
			{
				dp[now]=dp[to[i]]+val[now];
				nxt[now]=to[i];
			}
		}
}

struct node
{
	int id;
	node(){}node(int id):id(id){}
	bool operator <(const node &nxt)const
	{
		return dp[id]==dp[nxt.id]?id<nxt.id:dp[id]>dp[nxt.id];
	}
};

int main()
{
	
	freopen("game.in","r",stdin);
	freopen("game.out","w",stdout);
	
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++) scanf("%lld",&val[i]);
	for(int i=1,u,v;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		Node(u,v);
		Node(v,u);
	}
	dfs(1,0);
	set<node>st;
	st.insert(node(1));
	LL ans=0;
	for(int i=1,now;i<=k;i++)
	{
		if(st.empty()) break;
		now=(*st.begin()).id;
		st.erase(st.begin());
		ans+=dp[now];
		for(int j=now;j;j=nxt[j])
		{
			for(int k=info[j];k;k=Prev[k])
				if(dep[to[k]]>dep[j] && to[k]!=nxt[j])
					st.insert(node(to[k]));
		}
	}
	
	printf("%lld\n",ans);
}