此题的队列是完全没用的数据结构,栈的作用是困住无辜的人,容量为
b
b
b 的栈,会困住
b
−
1
b-1
b−1 个无辜的人,所以我们需要把数组中每个数减一,然后看他们如何拼满
k
k
k 个人。
此题最重要的是进行题目解析,减一后转换为背包装满问题就很简单了。
之前比赛时压力比较大,理解为从
n
n
n 个物品中做选择,如果选
m
m
m 个,求出能装满容量为
m
+
k
m+k
m+k 的背包的方案数。其实就每个减一就行,但是脑子没有转过弯,就很难受。
class Solution {
public:
int securityCheck(vector<int>& capacities, int k) {
vector<int> dp(k+1);
dp[0] = 1;
int MOD = 1e9+7;
for(auto& num: capacities)num--;
for(auto num: capacities)
{
for(int j=k;j>=num;j--)
{
dp[j] = (dp[j-num] + dp[j])%MOD;
}
}
return dp[k];
}
};