Compute‘s Wallpaper 一个有趣的dp题

传送门：ICompute’s Wallpaper

思路：
这道题很容易就可以看出是一道dp题，如果是dp，那么最关键的是推出其状态转移方程。

题目问出现至少两张美女图的方案有多少种（出题人一看就是正经人 ），那么我们就可以从1 到 n-1 枚举起始位置，并且第二次出现的位置一定在i+s[i] 之后（包括本身），那么我们问题就应该转变为如何确定i+s[i] 之后的所有方案数。

如果我们用sum[i] 表示第i 位后有多少种可能的情况，那么决定sum[i] 的是 sum[i-1] 和当前位对其的影响，如果当前美女的吸引度为s[i] 那么决定当前位对其整体的影响就应该是sum[i+s[i]]+1 。为什么要+1呢？因为可以能有这么一种情况：选取了i th 美女后就再也不选了。（直男本色 ）

因此我们只用从后向前扫一遍，逐一求出sum[i] 就行，之后再枚举起始位置，相加和就AC了。

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e6+5;
const int mod=1e9+7;
#define ll long long
int s[MAXN];
ll sum[2*MAXN];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n;
	ll ans=0;
	cin>>n;
	for(int i=1;i<=n;i++)
    {cin>>s[i];if(s[i]==0)s[i]=1;}
	for(int i=n;i>=1;i--)sum[i]=(sum[i+1]+sum[i+s[i]]+1)%mod;
	for(int i=1;i<n;i++)ans=(ans+sum[i+s[i]])%mod;
	cout<<ans<<endl;
	return 0;
 }