CF Round 174D. Cow Program 思维,记忆化

题意:长度为n的序列a. 第i轮游戏中 令a[1]=i. 初始x=1,y=0.任何步骤出现:x<=0或者y>n时终止.
每轮:步骤1:y+=a[x],x+=a[x]. 步骤2:y+=a[x], x-=a[x]. 
2<=n<=2e5,1<=a[i]<=1e9.问i=[1...n-1]时 最后y是多少? 永远无法终止时y=-1.

暴力模拟时O(n^2) 只要令dp[u][0/1]表示从u出发,方向向右或向左时的结果即可.
dp[u][1]=a[u]+dp[u+a[u]][0].
因为可能有环 初始令dp[u][0/1]为inf,表示无解.设vis[u][0/1]标记是否访问过该状态.若访问过,则返回dp[u][0/1].

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
const ll inf=2e16;
bool mk[N],vis[N][2];
ll n,a[N],res[N],dp[N][2];
ll dfs(int u,int st){
	if(u<=0||u>n)	return 0;
	if(vis[u][st])	return dp[u][st];
	vis[u][st]=true;
	ll num=1;
	if(st==0)	num=-1;		
	ll val=a[u]+dfs(u+num*a[u],1-st);
	return dp[u][st]=val;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++)	dp[i][0]=dp[i][1]=inf; 
	vis[1][1]=true;
	for(int i=2;i<=n;i++)	cin>>a[i];
	for(int i=2;i<=n;i++)
	{
		a[1]=i-1;
		res[i]=a[i]+dfs(i-a[i],1);
	}
	for(int i=2;i<=n;i++)
	{
		res[i]+=i-1;
		if(res[i]>=inf)	res[i]=-1;
		cout<<res[i]<<'\n';
	}
	return 0;
}