力扣 5865. 访问完所有房间的第一天 dp 前缀和优化

https://leetcode-cn.com/problems/first-day-where-you-have-been-in-all-the-rooms/

思路：经典读错题+想歪。分析后不难发现，第一次到达房间$i$，下一次必定要回到$<=i$的房间$j$(不妨把这个操作叫做回访)，且此时$<i$的房间都到达了偶数次，那么对于$[j,i)$的每个房间，我们都需要再次回访，如果用$d{p}_i$表示在位置$i$进行回访操作后再重新回到位置$i$所需要的天数，那么有：
$$d{p}_i=i-j+1+\sum _{k=j}^{i-1}d{p}_k$$
显然最终结果就等于${\sum }_{i=0}^{n-2}d{p}_i+n-1$。但是这样转移复杂度是$O(n^2)$的。那么考虑记录$dp$的前缀和$s$，则转移方程转换为：
$$s_i=s_{i-1}+d{p}_i=s_{i-1}+s_{i-1}-s_{j-1}+i-j+1$$
最终结果等于$s_{n-2}+n-1$，这样复杂度就优化到了$O(n)$。

class Solution {
public:
    int firstDayBeenInAllRooms(vector<int>& nextVisit) {
        int n=nextVisit.size();
        using ll=long long;
        const int mod=1e9+7;
        vector<ll> sum(n-1);
        sum[0]=1;
        for(int i=1;i<n-1;i++)
        {
            ll val=0;
            if(nextVisit[i]>0)
                val=sum[nextVisit[i]-1];
            sum[i]=(sum[i-1]+sum[i-1]-val+i-nextVisit[i]+1+mod)%mod;
        }
        return (sum[n-2]+n-1)%mod;
    }
};

等价的代码：

class Solution {
public:
    int firstDayBeenInAllRooms(vector<int>& nextVisit) {
        int n=nextVisit.size();
        using ll=long long;
        const int mod=1e9+7;
        vector<ll> sum(n);
        for(int i=0;i<n-1;i++)
            sum[i+1]=(sum[i]+sum[i]-sum[nextVisit[i]]+i-nextVisit[i]+1+mod)%mod;
        return (sum[n-1]+n-1)%mod;
    }
};