제출 #1222979

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1222979		nikd	봉쇄 시간 (IOI23_closing)	C++20	8 / 100	112 ms	29112 KiB

closing

#include "closing.h"
#include <bits/stdc++.h>
#include <vector>
using ll = long long;
using namespace std;

int max_score(int n, int x, int y, long long k,
              std::vector<int> U, std::vector<int> V, std::vector<int> W)
{
    vector<ll> dst;
    vector<vector<array<ll, 2>>> adj(n);
    for(int i = 0; i<n-1; i++){
        adj[U[i]].push_back({V[i], W[i]});
        adj[V[i]].push_back({U[i],W[i]});
    }
    function<void(int, int, ll)> find_dst = [&](int v, int p, ll d){
        dst.push_back(d);
        for(auto [u, w]: adj[v]){
            if(u == p) continue;
            if(d+w > k) continue;
            find_dst(u, v, d+w);
        }
        return;
    };
    find_dst(x, -1, 0);
    find_dst(y, -1, 0);
    int sol = 0;
    sort(dst.begin(), dst.end());
    for(ll d: dst){
        if(k-d >= 0){
            k-= d;
            sol++;
        }
        else break;
    }
    return sol;
}

• Subtask 0: 0 / 0
• Subtask 1: 8 / 8
• Subtask 2: 0 / 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