Submission #436737

#TimeUsernameProblemLanguageResultExecution timeMemory
436737TangentRail (IOI14_rail)C++17
56 / 100
949 ms98756 KiB
#include "rail.h"
#include "bits/stdc++.h"
 
using namespace std;
 
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vii;
typedef vector<ll> vll;
typedef vector<pii> vpii;
typedef vector<pll> vpll;
typedef vector<vii> vvii;
typedef vector<vll> vvll;
typedef vector<vpii> vvpii;
typedef vector<vpll> vvpll;
 
#define ffor(i, a, b) for (ll i = (a); i < (ll)(b); i++)
#define fford(i, a, b) for (ll i = (a); i > (ll)(b); i--)
#define rep(i, n) ffor(i, 0, n)
#define forin(x, a) for (auto &x: a)
#define all(a) a.begin(), a.end()

vvii memo;
int dist(int i, int j) {
    if (i > j) {
        swap(i, j);
    }
    if (!memo[i][j]) {
        memo[i][j] = getDistance(i, j);
    }
    return memo[i][j];
}

void findLocation(int N, int first, int location[], int stype[]) {
    memo.assign(N, vii(N));
    location[0] = first;
    stype[0] = 1;
    int closest = 1, mindist = dist(0, 1);
    ffor(i, 2, N) {
        if (dist(0, i) < mindist) {
            mindist = dist(0, i);
            closest = i;
        }
    }

    location[closest] = first + dist(0, closest);
    stype[closest] = 2;

    vii left, right;
    ffor(i, 1, N) {
        if (i != closest) {
            if (dist(0, i) == dist(0, closest) + dist(closest, i)) {
                left.emplace_back(i);
            } else {
                right.emplace_back(i);
            }
        }
    }

    int bound = closest;
    while(!left.empty()) {
        int closest2 = -1, mindist2 = 10000000;
        forin(i, left) {
            if (dist(i, bound) < mindist2) {
                mindist2 = dist(i, bound);
                closest2 = i;
            }
        }
        location[closest2] = location[bound] - dist(closest2, bound);
        stype[closest2] = 1;
        vii left2;
        int nbound = bound;
        forin(i, left) {
            if (i != closest2) {
                if (dist(bound, i) == dist(bound, closest2) + dist(closest2, i)) {
                    location[i] = location[closest2] + dist(closest2, i);
                    stype[i] = 2;
                    if (location[i] < location[nbound]) {
                        nbound = i;
                    }
                } else {
                    left2.emplace_back(i);
                }
            }
        }
        left = move(left2);
        bound = nbound;
    }

    bound = 0;
    while(!right.empty()) {
        int closest2 = -1, mindist2 = 10000000;
        forin(i, right) {
            if (dist(i, bound) < mindist2) {
                mindist2 = dist(i, bound);
                closest2 = i;
            }
        }
        location[closest2] = location[bound] + dist(closest2, bound);
        stype[closest2] = 2;
        vii right2;
        int nbound = bound;
        forin(i, right) {
            if (i != closest2) {
                if (dist(bound, i) == dist(bound, closest2) + dist(closest2, i)) {
                    location[i] = location[closest2] - dist(closest2, i);
                    stype[i] = 1;
                    if (location[i] > location[nbound]) {
                        nbound = i;
                    }
                } else {
                    right2.emplace_back(i);
                }
            }
        }
        right = move(right2);
        bound = nbound;
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...