广东工业大学2016校赛决赛 A: Krito的讨伐

Problem A: Krito的讨伐

Description

Krito终于干掉了99层的boss，来到了第100层。第100层可以表示成一颗树，这棵树有n个节点（编号从0到n-1），树上每一个节点可能有很多只怪物。 Krito现在在0号节点，现在它想要区清除这一层所有的怪物。他现在有atk大小的攻击力。只有当你的攻击力大于这只怪物的防御力时，你才可以打败他，同时每打败只怪物,你会获得一定的攻击力加成。一个节点可能存在着不止一只怪兽，你要打败这个节点的所有怪物才能可以从这个节点通过，请问他能不能完成这个任务？注意：不要求一次性杀光一个节点里面的所有怪物。

相关知识： 摘自维基百科 在计算机科学中，树（英语：tree）是一种抽象资料型别（ADT）或是实作这种抽象资料型别的数据结构，用来模拟具树状结构性质的资料集合。它是由n（n>=1）个有限节点组成一个具有层次关系的集合。把它叫做“树”是因为它看起来像一棵倒挂的树，也就是说它是根朝上，而叶朝下的。它具有以下的特点： 1.每个节点有零个或多个子节点； 2.没有父节点的节点称为根节点； 3.每一个非根节点有且只有一个父节点； 4.除了根节点外，每个子节点可以分为多个不相交的子树；

Input

第1行：一个数T，表示有T个测试样例（0<=T<=50) ,接下来有T个测试样例

对于每一个测试样例：

第1行：两个整数n，m表示这棵树有n个节点，m只怪兽(0<=n<=1000 ,0<=m <=100)

第2至n-1行： 两个整数u，v表示编号为u，v之间的节点有一条无向边，题目保证不会成环。（0<=u,v<n , u!=v)

第3行： 一个整数atk，表示Krito的初始化攻击力(0<=atk<=100)

第4至3+m行：两个整数id，def，add_atk，表示在编号为id的点上，有一只防御力为def的怪物,打败后可以增加add_atk点的攻击力。(0<=add_atk,def<=100)

Output

对于每一个测试样例，如果Krito能够清除所有的怪物，则输出“Oh yes.” 否则，输出“Good Good Study,Day Day Up.”

Sample Input

15 20 10 22 32 4113 10 21 11 0

Sample Output

Oh yes.

题意：告诉一颗树，然后一些结点，每个结点中可能有一些怪，每个怪都有一定防御力，开始攻击力有初始值，当攻击力大于怪的防御力时候可以打死怪，然后获得一定攻击力增长，只有打死一个结点的所有怪之后才能通过该点继续扩展，问能够消灭所有怪。

思路:从根节点开始，每次按顺序扩展，把可以扩展到的怪都加到优先队列，然后把能打的怪都打死，一次扩展相邻结点

#include <iostream>
#include <string>
#include <cstring>
#include <stdio.h>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>

using namespace std;

int n,m;
int u,v;
int atk;
int num[2009];
int vis[2009];

struct Node
{
int id,def,add;
Node() {};
Node(int a,int b,int c)
{
id=a;
def=b;
add=c;
}
int operator<(const Node &n)const
{
return def>n.def;
}
};

vector<int>vt[2009];
vector<Node>guai[2009];

priority_queue< Node >pq;

void dfs(int x)
{
if(num[x]!=0)
{
for(int i=0; i<guai[x].size(); i++)
{
pq.push(guai[x][i]);
}
guai[x].clear(); //加入一次后要清空，若再进入会重新加
return;
}

for(int i=0; i<vt[x].size(); i++) //当前节点没有怪，则扩展到其他节点
{
if(vis[vt[x][i]]==0)
{
vis[vt[x][i]]=1;
dfs(vt[x][i]);
}
}
}

int fun()
{
while(!pq.empty()) pq.pop(); //每次开始事要先清空

int cnt=0;
vis[cnt]=1;
dfs(cnt);

while(!pq.empty())
{
Node tmp=pq.top();
pq.pop();
if(atk<=tmp.def)
return 0;

atk+=tmp.add;
cnt=tmp.id;
num[cnt]--;
dfs(cnt);
}
return 1;
}

int main()
{
int T;

scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);

for(int i=0; i<=n; i++)
{
vt[i].clear();
guai[i].clear();
vis[i]=0;
num[i]=0;
}

// memset(vis,0,sizeof vis);
// memset(num,0,sizeof num);

for(int i=0; i<n-1; i++)
{
scanf("%d%d",&u,&v);
vt[u].push_back(v);
vt[v].push_back(u);
}
scanf("%d",&atk);
int a,b,c;
for(int i=0; i<m; i++)
{
scanf("%d%d%d",&a,&b,&c);
num[a]++; //当前该节点怪的数量
guai[a].push_back(Node(a,b,c)); //当前节点怪的集合
}

if(fun()==0)
puts("Good Good Study,Day Day Up.");
else
puts("Oh yes.");

}
return 0;
}