codeforces B. Cow Program (记忆化搜索)

来源：互联网发布：淘宝卖家怎么提高流量编辑：程序博客网时间：2024/06/05 11:43

题目链接：

codeforces 283B

题目大意：

给出n个数，奇数次操作x,y都加上a[x],偶数次操作y加上a[x],x减去a[x],走出了范围就结束。
问结束时的y值，如果无法结束，那么输出-1

题目分析：

记录状态dp[x][2]为在奇数次或偶数次到达x点时走完还会获得的权值。
直接搜索，搜索到搜过的状态直接返回。

AC代码：

#include <iostream>#include <cstdio>#include <algorithm>#include <cstring>#define MAX 200007using namespace std;typedef long long LL;LL dp[MAX][2];int vis[MAX][2];int a[MAX],n;void dfs ( int x , int d ){    if ( vis[x][d%2] ) return;    vis[x][d%2] = 1;    if ( d&1 )    {        int y = x-a[x];        if ( y <= 0 )         {            dp[x][d%2] = a[x];            return;        }        dfs ( y , d+1 );        if ( dp[y][(d+1)%2] != -1 )             dp[x][d%2] = dp[y][(d+1)%2] + a[x];    }    else    {        int y = x+a[x];        if ( y > n )         {            dp[x][d%2] = a[x];            return;        }        dfs ( y , d+1 );        if ( dp[y][(d+1)%2] != -1 )            dp[x][d%2] = dp[y][(d+1)%2] + a[x];    }}int main ( ){    while ( ~scanf ( "%d" , &n ) )    {        for ( int i = 2 ; i <= n ; i++ )            scanf ( "%d" , &a[i] );        memset ( vis , 0 , sizeof ( vis ) );        memset ( dp , -1, sizeof ( dp ) );        dp[1][0] = -1;        vis[1][0] = 1;        dp[1][1] = 0;        vis[1][1] = 1;        for ( int i = 2 ; i <= n ; i++ )            dfs ( i , 1 );        for ( int i = 2 ; i <= n ; i++ )            if ( dp[i][1] != -1 )                dp[i][1] += i-1;        for ( int i = 2 ; i <= n ; i++ )            printf ( "%lld\n" , dp[i][1] );    }}

0 0