GDUT Krito的讨伐(bfs&&优先队列)

1
2
3
4
5
6
7
8
9
10
11
12
第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)

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

1
2
3
4
5
6
7
8
9
10
1
5 2
0 1
0 2
2 3
2 4
11
3 10 2
1 11 0

1
2
Oh yes.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#include <stdio.h>
#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
#include <stack>
#include <queue>
using namespace std;
#define LL long long

struct Node
{
    int id, def, add;
    friend bool operator < (Node a, Node b)
    {
        return a.def > b.def;
    }
};

bool g[1009][1009];
int cnt[1009];
vector<Node > v[1009];
bool vis[1009];
int n, m, k;

void init()
{
    memset(cnt, 0, sizeof(cnt));
    memset(g, 0, sizeof(g));
    memset(vis, 0, sizeof(vis));
    for(int i=0; i<n; i++)
        v[i].clear();
}

bool bfs()
{
    priority_queue<Node> q;
    for(int i=0; i<v[0].size(); i++)
        q.push(v[0][i]);

    if(cnt[0] == 0)
    {
        Node t = {0, -1, 0};
        q.push(t);
    }
    vis[0] = 1;
    while(!q.empty())
    {
        Node t = q.top();
        q.pop();

        if(t.def == -1)
        {
            for(int i=0; i<n; i++)
            {
                if(!vis[i] && g[t.id][i] == 1)
                {
                    vis[i] = 1;
                    for(int j=0; j<cnt[i]; j++)
                        q.push(v[i][j]);
                    if(cnt[i] == 0)
                    {
                        Node x = {i, -1, 0};
                        q.push(x);
                    }
                }
            }

            continue;
        }
        if(t.def < k)
        {
            k += t.add;
            if(--cnt[t.id] == 0)
            {

                t.def = -1;
                q.push(t);
            }
        }
        else
            return false;
    }
    return true;
}


int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        init();
        scanf("%d%d", &n, &m);
        for(int i=1; i<n; i++)
        {
            int a, b;
            scanf("%d%d", &a, &b);
            g[a][b] = g[b][a] = 1;
        }

        scanf("%d", &k);

        for(int i=0; i<m; i++)
        {
            int a, b, c;
            scanf("%d%d%d", &a, &b, &c);
            Node t={a, b, c};
            v[a].push_back(t);
            cnt[a]++;
        }

        if(bfs())
            printf("Oh yes.n");
        else
            printf("Good Good Study,Day Day Up.n");
    }
    return 0;
}