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

最新推荐文章于 2025-01-02 00:05:29 发布

原创最新推荐文章于 2025-01-02 00:05:29 发布 · 1.4k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#codeforces #搜索 #dp #记忆化搜索

codeforces的dp专题专栏收录该内容

78 篇文章

订阅专栏

本文介绍Codeforces283B题目的解题思路，通过使用深度优先搜索算法来记录状态并避免重复计算，最终求解出给定序列下特定的操作序列所能达到的状态值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：

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 200007

using 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] );
    }
}