Submission #931004

#	Time	Username	Problem	Language	Result	Execution time	Memory
931004		vjudge1	Rail (IOI14_rail)	C++17	30 / 100	51 ms	4692 KiB

rail

#include<bits/stdc++.h>

#include "rail.h"

using namespace std;

#define F first
#define S second
#define ll long long
#define maksim gay
// #define int ll
#define pb push_back
#define sz(s) (int)s.size()
#define pii pair<int,int>
#define all(v) v.begin(),v.end()
#define mem(a,i) memset(a,i,sizeof(a))
#define in insert
#define lb lower_bound
#define ub upper_bound
#define y1 yy
#define ppb pop_back

const int MAX=2e6+10;
const int inf=2e9;
const int N=2e5;
const int mod=998244353;

int d[MAX];
int use[MAX];
int d1[MAX];
int d2[MAX];

void findLocation(int n, int first, int location[], int stype[])
{
  int fin=-1;
  int st=0;
  for(int i=0;i<n;i++){
    stype[i]=0;
    if(i==st)continue;
    d[i]=getDistance(st,i);
    if(fin==-1||d[fin]>d[i])fin=i;
  }
  location[0]=first;
  stype[0]=1;
  if(n==1)return;
  for(int i=0;i<n;i++){
    if(i==fin)continue;
    d1[i]=getDistance(fin,i);
  }
  location[fin]=first+d[fin];
  stype[fin]=2;
  vector<pii> vec,vec1;
  for(int i=0;i<n;i++){
    if(i==st||i==fin)continue;
    if(d[i]==d[fin]+d1[i]){
      vec.pb({d[i],i});
    }
    else{
      vec1.pb({d1[i],i});
    }
  }
  sort(all(vec));
  int last=st;
  vector<pii> Op;
  Op.pb({location[st],st});
  for(auto [ff,v]:vec){
    int dd=getDistance(v,fin);
    if(dd==d1[v]){
      stype[v]=1;
      location[v]=location[fin]-d1[v];
      if(location[v]<=location[st])Op.pb({location[v],v});
    }
    else{
      stype[v]=2;
      for(int i=1;i<sz(Op);i++){
        int pos=dd-d1[Op[i].S];
        int f=location[Op[i].S]+pos;
        if(pos>0&&f<Op[i-1].F){
          location[v]=f;
          break;
        }
      }
    }
  }
  Op.clear();
  Op.pb({location[fin],fin});
  sort(all(vec1));
  for(auto [ff,v]:vec1){
    int dd=getDistance(v,st);
    if(dd==d[v]){
      stype[v]=2;
      location[v]=location[st]+d[v];
      if(location[v]>location[fin])Op.pb({location[v],v});
    }
    else{
      stype[v]=1;
      for(int i=1;i<sz(Op);i++){
        int pos=dd-d[Op[i].S];
        int f=location[v]-pos;
        if(pos>0&&f>Op[i-1].F){
          location[v]=f;
          break;
        }
      }
    }
  }
}

Compilation message (stderr)

rail.cpp: In function 'void findLocation(int, int, int*, int*)':
rail.cpp:63:7: warning: unused variable 'last' [-Wunused-variable]
   63 |   int last=st;
      |       ^~~~

• Subtask 1: 8 / 8
• Subtask 2: 22 / 22
• Subtask 3: 0 / 26
• Subtask 4: 0 / 44