Submission #349061

#TimeUsernameProblemLanguageResultExecution timeMemory
349061KerimHighway Tolls (IOI18_highway)C++17
69 / 100
312 ms19580 KiB
#include "highway.h"
#include "bits/stdc++.h"
#define MAXN 130009
#define INF 1000000007
#define mp(x,y) make_pair(x,y)
#define all(v) v.begin(),v.end()
#define pb(x) push_back(x)
#define wr cout<<"----------------"<<endl;
#define ppb() pop_back()
#define tr(ii,c) for(__typeof((c).begin()) ii=(c).begin();ii!=(c).end();ii++)
#define ff first
#define ss second
#define my_little_dodge 46
#define debug(x)  cerr<< #x <<" = "<< x<<endl;
using namespace std;
 
typedef long long ll;
typedef pair<int,int> PII;
template<class T>bool umin(T& a,T b){if(a>b){a=b;return 1;}return 0;}
template<class T>bool umax(T& a,T b){if(a<b){a=b;return 1;}return 0;}
int M,N;
PII edge[MAXN];
vector<int>A,B,AA,BB;
vector<PII>adj[MAXN],gr[MAXN];
bool check(){
	if(A.empty() or B.empty())
		return 0;	
	vector<int>col(N),w(M);
	tr(it,A)col[*it]=0;
	tr(it,B)col[*it]=1;
	for(int i=0;i<M;i++)
		w[i]=(col[edge[i].ff]==col[edge[i].ss]);
	return ask(w)%2;
}
int col[MAXN];
void dfs(int nd,int pr){
	tr(it,gr[nd])
		if(it->ff!=pr){
			col[it->ss]=1;
			dfs(it->ff,nd);
		}
}
void find_pair(int N, vector<int> U, vector<int> V, int a, int b) {
  	M=U.size();::N=N;
  	for(int i=0;i<M;i++)
  		edge[i]=mp(U[i],V[i]);
  	if(a==1 and b==2){//Subtask 5
	  	int hor=0,who=-1;
	  	for(int i=0;i<17;i++){
	  		A.clear();B.clear();
	  		for(int j=0;j<N;j++){
	  			if(j>>i&1)A.pb(j);
	  			else B.pb(j);
	  		}
	  		if(check()) hor^=(1<<i),who=i;
	  	}A.clear();B.clear();assert(who!=-1);
		for(int j=0;j<N;j++){
			if(j>>who&1)A.pb(j);
			else B.pb(j);
		}int sz=A.size();
		while(sz>1){
			AA=A;BB=B;
			A.clear();
			for(int i=0;i<sz/2;i++)A.pb(AA[i]);
			for(int i=sz/2;i<sz;i++)B.pb(AA[i]);
			if(!check()){
				B=BB;A.clear();
				for(int i=sz/2;i<sz;i++)
					A.pb(AA[i]);
			}sz=A.size();
		}
	  	answer(A[0],A[0]^hor);
	  	return;
	}
	vector<int>w(M,0),ans;
	ll dis=ask(w);
	int low=0,high=M-1;
	while(low<high){
		int mid=(low+high)>>1;
		for(int i=0;i<M;i++)
			w[i]=(i>mid);
		if(ask(w)==dis)
			high=mid;
		else
			low=mid+1;
	}PII v=edge[low];
	for(int i=0;i<M;i++){
		w[i]=1;
		gr[edge[i].ff].pb(mp(edge[i].ss,i));
		gr[edge[i].ss].pb(mp(edge[i].ff,i));
	}
	queue<int>p;
	vector<int>d(N,INF);
	p.push(v.ff);d[v.ff]=0;
	while(!p.empty()){
		int nd=p.front();p.pop();
		tr(it,gr[nd])
			if(umin(d[it->ff],d[nd]+1))
				w[it->ss]=0,p.push(it->ff);	
	}
	for(int i=0;i<N;i++)
		gr[i].clear();
	int ban=-1;
	for(int i=0;i<M;i++){
		if(edge[i]==v)ban=i;
		if(!w[i]){
			gr[edge[i].ff].pb(mp(edge[i].ss,i));
			gr[edge[i].ss].pb(mp(edge[i].ff,i));
		}
		else
			col[i]=2;	
	}
	dfs(v.ss,v.ff);
	for(int j=0;j<2;j++){
		for(int i=0;i<N;i++)
			adj[i].clear();
		vector<int>vis(N,0);
		int s=v.ff,tmp=0;if(j)s=v.ss;
		vector<int>id(N),par(M);
		for(int i=0;i<M;i++){
			if(i!=ban and col[i]==j){
				adj[edge[i].ff].pb(mp(edge[i].ss,i));
				adj[edge[i].ss].pb(mp(edge[i].ff,i));
			}
			if(col[i]==2)
				w[i]=1;
			else
				w[i]=0;
		}
		queue<int>q;q.push(s);vis[s]=1;
		while(!q.empty()){
			int nd=q.front();q.pop();
			tr(it,adj[nd])
				if(!vis[it->ff]){
					id[tmp++]=it->ss;par[it->ss]=it->ff;
					vis[it->ff]=1;q.push(it->ff);	
				}
		}
		int st=-1,en=tmp-1;
		while(st<en){
			int mid=(st+en+1)>>1;
			for(int i=0;i<tmp;i++)
				w[id[i]]=(i>=mid);
			if(ask(w)==dis)en=mid-1;
			else st=mid;
		}
		if(st==-1)
			ans.pb(s);
		else
			ans.pb(par[id[st]]);
	}
	answer(ans[0],ans[1]);
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...