BOJ)2507 공주 구하기

문제: icpc.me/2507

0에서 n-1을 갔다가 0으로 다시 돌아와야하는 bitonic tour의 변형문제이다.

바이토닉 투어는 다이나믹 프로그래밍으로 해결해줄 수 있다.

dp[x][y] = 올 때 x번 정점을 밟고 갈 때 y번 정점을 밟을 때의 경로의 개수

점화식은 중복을 제거해주기 위해 max(x,y)보다 큰 섬으로만 보낼 수 있어야 한다.

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int n, w[505], g[505], b[505], a[505][505], dp[505][505];
const int MOD = 1000;
int func(int x, int y) {
    int &ret = dp[x][y];
    if (ret != -1)return ret;
    ret = 0;
    if (x != n - 1 && y != n - 1) {
        if (a[n - 1][x] && a[y][n - 1])ret = 1;
    }
    for (int i = max(x, y) + 1; i < n; i++) {
        if (a[i][x])
            ret += func(i, y);
        ret %= MOD;
        if (a[y][i])
            ret += func(x, i);
        ret %= MOD;
    }
    return ret;
}
int main() {
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
        scanf("%d%d%d", &w[i], &g[i], &b[i]);
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            if (abs(w[i] - w[j]) <= g[i] && !(i>j&&!b[i]))
                a[i][j] = 1;
        }
    }
    memset(dp, -1, sizeof(dp));
    printf("%d\n", func(0, 0));
    return 0;
}