oj.uz

Submission #850496

# Submission time Handle Problem Language Result Execution time Memory
850496 2023-09-16T17:45:37 Z eltu0815 Closing Time (IOI23_closing) C++17
35 / 100
 224 ms 125648 KB 
#include "closing.h"

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

typedef long long ll;
typedef pair<int, int> pii;

int visited[MAX], parent[MAX];
vector<pii> graph[MAX];

ll cost1[MAX], cost2[MAX];
ll dist1[MAX], dist2[MAX];

vector<pair<ll, ll> > vec;
multiset<pair<ll, ll> > st1;
multiset<pair<ll, ll> > st2;

void init(int node, int par) {
    parent[node] = par;
    for(auto [v, w] : graph[node]) if(v != par) init(v, node);
}

void dfs(int node, int par, ll dist[]) {
    for(auto [v, w] : graph[node]) if(v != par) {
        dist[v] = dist[node] + w;
        dfs(v, node, dist);
    }
}

ll c[MAX];
vector<int> path;

int max_score(int N, int X, int Y, long long K, vector<int> U, vector<int> V, vector<int> W)
{
    path.clear();
    for(int i = 0; i < N; ++i) graph[i].clear();
    for(int i = 0; i < N; ++i) dist1[i] = dist2[i] = visited[i] = c[i] = 0;
    for(int i = 0; i < N - 1; ++i) {
        graph[U[i]].push_back({V[i], W[i]});
        graph[V[i]].push_back({U[i], W[i]});
    }
    
    init(0, -1);
    dfs(X, -1, dist1);
    dfs(Y, -1, dist2);
    
    int mx = 0; ll sum = 0;
    priority_queue<pair<ll, int> > pq;
    pq.push({0, X}); pq.push({0, Y});
    while(!pq.empty()) {
        ll d = -pq.top().first;
        int node = pq.top().second;
        pq.pop();
        
        if(sum + d > K) break;
        ++mx; sum += d; visited[node] = 1;
        for(auto [v, w] : graph[node]) {
            if(visited[v]) continue;
            pq.push({-(d + w), v});
        }
    }
    
    int node = X;
    for(int i = 0; i < N; ++i) visited[i] = 0;
    while(node != -1) visited[node]++, node = parent[node];
    
    node = Y; int flag = 0;
    while(node != -1) {
        if(visited[node] == 1 && !flag) flag = 1;
        else if(flag) visited[node]--;
        else visited[node]++;
        node = parent[node];
    }
    
    node = X;
    while(node != -1 && visited[node]) {
        path.push_back(node);
        node = parent[node];
    }
    path.pop_back();
    
    node = Y;
    vector<int> path2;
    while(node != -1 && visited[node]) {
        path2.push_back(node);
        node = parent[node];
    }
    reverse(path2.begin(), path2.end());
    for(auto v : path2) path.push_back(v);
    
    int p = 0, q = path.size() - 1, r = 0; cost1[0] = 0;
    for(int i = 1; i <= 2 * N; ++i) cost1[i] = (ll)(2e18);
    for(int i = 1; i <= 2 * N; ++i) {
        int u = p == (int)path.size() ? -1 : path[p];
        int v = q == -1 ? -1 : path[q];
        
        if(u == -1 && v == -1) break;
        if(u == v) r = i;
        if(u == -1) {
            cost1[i] = cost1[i - 1] + dist2[v] - c[v];
            c[v] = dist2[v]; --q;
        }
        else if(v == -1) {
            cost1[i] = cost1[i - 1] + dist1[u] - c[u];
            c[u] = dist1[u]; ++p;
        }
        else if(dist1[u] - c[u] <= dist2[v] - c[v]) {
            cost1[i] = cost1[i - 1] + dist1[u] - c[u];
            c[u] = dist1[u]; ++p;
        }
        else {
            cost1[i] = cost1[i - 1] + dist2[v] - c[v];
            c[v] = dist2[v]; --q;
        }
    }
    for(int i = 0; i < r; ++i) cost1[i] = (ll)(2e18);
    
    vec.clear(); st1.clear(); st2.clear();
    for(int i = 1; i <= 2 * N; ++i) cost2[i] = (ll)(2e18);
    for(int i = 0; i < N; ++i) if(!visited[i]) {
        ll c1 = min(dist1[i], dist2[i]);
        ll c2 = max(dist1[i], dist2[i]);
        st1.insert({c1, c2}); st2.insert({c2, c1});
    }
    for(int i = 1; i <= 2 * N; ++i) {
        if(st1.empty()) break;
        if(st2.empty() || vec.empty() || vec.back().first + st1.begin()->first <= st2.begin()->first) {
            ll d1 = st1.begin()->first;
            ll d2 = st1.begin()->second;
            st1.erase(st1.begin());
            cost2[i] = cost2[i - 1] + d1;
            vec.push_back({d1, d2});
            
            if(d2 != -1) {
                st1.insert({d2 - d1, -1});
                assert(st2.find({d2, d1}) != st2.end());
                st2.erase(st2.find({d2, d1}));
            }
        }
        else {
            ll d2 = st2.begin()->first;
            ll d1 = st2.begin()->second;
            cost2[i] = cost2[i - 2] + d2;
            st2.erase(st2.begin());
            
            st1.insert(vec.back());
            vec.pop_back();
            
            vec.push_back({d1, d2});
            vec.push_back({d2 - d1, -1});
        }
    }
    
    for(int i = 2 * N, p = 0; i >= 0; --i) {
        while(p < 2 * N && cost1[i] + cost2[p + 1] <= K) ++p;
        if(cost1[i] + cost2[p] <= K) mx = max(mx, i + p);
    }
    return mx;
}

