codeforces D. Long Path

http://codeforces.com/contest/408/problem/D

 题意：有一排房间每个房间有两扇门，一扇通往第i+1个房间，另一扇通往第p[i]个房间，（p[i]<=i）然后他每经过一个房间就做一个标记，只有偶数个标记时他才会走第一扇门。问你他走到第n+1个房间需要多少单位时间。

思路：dp，一个人进入一个房间之后，如果标记为奇数的话，他会进入p[i]房间，又重新进入这个房间，重复一次的过程，所以dp[i]=dp[i-1]+dp[i-1]-dp[p[i]-1]+2;

 

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 100010
using namespace std;
const int mod=1000000007;

long long dp[maxn];
int p[maxn];
int num[maxn];
int n;

int main()
{
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&p[i]);
    }
    dp[1]=2;
    for(int i=2; i<=n; i++)
    {
       if(i==p[i])
       {
           dp[i]=dp[i-1]+2;
       }
       else
       {
           dp[i]=(dp[i-1]+dp[i-1]-dp[p[i]-1]+2+mod)%mod;
       }
    }
    printf("%lld\n",dp[n]);
    return 0;
}