广东工业大学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.

HINT

解题思路:从根节点dfs依次把所有可达的怪物加到一个优先队列，每次从队列中取出一个判断是否满足题意，如果不满足则退出，满足则增加攻击力，并把该节点上的怪物数量减一，并从该节点开始dfs把所有可达怪物加到优先队列即可。

AC代码如下:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
#include <vector>
#include <list>
#include <map>
#include <set>
#include <queue>
#include <cctype>
#include <cstdlib>
#include <sstream>
#define sqr(x) ((x)*(x))
#define lson le,mid,rt<<1
#define rson mid+1,ri,rt<<1|1
#define inf 0x3f3f3f3f
#define eps 1e-6
typedef long long ll;
using namespace std;
const double PI = 4.0*atan(1.0);
const int maxn = 1010;

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

vector<int> ve[maxn];
vector<node> vn[maxn];
int atk, m, n, cnt[maxn], vis[maxn];
priority_queue<node> pq;

void dfs(int root){
if(cnt[root] != 0){
for(int i = 0;i < vn[root].size();++i)
pq.push(vn[root][i]);
vn[root].clear();
return;
}
for(int i = 0;i < ve[root].size();++i){
if(!vis[ve[root][i]]){
vis[ve[root][i]] = 1;
dfs(ve[root][i]);
}
}
}

bool solve(){
int root = 0;
while(!pq.empty())	pq.pop();
vis[root] = 1;
dfs(root);
while(!pq.empty()){
node t = pq.top();
pq.pop();
if(atk <= t.def){
return 0;
}
atk += t.add_atk;
root = t.id;
cnt[root]--;
dfs(root);
}
return 1;
}

int main(){
int T,a,b,c;
scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&m);
for(int i = 0;i <= n;++i){
ve[i].clear();
vn[i].clear();
}
for(int i = 0;i < n-1;++i){
scanf("%d%d",&a,&b);
ve[b].push_back(a);
ve[a].push_back(b);
}
scanf("%d",&atk);
memset(cnt,0,sizeof(cnt));
memset(vis,0,sizeof(vis));
for(int i = 0;i < m;++i){
scanf("%d%d%d",&a,&b,&c);
vn[a].push_back(node(b,c,a));
cnt[a]++;
}
printf(solve()?"Oh yes.\n":"Good Good Study,Day Day Up.\n");
}
return 0;
}