Submission #425445

#	Time	Username	Problem	Language	Result	Execution time	Memory
425445		ngrace	Shortcut (IOI16_shortcut)	C++14	23 / 100	2066 ms	2340 KiB

shortcut

#include "shortcut.h"
#include <vector>
#include <queue>
#include <utility>
#include <iostream>
#include <limits.h>
using namespace std;
#define v vector
#define pii pair<int,int>
#define fi first
#define se second

v<v<long long>> adj;
int N;

long long floyd(){
    for(int k=0;k<2*N;k++){
        for(int i=0;i<2*N;i++){
            for(int j=0;j<2*N;j++){
                if(adj[i][k]==LLONG_MAX || adj[k][j]==LLONG_MAX) continue;
                if(adj[i][j] > adj[i][k] + adj[k][j]){
                    adj[i][j] = adj[i][k] + adj[k][j];
                    adj[j][i] = adj[i][k] + adj[k][j];
                }
            }
        }
    }
    long long out=0;
    for(int i=0;i<2*N;i++){
        for(int j=0;j<2*N;j++){
            if(adj[i][j]>out && adj[i][j]!=LLONG_MAX) out=adj[i][j];
        }
    }
    return out;
}

long long find_shortcut(int n, std::vector<int> l, std::vector<int> d, int c)
{
    if(n<=500){
        N=n;
        adj=v<v<long long>> (2*n,v<long long>(2*N, LLONG_MAX));
        for(int i=0;i<n-1;i++){
            adj[i][i+1] = l[i];
            adj[i+1][i] = l[i];
            adj[i][i]=0;
        }
        for(int i=0;i<n;i++){
            adj[i][i+n] = d[i];
            adj[i+n][i] = d[i];
            adj[i+n][i+n]=0;
        }

        long long origDiam = floyd();
        long long out=origDiam;
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                long long diam=0;
                for(int a=n;a<2*n;a++){
                    for(int b=a+1;b<2*n;b++){
                        long long dist=adj[a][b];
                        long long dist2=adj[a][i]+adj[b][j]+c;
                        long long dist3=adj[a][j]+adj[b][i]+c;
                        if(dist2<dist) dist=dist2;
                        if(dist3<dist) dist=dist3;
                        if(dist>diam) diam=dist;
                        if(diam>out) break;
                    }
                    if(diam>out) break;
                }
                if(diam<out) out=diam;
            }
        }

        return out;
    }
}

Compilation message (stderr)

shortcut.cpp: In function 'long long int find_shortcut(int, std::vector<int>, std::vector<int>, int)':
shortcut.cpp:76:1: warning: control reaches end of non-void function [-Wreturn-type]
   76 | }
      | ^

• Subtask 1: 9 / 9
• Subtask 2: 14 / 14
• Subtask 3: 0 / 8
• Subtask 4: 0 / 7
• Subtask 5: 0 / 33
• Subtask 6: 0 / 22
• Subtask 7: 0 / 4
• Subtask 8: 0 / 3