【Leetcode】1298. Maximum Candies You Can Get from Boxes

题目地址：

https://leetcode.com/problems/maximum-candies-you-can-get-from-boxes/description/

给定$n$种盒子，种类编号为$0\sim n-1$，再给定$4$个长$n$的数组（或数组的数组），$s[i]$表示当前编号$i$的盒子是开的还是关的（$1$表示开，$0$表示关），$c[i]$表示$i$号盒子里的糖果数，$k[i]$表示$i$号盒子里含哪些盒子的钥匙，$b[i]$表示$i$号盒子里还有哪些编号的盒子（注意这里编号为$i$的盒子需要视为一种种类为$i$的盒子）。给定一个初始的已经开了的盒子列表，问最多能得多少糖果。

可以维护三个集合，可以开的盒子编号，暂时不能开的盒子编号和钥匙编号，每次挑一个开的盒子，将其糖果计入计数，然后看一下它含的钥匙，如果对于当前钥匙，恰好有个暂时不能开的盒子编号与之契合，则将这个盒子加入可以开的盒子集合；接着看其所含的盒子，将可以开的盒子放入对应集合，将暂时开不了的盒子也放入对应集合。代码如下：

class Solution {
 public:
  int maxCandies(vector<int>& status, vector<int>& candies,
                 vector<vector<int>>& keys, vector<vector<int>>& containedBoxes,
                 vector<int>& initialBoxes) {
    unordered_set<int> s_open, s_closed, s_keys;
    for (int x : initialBoxes) {
      if (status[x]) s_open.insert(x);
      else s_closed.insert(x);
    }

    int res = 0;
    while (s_open.size()) {
      auto it = s_open.begin();
      int x = *it;
      s_open.erase(it);
      res += candies[x];
      for (int k : keys[x]) {
        if (s_closed.count(k)) {
          s_open.insert(k);
          s_closed.erase(k);
        } else s_keys.insert(k);
      }

      for (int y : containedBoxes[x]) {
        if (status[y]) s_open.insert(y);
        else if (s_keys.count(y)) {
          s_open.insert(y);
          s_keys.erase(y);
        } else s_closed.insert(y);
      }
    }
    
    return res;
  }
};

时间复杂度与数据有关。