oj.uz

Submission #1114781

# Submission time Handle Problem Language Result Execution time Memory
1114781 2024-11-19T15:23:46 Z FucKanh Toll (BOI17_toll) C++14
7 / 100
 50 ms 29508 KB 
#include<bits/stdc++.h>
#define ll long long
#define int ll
#define pii pair<int,int>
#define p3i pair<int,pii>

using namespace std;

const int inf = 1e15;
const int maxn = 5e4 + 2;
const int maxq = 1e4 + 2;

int k,n,m,q;
int ans[maxq];
int d[maxn][7][7];
vector<pii> a[maxn],b[maxn];

int cv(int tk, int lvl) {
    return (lvl-1) * k + tk;
}

pii cv(int u) {
    return make_pair(u % k, u / k + 1);
}

void query(int l, int r, vector<p3i> qr) {
    if (qr.size()==0) return;
    if (l==r) {
        for (p3i val : qr) {
            int id = val.first;
            ans[id] = 0;
        }
        return;
    }

    for (int i = 1; i <= n; i++) {
        for (int u = 0; u < k; u++) {
            for (int v = 0; v < k; v++) {
                d[i][u][v] = inf;
            }
        }
    }

    int mid = l + r >> 1;
    for (int i = 0; i < k; i++) d[mid][i][i] = 0;
    for (int i = mid - 1; i >= l; i--) {
        for (int u = 0; u < k; u++) {
            int f = cv(u,i);
            for(pii val : a[f]) {
                int g,t; tie(g,t) =  val;
                int um = cv(g).first;
                for (int v = 0; v < k; v++) {
                    int h = cv(v,mid);
                    if (cv(v,mid) >= n) continue;
                    if (d[i+1][um][v] < inf) {
                        d[i][u][v] = min(d[i][u][v], d[i+1][um][v] + t);
//                        cerr << "1. from " << f << " to " << g << " to " << h << " | " << u << " " << um << " " << v << " : " << d[i][u][v] << endl;
                    }
                }
            }
        }
    }

    for (int i = mid + 1; i <= r; i++) {
        for (int u = 0; u < k; u++) {
            int f = cv(u,i);
            if (f >= n) continue;
            for(pii val : b[f]) {
                int g,t; tie(g,t) =  val;
                int um = cv(g).first;
                for (int v = 0; v < k; v++) {
                    int h = cv(v,mid);
                    if (h >= n) continue;
                    if (d[i-1][v][um] < inf) {
                        d[i][v][u] = min(d[i][v][u], d[i-1][v][um] + t);
//                        cerr << "2. from " << h << " to " << g << " to " << f << " | " << v << " " << um << " " << u << " : " << d[i][v][u] << endl;
                    }
                }
            }
        }
    }
    vector<p3i> qrl,qrr;
    for (p3i val : qr) {
        int l,r; tie(l,r) = val.second;
        int u,lvlu; tie(u,lvlu) = cv(l);
        int v,lvlv; tie(v,lvlv) = cv(r);
        int id = val.first;
//        cerr  << id << ":: " << u << " " << lvlu << " " << v << " " << lvlv << endl;
        if (lvlv <= mid) qrl.push_back(val);
        else if (lvlu >mid) qrr.push_back(val);
        else {
            ans[id] = inf;
            for (int f = 0; f < k; f++) {
                ans[id] = min(ans[id], d[lvlu][u][f] + d[lvlv][f][v]);
            }
        }
        if (ans[id] >= inf) ans[id] = -1;
    }

    query(l,mid,qrl);
    query(mid+1,r,qrr);
}

void solve() {
    cin >> k >> n >> m >> q;
    int lvl = n / k + (n%k != 0);
    for (int i = 1; i <= m; i++) {
        int x,y,t; cin >> x >> y >> t;
        a[x].push_back({y,t});
        b[y].push_back({x,t});
    }

//    for (int i = 0; i < n; i++) {
//        cout << i << " : " << cv(i).first << " " << cv(i).second << endl;
//    }

    vector<p3i> qr;
    for (int i = 1; i <= q; i++){
        int x,y; cin >> x >> y;
        qr.push_back({i,{x,y}});
    }
    query(1,lvl,qr);
    for (int i = 1; i <= q; i++) {
        cout << ans[i] << '\n';
    }
}

signed main() {
    cin.tie(0) -> sync_with_stdio(0);
    solve();
    return 0;
}

Compilation message

toll.cpp: In function 'void query(long long int, long long int, std::vector<std::pair<long long int, std::pair<long long int, long long int> > >)':
toll.cpp:44:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   44 |     int mid = l + r >> 1;
      |               ~~^~~
toll.cpp:53:25: warning: unused variable 'h' [-Wunused-variable]
   53 |                     int h = cv(v,mid);
      |                         ^

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 42 ms 26884 KB Output is correct
2 Correct 2 ms 4688 KB Output is correct
3 Correct 1 ms 4748 KB Output is correct
4 Correct 2 ms 4688 KB Output is correct
5 Correct 2 ms 4688 KB Output is correct
6 Correct 3 ms 4688 KB Output is correct
7 Correct 3 ms 4756 KB Output is correct
8 Correct 46 ms 26692 KB Output is correct
9 Correct 42 ms 26704 KB Output is correct
10 Correct 26 ms 22960 KB Output is correct

Subtask #2
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 50 ms 29508 KB Output is correct
2 Correct 1 ms 4688 KB Output is correct
3 Incorrect 2 ms 4688 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 4688 KB Output is correct
2 Incorrect 2 ms 4856 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 31.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 4688 KB Output is correct
2 Incorrect 2 ms 4856 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #5
0 / 44.0

# Verdict Execution time Memory Grader output
1 Correct 42 ms 26884 KB Output is correct
2 Correct 2 ms 4688 KB Output is correct
3 Correct 1 ms 4748 KB Output is correct
4 Correct 2 ms 4688 KB Output is correct
5 Correct 2 ms 4688 KB Output is correct
6 Correct 3 ms 4688 KB Output is correct
7 Correct 3 ms 4756 KB Output is correct
8 Correct 46 ms 26692 KB Output is correct
9 Correct 42 ms 26704 KB Output is correct
10 Correct 26 ms 22960 KB Output is correct
11 Correct 50 ms 29508 KB Output is correct
12 Correct 1 ms 4688 KB Output is correct
13 Incorrect 2 ms 4688 KB Output isn't correct
14 Halted 0 ms 0 KB -