/*
4
7 3 2 10
3 1 2
3 0 3
1 2 4
2 4 2
2 5 5
5 6 3

4 0 3 18
0 1 18
1 2 1
2 3 19

8 0 6 100000
1 6 40000
3 6 30000
6 7 30000
1 2 10000
0 2 40000
0 4 100000
0 5 100000

12 0 11 9
0 1 1
1 11 10
1 2 1
2 3 1
3 4 1
4 5 1
5 6 1
6 7 1
7 8 1
8 9 1
9 10 1
*/

Subtask #0
0.0 / 0.0

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

Subtask #1
0.0 / 8.0

# Verdict Execution time Memory Grader output
1 Runtime error 224 ms 125648 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #2
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 20824 KB Output is correct
2 Correct 3 ms 20828 KB Output is correct
3 Correct 3 ms 20828 KB Output is correct
4 Correct 4 ms 20828 KB Output is correct
5 Correct 3 ms 20828 KB Output is correct
6 Correct 3 ms 20956 KB Output is correct
7 Correct 4 ms 20828 KB Output is correct
8 Correct 3 ms 20824 KB Output is correct
9 Correct 3 ms 20828 KB Output is correct
10 Correct 3 ms 20828 KB Output is correct
11 Correct 3 ms 20828 KB Output is correct

Subtask #3
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 20824 KB Output is correct
2 Correct 3 ms 20828 KB Output is correct
3 Correct 3 ms 20828 KB Output is correct
4 Correct 4 ms 20828 KB Output is correct
5 Correct 3 ms 20828 KB Output is correct
6 Correct 3 ms 20956 KB Output is correct
7 Correct 4 ms 20828 KB Output is correct
8 Correct 3 ms 20824 KB Output is correct
9 Correct 3 ms 20828 KB Output is correct
10 Correct 3 ms 20828 KB Output is correct
11 Correct 3 ms 20828 KB Output is correct
12 Correct 3 ms 20828 KB Output is correct
13 Correct 3 ms 20828 KB Output is correct
14 Correct 3 ms 20828 KB Output is correct
15 Correct 3 ms 20828 KB Output is correct
16 Correct 3 ms 20828 KB Output is correct
17 Correct 3 ms 20828 KB Output is correct
18 Correct 3 ms 20828 KB Output is correct
19 Correct 4 ms 21080 KB Output is correct
20 Correct 4 ms 20828 KB Output is correct
21 Correct 4 ms 20828 KB Output is correct
22 Correct 4 ms 20828 KB Output is correct
23 Correct 4 ms 20828 KB Output is correct
24 Correct 3 ms 20828 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 20824 KB Output is correct
2 Correct 3 ms 20828 KB Output is correct
3 Correct 3 ms 20828 KB Output is correct
4 Correct 4 ms 20828 KB Output is correct
5 Correct 3 ms 20828 KB Output is correct
6 Correct 3 ms 20956 KB Output is correct
7 Correct 4 ms 20828 KB Output is correct
8 Correct 3 ms 20824 KB Output is correct
9 Correct 3 ms 20828 KB Output is correct
10 Correct 3 ms 20828 KB Output is correct
11 Correct 3 ms 20828 KB Output is correct
12 Correct 3 ms 20828 KB Output is correct
13 Correct 3 ms 20828 KB Output is correct
14 Correct 3 ms 20828 KB Output is correct
15 Correct 3 ms 20828 KB Output is correct
16 Correct 3 ms 20828 KB Output is correct
17 Correct 3 ms 20828 KB Output is correct
18 Correct 3 ms 20828 KB Output is correct
19 Correct 4 ms 21080 KB Output is correct
20 Correct 4 ms 20828 KB Output is correct
21 Correct 4 ms 20828 KB Output is correct
22 Correct 4 ms 20828 KB Output is correct
23 Correct 4 ms 20828 KB Output is correct
24 Correct 3 ms 20828 KB Output is correct
25 Correct 5 ms 21016 KB Output is correct
26 Correct 6 ms 21596 KB Output is correct
27 Correct 5 ms 21596 KB Output is correct
28 Correct 4 ms 21164 KB Output is correct
29 Correct 5 ms 21340 KB Output is correct
30 Correct 6 ms 21596 KB Output is correct
31 Correct 6 ms 21340 KB Output is correct
32 Correct 4 ms 21340 KB Output is correct

