제출 #1198793

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1198793		simplemind_31	통행료 (IOI18_highway)	C++20	51 / 100	75 ms	14328 KiB

highway

#include "highway.h"
#include <bits/stdc++.h>
#define ALL(x) x.begin(),x.end()
using namespace std;
typedef long long ll;
vector<bool> visited1,visited2,bfstree1,bfstree2;
vector<vector<pair<int,int>>> grafo;
vector<int> dist1,dist2,aske;
vector<pair<int,int>> padre1,padre2;
vector<int> vv,uu;
int n,m;
ll res,a,b;
void dfs(int now,int ante){
  for(auto u:grafo[now]){
    if(u.first==ante){
      continue;
    }
    padre2[u.first].first=now;
    padre2[u.first].second=u.second;
    dist2[u.first]=dist2[now]+1;
    dfs(u.first,now);
  }
}
int bs2(int xx,int ante){
  dist2.assign(n,-1);
  padre2.assign(n,{-1,-1});
  dist2[xx]=0;
  padre2[xx]={xx,-1};
  dfs(xx,ante);
  int con=0,con1=0;
  vector<int> aske111(aske),second;
  for(int i=0;i<n;i++){
    if(dist2[i]>0){
      aske111[padre2[i].second]=0;
    }
  }
  ll ressss=(res-ask(aske111))/(b-a);
  ll resnow=res-ressss*(b-a);
  if(ressss==0){
    return xx;
  }
  vector<int> posibilidades;
  for(int i=0;i<n;i++){
    if(dist2[i]==ressss){
      posibilidades.push_back(i);
    }
  }
  int iz=0,de=posibilidades.size()-1;
  while(iz<de){
    int mid=(iz+de)>>1;
    for(int i=iz;i<=mid;i++){
      aske111[padre2[posibilidades[i]].second]=1;
    }
    if(ask(aske111)!=resnow){
      for(int i=iz;i<=mid;i++){
        aske111[padre2[posibilidades[i]].second]=0;
      }
      de=mid;
    }else{
      iz=mid+1;
    }
  }
  return posibilidades[iz];
}
void find_pair(int N,vector<int> U,vector<int> V, int A, int B){
  a=A;b=B;n=N;
  m=U.size();
  grafo.resize(n);
  vv=V;uu=U;
  dist1.assign(n,-1);
  padre1.assign(n,{-1,-1});
  visited1.assign(n,false);
  bfstree1.assign(m,false);
  bfstree2=bfstree1;
  visited2=visited1;
  dist2=dist1;
  padre2=padre1;
  for(int i=0;i<m;i++){
    grafo[U[i]].push_back({V[i],i});
    grafo[V[i]].push_back({U[i],i});
  }
  aske.assign(m,0);
  res=ask(aske);
  int iz=0,de=m-1;
  while(iz<de){
    int mid=(iz+de)>>1;
    for(int i=iz;i<=mid;i++){
      aske[i]=1;
    }
    if(ask(aske)==res){
      iz=mid+1;
    }else{
      for(int i=iz;i<=mid;i++){
        aske[i]=0;
      }
      de=mid;
    }
  }
  int x=U[iz],y=V[iz];
  aske.assign(m,1);
  aske[iz]=0;
  res/=a;
  res*=b;
  res+=-b+a;
  queue<int> bfs;
  bfs.push(x);
  dist1[x]=0;
  visited1[x]=true;
  bfs.push(y);
  dist1[y]=0;
  visited1[y]=true;
  while(!bfs.empty()){
    int now=bfs.front();
    bfs.pop();
    for(auto u:grafo[now]){
      if(visited1[u.first]){
        continue;
      }
      visited1[u.first]=true;
      dist1[u.first]=dist1[now]+1;
      padre1[u.first]={now,u.second};
      bfstree1[u.second]=true;
      bfs.push(u.first);
    }
  }
  grafo.clear();
  grafo.resize(n);
  for(int i=0;i<m;i++){
    if(bfstree1[i]){
      grafo[U[i]].push_back({V[i],i});
      grafo[V[i]].push_back({U[i],i});
    }
  }
  int i=bs2(x,y),j=bs2(y,x);
  answer(i,j);
}

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