Submission #823323

#	Time	Username	Problem	Language	Result	Execution time	Memory
823323		Trumling	Highway Tolls (IOI18_highway)	C++14	6 / 100	105 ms	4928 KiB

highway

#include "highway.h"
#include<bits/stdc++.h>
using namespace  std;
#define F first 
#define S second 
#define all(x) x.begin(),x.end()  
typedef long long ll;
#define INF 99999999999999
#define pb  push_back

vector<vector<pair<int,int> > >g;
vector<int>d,R;


void dfs(int start,int pre)
{
    for(pair<int,int> x:g[start])
        if(x.F!=pre)
        {
            d[x.F]=d[start]+1;
            R[x.F]=x.S;
            dfs(x.F,start);
        }
}

void find_pair(int N, vector<int> U, vector<int> V, int A, int B) {
  int M = U.size();
  g.assign(N,vector<pair<int,int> >());
  d.assign(N,0);
  R.assign(N,0);


vector<int>a(M,0);
ll ans=ask(a);

vector<int>v;

for(int i=ans/A/2;i<M;i+=ans/A)
v.pb(i);

 ll l=0,r=v.size()-1;
 ll idx=0;
//cout<<'\n';
 
 while(l<r)
 {
    
    ll  mid=(l+r)/2;
    for(int  i=l;i<=mid;i++)
    a[v[i]]=(idx+1)%2;
 
    ll curr=ask(a);
    if(idx==0)
    {
        if(curr>ans)
        {
           r=mid; 
           idx=1;
        }
        else
        l=mid+1;
    }
    else
    {
        if(curr==ans)
        {
            r=mid;
            idx=0;
        }
        else
        l=mid+1;
    }
 }
 a.assign(M,0);
 ll m=v[l];
 l=m;
 r=M-1;

 while(l<r)
 {
   // cout<<l<<' '<<r<<'\n';
    ll mid=(l+r)/2;

    a[mid]=1;
    ll curr=ask(a);

    if(curr>ans)
    l=mid+1;
    else
    r=mid;

    a[mid]=0;
 }
 
 answer(l,l-ans/A);

}

/*
6 5 1 2 
0 4
0 1  
1 2
2 3
3 4
4 5*/

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