제출 #1210631

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1210631		vanea	Toll (BOI17_toll)	C++20	100 / 100	101 ms	98380 KiB

toll

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

const int mxN = 5e4+10;

int dp[mxN][20][5][5], ans[5][5], tmp[5][5], k;

void comb(int dp[5][5], int a[5][5], int b[5][5]) {
    for(int x = 0; x < k; x++) {
        for(int y = 0; y < k; y++) {
            for(int z = 0; z < k; z++) {
                dp[x][y] = min(dp[x][y], a[x][z]+b[z][y]);
            }
        }
    }
}

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n, m, q;
    cin >> k >> n >> m >> q;
    memset(dp, 0x3F, sizeof dp);
    for(int i = 0; i < m; i++) {
        int a, b, w;
        cin >> a >> b >> w;
        dp[a/k][0][a%k][b%k] = w;
    }
    for(int j = 1; j < 17; j++) {
        for(int i = 0; i + (1 << j) < (n + k - 1) / k; i++) {
            comb(dp[i][j], dp[i][j-1], dp[i + (1 << (j - 1))][j-1]);
        }
    }
    while(q--) {
        int a, b;
        cin >> a >> b;
        if(a > b) {
            cout << -1 << '\n';
            continue;
        }
        int dist = (b/k)-(a/k), curr = (a/k);
        memset(ans, 0x3F, sizeof ans);
        for(int i = 0; i < k; i++) ans[i][i] = 0;
        for(int j = 0; j <= 16; j++) {
            if(dist & (1 << j)) {
                memset(tmp, 0x3F, sizeof tmp);
                comb(tmp, ans, dp[curr][j]);
                memcpy(ans, tmp, sizeof ans);
                curr += (1 << j);
            }
        }
        cout << (ans[a%k][b%k] == 0x3F3F3F3F ? -1 : ans[a%k][b%k]) << '\n';
    }
    return 0;
}

• Subtask 1: 7 / 7
• Subtask 2: 10 / 10
• Subtask 3: 8 / 8
• Subtask 4: 31 / 31
• Subtask 5: 44 / 44