dp
又是动态规划,这是字节跳动校招的一道附加题-传送门:
https://www.nowcoder.com/practice/58b04ed2865f4ff4921290f1bd4ee486?tpId=90&tqId=30811&tPage=2&rp=2&ru=/ta/2018test&qru=/ta/2018test/question-ranking
解析
这道题首先一看先需要确定这是一个怎样的过程。到
i
i
i个点,如果是第一次到,显然是会移动到
p
[
i
]
p[i]
p[i],而这个
1
≤
p
[
i
]
≤
i
1 \leq p[i] \leq i
1≤p[i]≤i, 所以他是不会前进的,而一个显然的条件是,当它到达
i
i
i点时,前面的点访问过的次数肯定都是偶数次,这个可以很容易的证明。那么当它到达
i
i
i点后移动到
p
[
i
]
p[i]
p[i]时,由于前面的点都是偶数次,相当于从那个点再到第
i
i
i个点这样一个过程,那么就可以dp了,设
d
p
[
i
]
dp[i]
dp[i]表示到达
i
i
i点的方案数,那么可以得出状态转移方程为:
d
p
[
i
]
=
(
d
p
[
i
−
1
]
+
1
+
d
p
[
i
−
1
]
−
d
p
[
p
[
i
]
−
1
]
+
1
)
dp[i]=(dp[i-1]+1+dp[i-1]-dp[p[i]-1]+1)
dp[i]=(dp[i−1]+1+dp[i−1]−dp[p[i]−1]+1)然后在减的那个地方记得加上膜数就行(防止出现小于0的情况)。
AC代码:
// 小学生一发的刷题之路
// 3月3号;
//
//
//
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <queue>
#include <deque> //双向队列;
#include <cmath>
#include <set>
#include <stack>
#include <map>
#include <vector>
#include <cstdlib>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
const double PI=acos(-1.0);
const double eps=1e-10;
const int maxn=1e5+5;
const int maxm=1e3+5;
const ll mod=1e9+7;
const int INF=1e8;
template<class T>
inline void read(T &ret){ //快速输入模版;
ret=0;
int f=1;
char c=getchar();
while(c<'0'||c>'9'){
if(c=='-') f=-1;
c=getchar();
}
while(c>='0'&&c<='9'){
ret=ret*10+c-'0';
c=getchar();
}
ret*=f;
}
template <class T>
inline void out(T ret){ //快速输出模版;
if(ret>9)
{
out(ret/10);
}
putchar(ret%10+'0');
}
int n,pos[maxn];
ll dp[maxn];
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&pos[i]);
}
//dp[i]表示从1到i的步骤数;
dp[0]=0;
for(int i=1;i<=n;i++){
dp[i]=(dp[i-1]+2)%mod; //先走到i;
dp[i]=(dp[i]+dp[i-1]-dp[pos[i]-1]+mod)%mod;
}
printf("%lld\n",dp[n]);
return 0;
}
新的开始,每天都要快乐哈!