Medium
First Day Where You Have Been in All the Rooms — C++
Full explanation · Time O(n) · Space O(n)
// Time: O(n)
// Space: O(n)
class Solution {
public:
int firstDayBeenInAllRooms(vector<int>& nextVisit) {
static const int MOD = 1e9 + 7;
vector<int> dp(size(nextVisit));
for (int i = 1; i < size(nextVisit); ++i) {
dp[i] = (dp[i - 1] + 1ull + (dp[i - 1] - dp[nextVisit[i - 1]]) + 1 + MOD) % MOD;
}
return dp.back();
}
};