Subtask #5
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 20828 KB Output is correct
2 Correct 3 ms 20824 KB Output is correct
3 Correct 3 ms 20828 KB Output is correct
4 Correct 3 ms 20828 KB Output is correct
5 Correct 4 ms 20828 KB Output is correct
6 Correct 3 ms 20828 KB Output is correct
7 Correct 3 ms 20828 KB Output is correct
8 Correct 3 ms 20960 KB Output is correct
9 Runtime error 18 ms 42072 KB Execution killed with signal 6
10 Halted 0 ms 0 KB -

Subtask #6
0.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 20828 KB Output is correct
2 Correct 3 ms 20824 KB Output is correct
3 Correct 3 ms 20828 KB Output is correct
4 Correct 3 ms 20828 KB Output is correct
5 Correct 4 ms 20828 KB Output is correct
6 Correct 3 ms 20828 KB Output is correct
7 Correct 3 ms 20956 KB Output is correct
8 Correct 4 ms 20828 KB Output is correct
9 Correct 3 ms 20824 KB Output is correct
10 Correct 3 ms 20828 KB Output is correct
11 Correct 3 ms 20828 KB Output is correct
12 Correct 3 ms 20828 KB Output is correct
13 Correct 3 ms 20828 KB Output is correct
14 Correct 3 ms 20828 KB Output is correct
15 Correct 3 ms 20828 KB Output is correct
16 Correct 3 ms 20828 KB Output is correct
17 Correct 3 ms 20828 KB Output is correct
18 Correct 3 ms 20828 KB Output is correct
19 Correct 3 ms 20828 KB Output is correct
20 Correct 3 ms 20960 KB Output is correct
21 Runtime error 18 ms 42072 KB Execution killed with signal 6
22 Halted 0 ms 0 KB -

Subtask #7
0.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 20828 KB Output is correct
2 Correct 3 ms 20824 KB Output is correct
3 Correct 3 ms 20828 KB Output is correct
4 Correct 3 ms 20828 KB Output is correct
5 Correct 4 ms 20828 KB Output is correct
6 Correct 3 ms 20828 KB Output is correct
7 Correct 3 ms 20956 KB Output is correct
8 Correct 4 ms 20828 KB Output is correct
9 Correct 3 ms 20824 KB Output is correct
10 Correct 3 ms 20828 KB Output is correct
11 Correct 3 ms 20828 KB Output is correct
12 Correct 3 ms 20828 KB Output is correct
13 Correct 3 ms 20828 KB Output is correct
14 Correct 3 ms 20828 KB Output is correct
15 Correct 3 ms 20828 KB Output is correct
16 Correct 3 ms 20828 KB Output is correct
17 Correct 3 ms 20828 KB Output is correct
18 Correct 3 ms 20828 KB Output is correct
19 Correct 3 ms 20828 KB Output is correct
20 Correct 4 ms 21080 KB Output is correct
21 Correct 4 ms 20828 KB Output is correct
22 Correct 4 ms 20828 KB Output is correct
23 Correct 4 ms 20828 KB Output is correct
24 Correct 4 ms 20828 KB Output is correct
25 Correct 3 ms 20828 KB Output is correct
26 Correct 3 ms 20828 KB Output is correct
27 Correct 3 ms 20960 KB Output is correct
28 Runtime error 18 ms 42072 KB Execution killed with signal 6
29 Halted 0 ms 0 KB -

