oj.uz

Submission #992733

# Submission time Handle Problem Language Result Execution time Memory
992733 2024-06-05T06:31:57 Z stdfloat Closing Time (IOI23_closing) C++17
9 / 100
 432 ms 28864 KB 
#include <bits/stdc++.h>
#include "closing.h"
using namespace std;

#define ff  first
#define ss  second
#define pii pair<int, int>

using ll = long long;

vector<bool> vis;

vector<ll> c;

vector<vector<pii>> E;

bool dfs(int x, int p = -1, ll sm = 0) {
    bool tr = false;

    if (vis[x]) {
        tr = true;
        c[x] = max(c[x], sm);
    }

    for (auto [i, w] : E[x]) {
        if (i != p && dfs(i, x, sm + w)) {
            tr = true;
            c[x] = max(c[x], sm);
        }
    }

    return tr;
}

vector<ll> f(int st, int n) {
    queue<int> q;
    vector<ll> dis(n, -1);
    q.push(st); dis[st] = 0;
    while (!q.empty()) {
        auto x = q.front(); q.pop();

        for (auto [i, w] : E[x]) {
            if (dis[i] == -1) {
                dis[i] = dis[x] + w;
                q.push(i);
            }
        }
    }

    return dis;
}

int max_score(int n, int X, int Y, long long K, vector<int> U, vector<int> V, vector<int> W) {
    E.assign(n, {});
    for (int i = 0; i < n - 1; i++) {
        E[U[i]].push_back({V[i], W[i]});
        E[V[i]].push_back({U[i], W[i]});
    }

    vector<ll> disx = f(X, n), disy = f(Y, n);

    // if (n <= 20) {
        vector<pii> u;
        for (int i = 0; i < n; i++)
            u.push_back({min(disx[i], disy[i]), i});

        sort(u.begin(), u.end());

        int mx = 0;
        for (int mk = 0; mk < 1 << n; mk++) {
            c.assign(n, -1);
            vis.assign(n, false);
            for (int i = 0; i < n; i++)
                vis[i] = (mk >> i) & 1;

            dfs(X); dfs(Y);

            ll sm = 0;
            int cnt = 0;
            for (int i = 0; i < n && sm <= K; i++) {
                if (c[i] != -1) {
                    sm += c[i];
                    cnt += 1 + (c[i] == max(disx[i], disy[i]));
                }
            }

            if (sm > K) continue;

            for (int i = 0; i < n; i++) {
                if (c[u[i].ss] == -1) {
                    if ((sm += u[i].ff) > K) break;
                    cnt += 1 + (disx[u[i].ss] == disy[u[i].ss]);
                }
            }

            mx = max(mx, cnt);
        }

        return mx;
    // }

    // vector<pii> v;
    // for (int i = 0; i < n; i++) {
    //     if ((int)E[i].size() == 1) {
    //         v.push_back({i, 0});
            
    //         queue<int> q;
    //         vector<bool> vis(n);
    //         q.push(i); vis[i] = true;
    //         while (!q.empty()) {
    //             auto x = q.front(); q.pop();

    //             for (auto [j, w] : E[x]) {
    //                 if (!vis[j]) {
    //                     q.push(j);
    //                     vis[j] = true;
    //                     v.push_back({j, w});
    //                 }
    //             }
    //         }
        
    //         break;
    //     }
    // }

    // for (auto [i, w] : v) {
    //     if (i == X || i == Y) {
    //         if (i == Y) swap(X, Y);
    //         break;
    //     }
    // }

    // int a = -1, b = -1;
    // for (int i = 0; i < (int)v.size(); i++) {
    //     if (v[i].ff == X) a = i;
    //     else if (v[i].ff == Y) b = i;
    // }

    // vector<int> u;
    // for (int k = 0; k < n; k++)
    //     u.push_back(min(disx[v[k].ff], disy[v[k].ff]));

    // sort(u.begin(), u.end());

    // ll sm = 0;
    // int cnt = 0;
    // for (auto i : u) {
    //     if ((sm += i) <= K) cnt++;
    //     else break;
    // }

    // int mx = cnt;
    // for (int i = 0; i < n; i++) {
    //     for (int j = i; j < n; j++) {
    //         vector<ll> c(n, -1);
    //         for (auto x : {a, b}) {
    //             for (auto y : {i, j}) {
    //                 ll sm = 0;
    //                 c[x] = max(c[x], 0LL);
    //                 if (y <= x) {
    //                     for (int k = x - 1; k >= y; k--)
    //                         c[k] = max(c[k], sm += v[k + 1].ss);
    //                 }
    //                 else {
    //                     for (int k = x + 1; k <= y; k++)
    //                         c[k] = max(c[k], sm += v[k].ss);
    //                 }
    //             }
    //         }

    //         ll sm = 0;
    //         for (auto k : c)
    //             if (k != -1) sm += k;

    //         if (sm > K) break;

    //         vector<int> u;
    //         for (int k = 0; k < n; k++)
    //             if (c[k] == -1) u.push_back(min(disx[v[k].ff], disy[v[k].ff]));

    //         sort(u.begin(), u.end());

    //         int cnt = 0;
    //         for (int i = 0; i < n; i++)
    //             if (c[i] != -1) cnt += 1 + (c[i] != min(disx[i], disy[i]));

    //         for (auto i : u) {
    //             if ((sm += i) <= K) cnt++;
    //             else break;
    //         }

    //         mx = max(mx, cnt);
    //     }
    // }

    // return mx;
}

