Submission #745641

#	Time	Username	Problem	Language	Result	Execution time	Memory
745641		Username4132	Rail (IOI14_rail)	C++14	30 / 100	299 ms	98848 KiB

rail

#include "rail.h"
#include<iostream>
#include<vector>
using namespace std;
using pii = pair<int, int>;
#define forn(i, n) for(int i=0; i<(int)n; ++i)
#define forsn(i, s, n) for(int i=s; i<(int)n; ++i)
#define PB push_back
#define F first
#define S second

const int MAXN=5010, INF=1000000010;
int *rel, *res;
int n, dis[MAXN][MAXN];
bool vis[MAXN];
vector<pii> g[MAXN];

void dfs(int v){
    vis[v]=true;
    for(pii to:g[v]) if(!vis[to.F]){
        if(res[v]==1) rel[to.F]=rel[v]+to.S;
        else rel[to.F]=rel[v]-to.S;
        res[to.F] = res[v]==1? 2 : 1;
        dfs(to.F);
    }
}

void findLocation(int N, int first, int location[], int stype[])
{
    n=N;
    forn(i, n) forsn(j, i, n) dis[i][j]=dis[j][i]=getDistance(i, j);
    forn(i, n){
        int mn=INF, ind=-1;
        forn(j, n) if(j!=i && mn>dis[i][j]) mn=dis[i][j], ind=j;
        g[i].PB({ind, mn}), g[ind].PB({i, mn});
    }
    rel=location, res=stype;
    rel[0]=first, res[0]=1;
    dfs(0);
}

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