Subtask #8
0.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 20828 KB Output is correct
2 Correct 3 ms 20824 KB Output is correct
3 Correct 3 ms 20828 KB Output is correct
4 Correct 3 ms 20828 KB Output is correct
5 Correct 4 ms 20828 KB Output is correct
6 Correct 3 ms 20828 KB Output is correct
7 Correct 3 ms 20956 KB Output is correct
8 Correct 4 ms 20828 KB Output is correct
9 Correct 3 ms 20824 KB Output is correct
10 Correct 3 ms 20828 KB Output is correct
11 Correct 3 ms 20828 KB Output is correct
12 Correct 3 ms 20828 KB Output is correct
13 Correct 3 ms 20828 KB Output is correct
14 Correct 3 ms 20828 KB Output is correct
15 Correct 3 ms 20828 KB Output is correct
16 Correct 3 ms 20828 KB Output is correct
17 Correct 3 ms 20828 KB Output is correct
18 Correct 3 ms 20828 KB Output is correct
19 Correct 3 ms 20828 KB Output is correct
20 Correct 4 ms 21080 KB Output is correct
21 Correct 4 ms 20828 KB Output is correct
22 Correct 4 ms 20828 KB Output is correct
23 Correct 4 ms 20828 KB Output is correct
24 Correct 4 ms 20828 KB Output is correct
25 Correct 3 ms 20828 KB Output is correct
26 Correct 5 ms 21016 KB Output is correct
27 Correct 6 ms 21596 KB Output is correct
28 Correct 5 ms 21596 KB Output is correct
29 Correct 4 ms 21164 KB Output is correct
30 Correct 5 ms 21340 KB Output is correct
31 Correct 6 ms 21596 KB Output is correct
32 Correct 6 ms 21340 KB Output is correct
33 Correct 4 ms 21340 KB Output is correct
34 Correct 3 ms 20828 KB Output is correct
35 Correct 3 ms 20960 KB Output is correct
36 Runtime error 18 ms 42072 KB Execution killed with signal 6
37 Halted 0 ms 0 KB -

Subtask #9
0.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 20828 KB Output is correct
2 Correct 3 ms 20824 KB Output is correct
3 Correct 3 ms 20828 KB Output is correct
4 Correct 3 ms 20828 KB Output is correct
5 Correct 4 ms 20828 KB Output is correct
6 Correct 3 ms 20828 KB Output is correct
7 Correct 3 ms 20956 KB Output is correct
8 Correct 4 ms 20828 KB Output is correct
9 Correct 3 ms 20824 KB Output is correct
10 Correct 3 ms 20828 KB Output is correct
11 Correct 3 ms 20828 KB Output is correct
12 Correct 3 ms 20828 KB Output is correct
13 Correct 3 ms 20828 KB Output is correct
14 Correct 3 ms 20828 KB Output is correct
15 Correct 3 ms 20828 KB Output is correct
16 Correct 3 ms 20828 KB Output is correct
17 Correct 3 ms 20828 KB Output is correct
18 Correct 3 ms 20828 KB Output is correct
19 Correct 3 ms 20828 KB Output is correct
20 Correct 4 ms 21080 KB Output is correct
21 Correct 4 ms 20828 KB Output is correct
22 Correct 4 ms 20828 KB Output is correct
23 Correct 4 ms 20828 KB Output is correct
24 Correct 4 ms 20828 KB Output is correct
25 Correct 3 ms 20828 KB Output is correct
26 Correct 5 ms 21016 KB Output is correct
27 Correct 6 ms 21596 KB Output is correct
28 Correct 5 ms 21596 KB Output is correct
29 Correct 4 ms 21164 KB Output is correct
30 Correct 5 ms 21340 KB Output is correct
31 Correct 6 ms 21596 KB Output is correct
32 Correct 6 ms 21340 KB Output is correct
33 Correct 4 ms 21340 KB Output is correct
34 Correct 3 ms 20828 KB Output is correct
35 Correct 3 ms 20960 KB Output is correct
36 Runtime error 18 ms 42072 KB Execution killed with signal 6
37 Halted 0 ms 0 KB -