Subtask #0
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct

Subtask #1
0.0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 91 ms 28864 KB 1st lines differ - on the 1st token, expected: '451', found: '200000'
2 Halted 0 ms 0 KB -

Subtask #2
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 397 ms 404 KB Output is correct
3 Correct 411 ms 596 KB Output is correct
4 Correct 432 ms 348 KB Output is correct
5 Correct 179 ms 348 KB Output is correct
6 Correct 280 ms 348 KB Output is correct
7 Incorrect 31 ms 348 KB 1st lines differ - on the 1st token, expected: '26', found: '17'
8 Halted 0 ms 0 KB -

Subtask #3
0.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 397 ms 404 KB Output is correct
3 Correct 411 ms 596 KB Output is correct
4 Correct 432 ms 348 KB Output is correct
5 Correct 179 ms 348 KB Output is correct
6 Correct 280 ms 348 KB Output is correct
7 Incorrect 31 ms 348 KB 1st lines differ - on the 1st token, expected: '26', found: '17'
8 Halted 0 ms 0 KB -

Subtask #4
0.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 397 ms 404 KB Output is correct
3 Correct 411 ms 596 KB Output is correct
4 Correct 432 ms 348 KB Output is correct
5 Correct 179 ms 348 KB Output is correct
6 Correct 280 ms 348 KB Output is correct
7 Incorrect 31 ms 348 KB 1st lines differ - on the 1st token, expected: '26', found: '17'
8 Halted 0 ms 0 KB -

Subtask #5
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 397 ms 404 KB Output is correct
4 Correct 411 ms 596 KB Output is correct
5 Correct 432 ms 348 KB Output is correct
6 Correct 179 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 344 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 83 ms 348 KB Output is correct
11 Correct 187 ms 344 KB Output is correct
12 Correct 9 ms 436 KB Output is correct
13 Correct 374 ms 348 KB Output is correct
14 Correct 183 ms 348 KB Output is correct
15 Correct 92 ms 344 KB Output is correct
16 Correct 92 ms 344 KB Output is correct
17 Correct 77 ms 416 KB Output is correct
18 Correct 170 ms 348 KB Output is correct
19 Correct 374 ms 600 KB Output is correct
20 Correct 93 ms 344 KB Output is correct
21 Correct 84 ms 344 KB Output is correct
22 Correct 377 ms 344 KB Output is correct
23 Correct 392 ms 348 KB Output is correct
24 Correct 83 ms 348 KB Output is correct

Subtask #6
0.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 397 ms 404 KB Output is correct
4 Correct 411 ms 596 KB Output is correct
5 Correct 432 ms 348 KB Output is correct
6 Correct 179 ms 348 KB Output is correct
7 Correct 280 ms 348 KB Output is correct
8 Incorrect 31 ms 348 KB 1st lines differ - on the 1st token, expected: '26', found: '17'
9 Halted 0 ms 0 KB -

Subtask #7
0.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 397 ms 404 KB Output is correct
4 Correct 411 ms 596 KB Output is correct
5 Correct 432 ms 348 KB Output is correct
6 Correct 179 ms 348 KB Output is correct
7 Correct 280 ms 348 KB Output is correct
8 Incorrect 31 ms 348 KB 1st lines differ - on the 1st token, expected: '26', found: '17'
9 Halted 0 ms 0 KB -

Subtask #8
0.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 397 ms 404 KB Output is correct
4 Correct 411 ms 596 KB Output is correct
5 Correct 432 ms 348 KB Output is correct
6 Correct 179 ms 348 KB Output is correct
7 Correct 280 ms 348 KB Output is correct
8 Incorrect 31 ms 348 KB 1st lines differ - on the 1st token, expected: '26', found: '17'
9 Halted 0 ms 0 KB -

Subtask #9
0.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 397 ms 404 KB Output is correct
4 Correct 411 ms 596 KB Output is correct
5 Correct 432 ms 348 KB Output is correct
6 Correct 179 ms 348 KB Output is correct
7 Correct 280 ms 348 KB Output is correct
8 Incorrect 31 ms 348 KB 1st lines differ - on the 1st token, expected: '26', found: '17'
9 Halted 0 ms 0 KB -