# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
986101 |
2024-05-19T17:58:23 Z |
tfgs |
Toll (BOI17_toll) |
C++17 |
|
76 ms |
20060 KB |
// spent some time confused b/c i thought that queries had to be between
// different blocks. but this is only edges, queries in the same block are
// valid. and of course the distance between u and u is 0
#include <bits/stdc++.h>
using namespace std;
const int inf = 1e9;
#ifdef LOCAL
#include "algo/debug.h"
#endif
#define f first
#define s second
template<class T> using V = vector<T>;
using vi = V<int>;
using vb = V<bool>;
using vs = V<string>;
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
#define pb push_back
#define lb lower_bound
#define ub upper_bound
template<class T> int lwb(V<T>& a, const T& b) { return lb(all(a),b)-begin(a); }
template<class T> int upb(V<T>& a, const T& b) { return ub(all(a),b)-begin(a); }
const int max_k = 5, max_n = 5e4, max_q = 1e4;
int k, n, m, q;
V<array<int, 2>> g[max_n][max_k];
V<V<vi>> t; // node, class a, class b = dist
V<vi> zero;
bool is_zero(const V<vi>& vec) {
for (int i = 0; i < k; i++) for (int j = 0; j < k; j++)
if (vec[i][j] != 0) return false;
return true;
}
void build(int v, int l, int r) {
if (l == r-1) {
for (int my_class = 0; my_class < k; my_class++)
t[v][my_class][my_class] = 0;
return;
}
int m = (l+r)/2;
build(2*v+1, l, m);
build(2*v+2, m, r);
for (int l_class = 0; l_class < k; l_class++) {
for (int lm_class = 0; lm_class < k; lm_class++) {
for (auto [w, rm_class] : g[m-1][lm_class]) {
for (int r_class = 0; r_class < k; r_class++) {
t[v][l_class][r_class] = min(t[v][l_class][r_class],
t[2*v+1][l_class][lm_class]+w+t[2*v+2][rm_class][r_class]);
}
}
}
}
}
V<vi> query(int v, int l, int r, int L, int R) {
if (r <= L || R <= l) return zero;
if (L <= l && r <= R) return t[v];
int m = (l+r)/2;
V<vi> res(k, vi(k, inf));
auto lv = query(2*v+1, l, m, L, R);
auto rv = query(2*v+2, m, r, L, R);
if (is_zero(lv)) return rv;
if (is_zero(rv)) return lv;
for (int l_class = 0; l_class < k; l_class++) {
for (int lm_class = 0; lm_class < k; lm_class++) {
for (auto [w, rm_class] : g[m-1][lm_class]) {
for (int r_class = 0; r_class < k; r_class++) {
res[l_class][r_class] = min(res[l_class][r_class],
lv[l_class][lm_class]+w+rv[rm_class][r_class]);
}
}
}
}
return res;
}
void init_tree() {
int size = 1;
while (size < 2*((n-1)/k+1)) size *= 2;
t = V<V<vi>>(size, V<vi>(k, vi(k, inf)));
build(0, 0, (n-1)/k+1);
}
void solve() {
cin >> k >> n >> m >> q;
for (int i = 0; i < m; i++) {
int u, v, w; cin >> u >> v >> w;
g[u/k][u%k].pb({w, v%k});
}
zero = V<vi>(k, vi(k));
init_tree();
for (int i = 0; i < q; i++) {
int u, v; cin >> u >> v;
auto res = query(0, 0, (n-1)/k+1, u/k, v/k+1);
int ans = res[u%k][v%k];
if (ans == inf) ans = -1;
cout << ans << '\n';
}
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
solve();
return 0;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
76 ms |
20060 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
65 ms |
19240 KB |
Output is correct |
2 |
Correct |
2 ms |
6232 KB |
Output is correct |
3 |
Correct |
2 ms |
6232 KB |
Output is correct |
4 |
Correct |
3 ms |
6236 KB |
Output is correct |
5 |
Correct |
4 ms |
6236 KB |
Output is correct |
6 |
Correct |
3 ms |
6236 KB |
Output is correct |
7 |
Incorrect |
14 ms |
6492 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
3 ms |
6232 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
3 ms |
6232 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
76 ms |
20060 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |