oj.uz

Submission #1082036

# Submission time Handle Problem Language Result Execution time Memory
1082036 2024-08-30T15:12:37 Z LaMatematica14 Closing Time (IOI23_closing) C++17
8 / 100
 122 ms 44872 KB 
#include <bits/stdc++.h>
using namespace std; 

struct tree {
    vector<vector<array<long long, 2>>> adj;
    void dfs(int ba, vector<long long> &dist, vector<int> &p) {
        for (auto x : adj[ba]) {
            if (x[0] == p[ba]) continue;
            dist[x[0]] = dist[ba]+x[1];
            p[x[0]] = ba;
            dfs(x[0], dist, p);
        }
    }

    int m(int N, int X, int Y, long long K, vector<int> &U, vector<int> &V, vector<int> &W) {
        vector<long long> dx(N, 0);
        vector<long long> dy(N, 0);
        vector<int> px(N, -1);
        vector<int> py(N, -1);
        adj.resize(N);
        for (long long i = 0; i < N-1; i++ ) {
            adj[U[i]].push_back({V[i], W[i]});
            adj[V[i]].push_back({U[i], W[i]});
        }
        
        dfs(X, dx, px);
        dfs(Y, dy, py);

        vector<int> acx(N, 0); vector<int> acy(N, 0); // ho gia' comprato la x,y
        vector<int> rex(N, 0); vector<int> rey(N, 0); //
        priority_queue<array<long long, 3>> d;
        for (long long i = 0; i < N; i++) {
            if (i == X) {
                acx[X] = 1; rex[X] = 1;
                for (auto x : adj[X]) {
                    if (x[0] == px[X]) continue;
                    rex[x[0]] = 1;
                }
            }
            else if (i == Y) {
                acy[Y] = 1; rey[Y] = 1;
                for (auto x : adj[Y]) {
                    if (x[0] == py[Y]) continue;
                    rey[x[0]] = 1;
                }
            }
            else if (dx[i] < dy[i]) d.push({-dx[i], i, 0});
            else d.push({-dy[i], i, 1});
        }
        vector<int> done(N, 0);
        long long att = 0, k = 2;
        while (k < 2*N && att-d.top()[0] <= K) {
            att -= d.top()[0]; k++;
            auto a = d.top(); d.pop();
            if (a[2]) {
                acy[a[1]] = 1;
                if (!done[a[1]] && rex[a[1]] == 1) {
                    done[a[1]] =1 ;
                    d.push({-abs(dx[a[1]]-dy[a[1]]), a[1], 0});
                }
                for (auto x : adj[a[1]]) {
                    if (x[0] == py[a[1]]) continue;
                    rey[x[0]] = 1;
                    if (!done[x[0]] && acx[x[0]] == 1) {
                        done[x[0]] = 1;
                        d.push({-abs(dx[x[0]]-dy[x[0]]), x[0], 1});
                    }
                }
                
            }
            else {
                acx[a[1]] = 1;
                if (!done[a[1]] && rey[a[1]] == 1) {
                    done[a[1]] = 1;
                    d.push({-abs(dx[a[1]]-dy[a[1]]), a[1], 1});
                }
                for (auto x : adj[a[1]]) {
                    if (x[0] == px[a[1]]) continue;
                    rex[x[0]] = 1;
                    if (!done[x[0]] && acy[x[0]] == 1) {
                        done[x[0]] = 1;
                        d.push({-abs(dx[x[0]]-dy[x[0]]), x[0], 0});
                    }
                }
            }
            if (a[1] == -1) continue;
        }
        return k;
    }

};

int max_score(int N, int X, int Y, long long K,vector<int> U, vector<int> V, vector<int> W) {
    tree a;
    return a.m(N, X, Y, K, U, V, W);
}

Subtask #0
0.0 / 0.0

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

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 91 ms 41676 KB Output is correct
2 Correct 122 ms 44872 KB Output is correct
3 Correct 60 ms 5724 KB Output is correct

Subtask #2
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '30', found: '17'
3 Halted 0 ms 0 KB -

Subtask #3
0.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '30', found: '17'
3 Halted 0 ms 0 KB -

Subtask #4
0.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '30', found: '17'
3 Halted 0 ms 0 KB -

Subtask #5
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '30', found: '17'
4 Halted 0 ms 0 KB -

Subtask #6
0.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '30', found: '17'
4 Halted 0 ms 0 KB -

Subtask #7
0.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '30', found: '17'
4 Halted 0 ms 0 KB -

Subtask #8
0.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '30', found: '17'
4 Halted 0 ms 0 KB -

Subtask #9
0.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '30', found: '17'
4 Halted 0 ms 0 KB -