제출 #1367304

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1367304		kahoul	Trains (BOI24_trains)	C++20	71 / 100	2106 ms	442892 KiB

Main

#include <bits/stdc++.h>
using namespace std;
#define int long long

const int mod = 1e9 + 7;
const int maxn = 1e5 + 10;

int x[maxn];
int d[maxn];

int dp[maxn];
map<int, int> resp[maxn];

int n;

int sum (int a, int b) {
    return (((a + b) % mod) + mod) % mod;
}

int aux(int i, int d) {
    if (d == 0) return 0;
    if (i >= n) return 0;
    if (i == n - 1) return 1;
    if (resp[i].count(d)) return resp[i][d];
    int ans = sum(dp[i], aux(i + d, d));

    return resp[i][d] = ans;
}

signed main () {
    cin >> n;

    for (int i = 0; i < n; i++) {
        cin >> d[i] >> x[i];
    }

    dp[n - 1] = 1;

    for (int i = n - 2; i >= 0; i--) {
        int t1 = aux(i + d[i], d[i]);
        int t2 = aux(i + (x[i] + 1) * d[i], d[i]);
        dp[i] = sum(
            sum(1, t1),
            -t2
        );
    }

    for (int i = 0; i < 1; i++) {
        cout << dp[i] << ' ';
    }
}

• 서브태스크 1: 8 / 8
• 서브태스크 2: 13 / 13
• 서브태스크 3: 16 / 16
• 서브태스크 4: 34 / 34
• 서브태스크 5: 0 / 29