Codeforces Round \ # 174 (Div. 1) B. Cow 프로그램 (dp + 기억 화)

1364 단어 codeforces
제목 링크: http://codeforces.com/contest/283/problem/B
사고방식: dp [now] [flag] 는 현재 위치 now 를 표시 하고 flag 는 다음 에 해 야 할 절 차 를 표시 한 다음 에 제목 의 기억 화 에 따라 기억 을 검색 하고 vis 배열 은 이미 방문 한 상 태 를 표시 합 니 다.
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define REP(i, a, b) for (int i = (a); i < (b); ++i)
#define FOR(i, a, b) for (int i = (a); i <= (b); ++i)
using namespace std;

const int MAX_N = (200000 + 100);
long long dp[MAX_N][2];
int vis[MAX_N][2];
int N, a[MAX_N];

long long dfs(int now, int dir)
{
    if (now <= 0 || now > N) return 0;
    if (~dp[now][dir]) return dp[now][dir];
    if (vis[now][dir]) return -1;
    vis[now][dir] = 1;
    long long res = dfs(dir ? now + a[now] : now - a[now], dir ^ 1);
    return dp[now][dir] = (res < 0 ? -1 : res + a[now]);
}

int main()
{
    while (cin >> N) {
        FOR(i, 2, N) cin >> a[i];
        memset(dp, -1, sizeof(dp));
        memset(vis, 0, sizeof(vis));
        FOR(i, 1, N - 1) {
            long long ans = dfs(i + 1, 0);
            cout << (ans < 0 ? -1 : ans + i) << endl;
        }
    }
    return 0;
}

좋은 웹페이지 즐겨찾기