給你 n 個盒子,每個盒子的格式爲 [status, candies, keys, containedBoxes] ,其中:
狀態字 status[i]:整數,如果 box[i] 是開的,那麼是 1 ,否則是 0 。
糖果數 candies[i]: 整數,表示 box[i] 中糖果的數目。
鑰匙 keys[i]:數組,表示你打開 box[i] 後,可以得到一些盒子的鑰匙,每個元素分別爲該鑰匙對應盒子的下標。
內含的盒子 containedBoxes[i]:整數,表示放在 box[i] 裏的盒子所對應的下標。
給你一個 initialBoxes 數組,表示你現在得到的盒子,你可以獲得裏面的糖果,也可以用盒子裏的鑰匙打開新的盒子,還可以繼續探索從這個盒子裏找到的其他盒子。
請你按照上述規則,返回可以獲得糖果的 最大數目 。
示例 1:
輸入:status = [1,0,1,0], candies = [7,5,4,100], keys = [[],[],[1],[]], containedBoxes = [[1,2],[3],[],[]], initialBoxes = [0]
輸出:16
解釋: 一開始你有盒子 0 。你將獲得它裏面的 7 個糖果和盒子 1 和 2。 盒子 1
目前狀態是關閉的,而且你還沒有對應它的鑰匙。所以你將會打開盒子 2 ,並得到裏面的 4 個糖果和盒子 1 的鑰匙。 在盒子 1 中,你會獲得
5 個糖果和盒子 3 ,但是你沒法獲得盒子 3 的鑰匙所以盒子 3 會保持關閉狀態。 你總共可以獲得的糖果數目 = 7 + 4 + 5 =
16 個。
示例 2:
輸入:status = [1,0,0,0,0,0], candies = [1,1,1,1,1,1], keys = [[1,2,3,4,5],[],[],[],[],[]], containedBoxes = [[1,2,3,4,5],[],[],[],[],[]], initialBoxes = [0]
輸出:6
解釋: 你一開始擁有盒子 0 。打開它你可以找到盒子 1,2,3,4,5 和它們對應的鑰匙。
打開這些盒子,你將獲得所有盒子的糖果,所以總糖果數爲 6 個。
示例 3:
輸入:status = [1,1,1], candies = [100,1,100], keys = [[],[0,2],[]], containedBoxes = [[],[],[]], initialBoxes = [1]
輸出:1
示例 4:
輸入:status = [1], candies = [100], keys = [[]], containedBoxes = [[]], initialBoxes = []
輸出:0
示例 5:
輸入:status = [1,1,1], candies = [2,3,2], keys = [[],[],[]], containedBoxes = [[],[],[]], initialBoxes = [2,1,0]
輸出:7
提示:
1 <= status.length <= 1000 status.length == candies.length ==
keys.length == containedBoxes.length == n status[i] 要麼是 0 要麼是 1 。 1 <=
candies[i] <= 1000 0 <= keys[i].length <= status.length 0 <=
keys[i][j] < status.length keys[i] 中的值都是互不相同的。 0 <=
containedBoxes[i].length <= status.length 0 <= containedBoxes[i][j] <
status.length containedBoxes[i] 中的值都是互不相同的。 每個盒子最多被一個盒子包含。 0 <=
initialBoxes.length <= status.length 0 <= initialBoxes[i] <
status.length
/**
* @param {number[]} status
* @param {number[]} candies
* @param {number[][]} keys
* @param {number[][]} containedBoxes
* @param {number[]} initialBoxes
* @return {number}
*/
var maxCandies = function(status, candies, keys, containedBoxes, initialBoxes) {
let allCandies = 0;
let allContainedBoxes =new Array(status.length).fill(0);
initialBoxes.forEach(e=>{
allContainedBoxes[e] = true;
})
for(let i=0;i<status.length;i++){
if(status[i]==1 && allContainedBoxes[i]){
//收集鑰匙,打開對應箱子
keys[i].forEach(e=>{
if(status[e] ==0){
status[e]=1;
}
})
//收集盒子
containedBoxes[i].forEach(e=>{
allContainedBoxes[e] = true;
})
allCandies+=candies[i];
//箱子裏面的東西拿出來了
status[i] = "-1";
//重新遍歷箱子
i=-1;
}
}
return allCandies
};