Submission #705403

#	Time	Username	Problem	Language	Result	Execution time	Memory
705403		aedmhsn	Journey (NOI18_journey)	C++17	100 / 100	1999 ms	38272 KiB

journey

#include <bits/stdc++.h>
using namespace std;
 
 
#define A first
#define B second
#define MP make_pair
#define ms(a, x) memset(a, x, sizeof(a)) 
 
 
#define boost() ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL)
 
typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<long long, long long> pll;
typedef pair<long double, long double> pld;
const int INF = 0x3f3f3f3f;
const double PI = acos(-1);

const int mxN=1e4+5;

ll n, m, h, dp[mxN][405];
vector <vector<pll>> adj(mxN);

ll solve(int node, int date){
    if(node == 0 && date == 0)
        return 1;
    if(dp[node][date] != -1)
        return dp[node][date];
    ll sum=0;
    bool over=false;
    for(auto [x, y]:adj[node]){
        if(x < node){
            for(int i=0; i<m; i++){
                if(date-y-i >= 0){
                    sum += solve(x, date-y-i);
                    if(sum >= 5e8+1){
                        sum = 5e8+1;
                        over=true;
                        break;
                    }
                }
                else
                    break;
            }
            if(over)
                break;
        }
    }
    return dp[node][date]=sum;
}


int main(){
    boost();
    ms(dp, -1);
    cin >> n >> m >> h;
    for(int i=0; i<n-1; i++){
        for(int j=0; j<h; j++){
            int x, y;
            cin >> x >> y;
            adj[x].push_back({i, y});
        }
    }
    for(int i=0; i<=m-1; i++)
        cout << solve(n-1, i) << " ";
}

• Subtask 1: 20 / 20
• Subtask 2: 23 / 23
• Subtask 3: 26 / 26
• Subtask 4: 31 / 31