您可以从C ++盒子中获得的最大糖果
假设我们有n个盒子,这里每个盒子都以[状态,糖果,键,containedBoxes]之类的格式给出,但有一些约束-
status[i]:打开box[i]时为1,关闭box[i]时为0。
candies[i]:是box[i]中的糖果数。
keys[i]:是一个数组,其中包含我们可以使用box[i]中的键打开的盒子的索引。
containsBoxes[i]:一个数组,其中包含在box[i]中找到的盒子的索引。
我们将从initialBoxes数组中给出的一些盒子开始。我们可以将所有糖果放入任何打开的盒子中,并且可以使用其中的键来打开新盒子,也可以使用在其中找到的盒子。
我们必须找到遵循上述规则可获得的最大糖果数量。
因此,如果输入的状态为[1,0,1,0],糖果=[8,6,5,101],键=[[],[],[1],[]],则containsBoxes=[[1,2],[3],[],[]],initialBoxes=[0],那么输出将为19。这是因为最初给我们提供了框0。我们将在其中和框中找到8个糖果。1和2。框1没有打开,我们没有键,因此我们将打开框2。在框2中,我们将找到5个糖果和框1的键。在框1中,您将找到6个糖果和框3,但找不到框3的键,因此框3将保持关闭状态。收集的糖果总数=8+5+6=19个糖果。
为了解决这个问题,我们将遵循以下步骤-
回答:=0
定义一个队列q
定义访问,打开,hasKey的集合
对于初始化i:=0,当i<ib的大小时,更新(将i增加1),执行-
ans:=ans+cnt[x]
将x插入打开
将x插入q
x:=ib[i]
将x插入已访问
如果st[x]与1相同,则-
当(不是q为空)时,执行-
x:=cb[curr,i]
将x插入已访问
如果x未打开且x在hasKey或st[x]与1相同,则-
将x插入打开
ans:=ans+cnt[x]
将x插入q
x:=k[curr,i]
将x插入hasKey
如果x未打开且x未访问,则-
ans:=ans+cnt[x]
将x插入q
将x插入打开
curr:=q的第一个元素
从q删除元素
对于初始化i:=0,当i<k[curr]的大小时,更新(将i增加1),执行-
对于初始化i:=0,当i<cb[curr]的大小时,更新(将i增加1),执行-
返回ans
让我们看下面的实现以更好地理解-
示例
#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
int maxCandies(vector<int>& st, vector<int>& cnt,
vector<vector<int>>& k, vector<vector<int>>& cb, vector<int>& ib) {
int ans = 0;
queue<int> q;
set<int> visited;
set<int> opened;
set<int> hasKey;
for (int i = 0; i < ib.size(); i++) {
int x = ib[i];
visited.insert(x);
if (st[x] == 1) {
ans += cnt[x];
opened.insert(x);
q.push(x);
}
}
while (!q.empty()) {
int curr = q.front();
q.pop();
for (int i = 0; i < k[curr].size(); i++) {
int x = k[curr][i];
hasKey.insert(x);
if (!opened.count(x) && visited.count(x)) {
ans += cnt[x];
q.push(x);
opened.insert(x);
}
}
for (int i = 0; i < cb[curr].size(); i++) {
int x = cb[curr][i];
visited.insert(x);
if (!opened.count(x) && (hasKey.count(x) || st[x] ==
1)) {
opened.insert(x);
ans += cnt[x];
q.push(x);
}
}
}
return ans;
}
};
main(){
Solution ob;
vector<int> v = {1,0,1,0}, v1 = {8,6,5,101}, v2 = {0};
vector<vector<int>> v3 = {{},{},{1},{}}, v4 = {{1,2},{3},{0},{}};
cout << (ob.maxCandies(v, v1, v3, v4, v2));
}输入项
{1,0,1,0}, {8,6,5,101}, {{},{},{1},{}}, {{1,2},{3},{0},{}}, {0}输出结果
19