Submission #1219033

#	Time	Username	Problem	Language	Result	Execution time	Memory
1219033		JooDdae	통행료 (IOI18_highway)	C++20	5 / 100	6 ms	1340 KiB

highway

#include "highway.h"
#include<bits/stdc++.h>
using namespace std;

void find_pair(int N, vector<int> U, vector<int> V, int A, int B) {
    int M = U.size();
    vector<int> w(M, 0), d(N, 0), p(N, 0);
    int D = ask(w);
    
    vector<vector<array<int, 2>>> v(N);
    for(int i=0;i<M;i++) {
        v[U[i]].push_back({V[i], i});
        v[V[i]].push_back({U[i], i});
    }

    queue<int> q;
    q.push(0), d[0] = 1;
    while(!q.empty()) {
        int u = q.front(); q.pop();
        for(auto [x, id] : v[u]) if(!d[x]) q.push(x), d[x] = d[u]+1, p[x] = id;
    }

    for(int i=0;i<N;i++) if(d[i] == D/A+1) {
        w[p[i]] = 1;
        if(ask(w) != D) {
            answer(0, i);
            return;
        }
        w[p[i]] = 0;
    }
}

• Subtask 1: 5 / 5
• Subtask 2: 0 / 7
• Subtask 3: 0 / 6
• Subtask 4: 0 / 33
• Subtask 5: 0 / 18
• Subtask 6: 0 / 31