LCP 47. 入场安检

LCP 47. 入场安检

「力扣挑战赛」 的入场仪式马上就要开始了，由于安保工作的需要，设置了可容纳人数总和为 M 的 N 个安检室，capacities[i] 记录第 i 个安检室可容纳人数。安检室拥有两种类型：

• 先进先出：在安检室中的所有观众中，最早进入安检室的观众最先离开

• 后进先出：在安检室中的所有观众中，最晚进入安检室的观众最先离开

恰好 M+1 位入场的观众（编号从 0 开始）需要排队依次入场安检， 入场安检的规则如下：

• 观众需要先进入编号 0 的安检室

• 当观众将进入编号 i 的安检室时（0 <= i < N），

  • 若安检室未到达可容纳人数上限，该观众可直接进入；

  • 若安检室已到达可容纳人数上限，在该观众进入安检室之前需根据当前安检室类型选择一位观众离开后才能进入；

• 当观众离开编号 i 的安检室时 （0 <= i < N-1)，将进入编号 i+1 的安检室接受安检。

若可以任意设定每个安检室的类型，请问有多少种设定安检室类型的方案可以使得编号 k 的观众第一个通过最后一个安检室入场。

注意：

• 观众不可主动离开安检室，只有当安检室容纳人数达到上限，且又有新观众需要进入时，才可根据安检室的类型选择一位观众离开；

• 由于方案数可能过大，请将答案对 1000000007 取模后返回。

---

纸老虎，我们只需要从前面几个安检室选几个使得总的容量为 k ，也就是说这几个安检室都是后进先出的栈形式：

func securityCheck(capacities []int, k int) int {
    f := make([]int, k + 1)
    f[0] = 1
    mod := 1000000000 + 7
    n := len(capacities)
    for i := 0 ; i < n; i ++ {
        for j := k; j >= capacities[i] - 1; j -- {
            f[j] = (f[j] + f[j - capacities[i] + 1]) % mod
        }
    }
    return f[k]
}