Submission #992662

#	Time	Username	Problem	Language	Result	Execution time	Memory
992662		goats_9	Toll (BOI17_toll)	C++17	100 / 100	57 ms	86868 KiB

toll

/* Author: goats_9 */

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

int k, n, m, o;

int spt[50000][17][5][5], ans[5][5], tmp[5][5];

void merge(int res[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++) {
				res[x][y] = min(res[x][y], a[x][z] + b[z][y]);
			}
		}
	}
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cin >> k >> n >> m >> o;
	memset(spt, 0x3f, sizeof(spt));
	for (int i = 0; i < m; i++) {
		int a, b, t;
		cin >> a >> b >> t;
		spt[a/k][0][a%k][b%k] = t;
	}
	for (int j = 1; j < 17; j++) {
		int bm = 1<<j;
		for (int i = 0; i + bm < (n + k - 1) / k; i++) {
			merge(spt[i][j], spt[i][j - 1], spt[i + bm/2][j - 1]);
		}
	}
	while (o--) {
		int a, b;
		cin >> a >> b;
		memset(ans, 0x3f, sizeof(ans));
		for (int i = 0; i < 5; i++) ans[i][i] = 0;
		for (int a1 = a / k, b1 = b / k, i = 16; i >= 0; i--) {
			if (a1 + (1<<i) <= b1) {
				memset(tmp, 0x3f, sizeof(tmp));
				merge(tmp, ans, spt[a1][i]);
				memcpy(ans, tmp, sizeof(ans));
				a1 += (1<<i);
			}
		}
		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