Submission #425434

#	Time	Username	Problem	Language	Result	Execution time	Memory
425434		ngrace	Shortcut (IOI16_shortcut)	C++14	0 / 100	2068 ms	1064 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 diameter(){
    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];
                }
            }
        }
    }
    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 out=diameter();
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                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;
                }
                //create edge between
                adj[i][j] = c;
                adj[j][i] = c;

                long long diam=diameter();
                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:77:1: warning: control reaches end of non-void function [-Wreturn-type]
   77 | }
      | ^

• Subtask 1: 0 / 9
• Subtask 2: 0 / 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