2327. 知道秘密的人数
代码/思路
1- 定义dp[i] 代表 第 i 天获取密码的人数
2- 那么他将会在第 i + delay 到 i + forget - 1 天分享秘密给新人,
状态转移方程为: dp[j] += dp[i]; 其中j 的范围就是 [i + delay, i + forget - 1]
3-最后统计 [n - forget + 1, n] 区间的获得秘密的人数即可
class Solution {
#define maxn 1010
#define mod 1000000007
int dp[maxn];
public:
int peopleAwareOfSecret(int n, int delay, int forget) {
dp[1] = 1;
for(int i = 1; i <= n; i++)
{
for(int j = i + delay; j < i + forget && j <= n; j++)
{
dp[j] += dp[i];
if(dp[j] >= mod)
{
dp[j] -= mod;
}
}
}
int sum = 0;
for(int i = n - forget + 1; i <= n; i++)
{
sum += dp[i];
if(sum >= mod)
{
sum -= mod;
}
}
return sum;
}
};