1298. 你能从盒子里获得的最大糖果数

Go语言使用广度优先

func maxCandies(status []int, candies []int, keys [][]int, containedBoxes [][]int, initialBoxes []int) int {
    n := len(status)
	canOpen := make([]bool, n)
	hasBox := make([]bool, n)
	used := make([]bool, n)

	for i := 0; i < n; i++ {
		canOpen[i] = (status[i] == 1)
	}
	q := []int{}
	ans := 0
	for _, box := range initialBoxes {
		hasBox[box] = true
		if canOpen[box] {
            q = append(q, box)
			used[box] = true
			ans += candies[box]
		}
	}

	for len(q) > 0 {
		bigBox := q[0]
        q = q[1:]
		for _, key := range keys[bigBox] {
			canOpen[key] = true
			if !used[key] && hasBox[key] {
                q = append(q, key)
				used[key] = true
				ans += candies[key]
			}
		}
		for _, box := range containedBoxes[bigBox] {
			hasBox[box] = true
			if !used[box] && canOpen[box] {
				q = append(q, box)
				used[box] = true
				ans += candies[box]
			}
		}
	}

	return ans
}