GDUT 2016 A Krito的讨伐（BFS + 优先队列）

最新推荐文章于 2021-03-06 22:19:13 发布

原创最新推荐文章于 2021-03-06 22:19:13 发布 · 468 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#bfs

ACM刷题专栏收录该内容

91 篇文章

订阅专栏

在挑战第100层时，Krito必须制定策略清除所有节点的怪物，仅需击败每个节点的怪物即可继续前进。通过广度优先搜索(BFS)并维护攻击力与怪物防御力的关系，实现清除所有怪物的目标。

Description

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

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

Output

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

Sample Input

1
5 2
0 1
0 2
2 3
2 4
11
3 10 2
1 11 0

Sample Output

Oh yes.

题解

用vector构造树，gra[i][j]表示i到j有一条路；
用vector储存怪物结点，mon[i][j]表示i的第j个怪物；从起点开始广搜；
用priority_queue维护一颗树，当结点def< k时杀怪，更新k；当结点def == -1时，往下搜索；

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <vector>
using namespace std;

class Node {
public:
  int id, def, atk;
  Node(int a, int c, int d) {
    id = a, def = c, atk = d;
  }
  bool operator < (const Node &a) const {
    return def == a.def ? atk < a.atk : def > a.def;
  }
};

const int MAXN = 1000;
int t, n, m, k;
int cnt[MAXN + 10];
vector<int> gra[MAXN + 10];
vector<Node> mon[MAXN + 10];

void Init() {
  memset(cnt, 0, sizeof(cnt));
  for (int i = 0; i <= MAXN; ++i) {
    gra[i].clear();
    mon[i].clear();
  }
}

bool Bfs() {
  priority_queue<Node> q;
  for (int i = 0; i < mon[0].size(); ++i) {
    q.push(mon[0][i]);
  }
  if (cnt[0] == 0) {
    q.push(Node(0, -1, 0));
  }

  while (!q.empty()) {
    Node head = q.top(); q.pop();
    if (head.def == -1) {
      for (int i = 0; i < gra[head.id].size(); ++i) {
        for (int j = 0; j < mon[gra[head.id][i]].size(); ++j) {
          q.push(mon[gra[head.id][i]][j]);
        }
        if (cnt[gra[head.id][i]] == 0) {
          q.push(Node(gra[head.id][i], -1, 0));
        }
      }

    } else if (head.def < k) {
      k += head.atk;
      if (--cnt[head.id] == 0) {
        q.push(Node(head.id, -1, 0));
      }

    } else {
      return false;
    }
  }
  return true;
}

int main() {
  scanf("%d", &t);
  while (t--) {
    Init();
    scanf("%d%d", &n, &m);
    for (int i = 1, a, b; i < n; ++i) {
      scanf("%d%d", &a, &b);
      gra[a].push_back(b);
    }
    scanf("%d", &k);
    for (int i = 0, id, def, atk; i < m; ++i) {
      scanf("%d%d%d", &id, &def, &atk);
      mon[id].push_back(Node(id, def, atk));
      ++cnt[id];
    }
    printf("%s\n", Bfs() ? "Oh yes." : "Good Good Study,Day Day Up.");
  }
  return 0;
}