Submission #1070785

#	Time	Username	Problem	Language	Result	Execution time	Memory
1070785		MohamedFaresNebili	Closing Time (IOI23_closing)	C++17	9 / 100	1098 ms	28620 KiB

closing

#include <bits/stdc++.h>
#include "closing.h"

        using namespace std;

        vector<pair<int, long long>> adj[200005];
        long long D[200005][2];
        void dfs(int X, int p, int st) {
            for(auto u : adj[X]) {
                if(u.first == p) continue;
                D[u.first][st] = D[X][st] + u.second;
                dfs(u.first, X, st);
            }
        }

        int max_score(int N, int X, int Y, long long K,
            vector<int> U, vector<int> V, vector<int> W) {
            for(int l = 0; l < N; l++) D[l][0] = D[l][1] = 2e18 + 7, adj[l].clear();
            for(int l = 0; l < N - 1; l++) {
                adj[U[l]].push_back({V[l], W[l]});
                adj[V[l]].push_back({U[l], W[l]});
            }
            if(X > Y) swap(X, Y);
            D[X][0] = 0, D[Y][1] = 0;
            dfs(X, X, 0); dfs(Y, Y, 1);
            int res = 0;
            for(int l = X; l >= 0; l--) {
                for(int i = X; i < N; i++) {
                    for(int j = Y; j >= 0; j--) {
                        for(int e = Y; e < N; e++) {
                            int lo = max(l, j), hi = min(i, e);
                            long long calc = 0;
                            if(lo > hi) {
                                for(int f = l; f <= i; f++)
                                    calc += D[f][0];
                                for(int f = j; f <= e; f++)
                                    calc += D[f][1];
                            }
                            else {
                                for(int f = l; f <= i; f++)
                                    calc += D[f][0];
                                for(int f = j; f <= e; f++)
                                    calc += D[f][1];
                                for(int f = lo; f <= hi; f++)
                                    calc -= min(D[f][0], D[f][1]);
                            }
                            if(calc <= K) res = max(res,  i - l + e - j + 2);
                        }
                    }
                }
            }
            return res;
        }

• Subtask 0: 0 / 0
• Subtask 1: 0 / 8
• Subtask 2: 9 / 9
• Subtask 3: 0 / 12
• Subtask 4: 0 / 14
• Subtask 5: 0 / 9
• Subtask 6: 0 / 11
• Subtask 7: 0 / 10
• Subtask 8: 0 / 10
• Subtask 9: 0 / 17