Submission #465416

#	Time	Username	Problem	Language	Result	Execution time	Memory
465416		MatheusLealV	Highway Tolls (IOI18_highway)	C++17	51 / 100	243 ms	14148 KiB

highway

#include "highway.h"
#define f first
#define ss second
#define pb push_back
#define mp make_pair
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> pii;
#define N 90005
vector<int> caras;
vector<pii> grafo[N];
int pai[N], n, id_pai[N],A,B;
void dfs(int x, int p){
	pai[x]=p;
	caras.pb(x);
	for(auto v: grafo[x]){
		if(v.f == p) continue;
		id_pai[v.f] =v.ss;
		dfs(v.f, x);
	}
}

int get(int root){
  caras.clear();
  dfs(root,root);
  int ini = 1, fim = sz(caras)-1,mid,best=-1;
  vector<int> w(n-1);
  int dist = ask(w)/A;
  while(fim >=ini){
    mid = (ini + fim)/2;
    vector<int> w(n-1);
    for(int i = 1; i <= mid; i++) w[id_pai[caras[i]]] = 1;
    int dmid = ask(w);
    if(dmid == B*dist){
      best=mid;
      fim=mid-1;
    }else ini=mid+1;
  }
  return caras[best];
}
void find_pair(int N_, std::vector<int> U, std::vector<int> V, int AA, int BB) {
  n=N_;A=AA;B=BB;
  if(sz(U) != n-1) return;
  for(int i = 0; i < sz(U); i++){
  	grafo[U[i]].pb({V[i],i}); grafo[V[i]].pb({U[i],i});
  }
  int S = get(0);
  int T = get(S);
  answer(S, T);
}

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