Submission #1064140

#	Time	Username	Problem	Language	Result	Execution time	Memory
1064140		Theo830	Rail (IOI14_rail)	C++17	30 / 100	854 ms	262144 KiB

rail

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll INF = 1e18;
ll MOD = 998244353;
typedef pair<ll,ll> ii;
#define iii pair<ll,ii>
#define f(i,a,b) for(ll i = a;i < b;i++)
#define pb push_back
#define vll vector<ll>
#define F first
#define S second
#define all(x) (x).begin(), (x).end()
///I hope I will get uprating and don't make mistakes
///I will never stop programming
///sqrt(-1) Love C++
///Please don't hack me
///@TheofanisOrfanou Theo830
///Think different approaches (bs,dp,greedy,graphs,shortest paths,mst)
///Stay Calm
///Look for special cases
///Beware of overflow and array bounds
///Think the problem backwards
///Training
#include "rail.h"
void findLocation(int n, int first, int location[], int stype[]){
    location[0] = first;
    stype[0] = 1;
    ll dist[n][n];
    f(i,0,n){
        dist[i][i] = 0;
        f(j,i+1,n){
            dist[i][j] = dist[j][i] = getDistance(i,j);
        }
    }
    ll mini = INF;
    ll pos;
    f(i,1,n){
        if(mini > dist[0][i]){
            mini = dist[0][i];
            pos = i;
        }
    }
    stype[pos] = 2;
    location[pos] = first + dist[0][pos];
    ll ex = 2;
    set<iii>dista;
    bool v[n] = {0};
    v[0] = v[pos] = 1;
    f(i,0,n){
        dista.insert(iii(dist[i][0],ii(i,0)));
        dista.insert(iii(dist[i][pos],ii(i,pos)));
    }
    while(ex != n){
        ll prev;
        while(1){
            auto it = dista.begin();
            iii w = (*it);
            dista.erase(w);
            if(!v[w.S.F]){
                mini = w.F;
                prev = w.S.S;
                pos = w.S.F;
                break;
            }
        }
        if(stype[prev] == 1){
            stype[pos] = 2;
            location[pos] = location[prev] + mini;
        }
        else{
            stype[pos] = 1;
            location[pos] = location[prev] - mini;
        }
        v[pos] = 1;
        ex++;
        f(i,0,n){
            dista.insert(iii(dist[i][pos],ii(i,pos)));
        }
    }
}

Compilation message (stderr)

rail.cpp: In function 'void findLocation(int, int, int*, int*)':
rail.cpp:45:40: warning: 'pos' may be used uninitialized in this function [-Wmaybe-uninitialized]
   45 |     location[pos] = first + dist[0][pos];
      |                             ~~~~~~~~~~~^

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