oj.uz

Submission #257150

# Submission time Handle Problem Language Result Execution time Memory
257150 2020-08-03T16:27:56 Z eohomegrownapps Towns (IOI15_towns) C++14
25 / 100
 29 ms 1024 KB 
#include "towns.h"
#include <bits/stdc++.h>
using namespace std;

vector<vector<int>> distv;
int n;

int gd;

int dist(int a, int b){
    if (a==b){return 0;}
    if (a>b){
        swap(a,b);
    }
    if (distv[a][b]!=-1){
        return distv[a][b];
    } else {
        gd++;
        assert(gd<=(n*(n-1))/2);
        return distv[a][b]=getDistance(a,b);
    }
}

int hubDistance(int N, int sub) {
    gd=0;
    n=N;
    distv.assign(n,vector<int>(n,-1));
	
    int adist = -1;
    int aind = -1;
    for (int i = 0; i<n; i++){
        int ds = dist(0,i);
        if (adist<ds){
            adist=ds;
            aind=i;
        }
    }


    int bdist = -1;
    int bind = -1;
    for (int i = 0; i<n; i++){
        int ds = dist(aind,i);
        if (bdist<ds){
            bdist=ds;
            bind=i;
        }
    }

    //from adist to aind
    //vector<int> diameter;
    int hubdist = 1e9;
    int hubind = -1;
    for (int i = 0; i<n; i++){
        if (i==aind||i==bind){continue;}
        int distfroma = dist(aind,i)-(dist(aind,i)+dist(bind,i)-dist(aind,bind))/2;
        int rv = max(distfroma,dist(aind,bind)-distfroma);
        if (rv<hubdist){
            hubdist=rv;
            hubind=i;
        }
    }
    if (sub<=2){return hubdist;}
	int hubfroma = dist(aind,hubind)-(dist(aind,hubind)+dist(bind,hubind)-dist(aind,bind))/2;;
    vector<int> nodesinhub;
    int befcnt=0;
    int aftcnt=0;
    for (int i = 0; i<n; i++){
        int distfroma = dist(aind,i)-(dist(aind,i)+dist(bind,i)-dist(aind,bind))/2;
        if (distfroma<hubfroma){
            befcnt++;
        } else if (distfroma==hubfroma){
            nodesinhub.push_back(i);
        } else {
            aftcnt++;
        }
    }
    //cout<<befcnt<<' '<<aftcnt<<' '<<nodesinhub.size()<<'\n';
    if (befcnt>n/2||aftcnt>n/2){return -hubdist;}
    vector<bool> visited(n,false);
    int nodesleft = nodesinhub.size();
    for (int i : nodesinhub){
        if (visited[i]){continue;}
        if (nodesleft<=n/2){break;}
        int nodesingrp = 1;
        visited[i]=true;
        for (int j : nodesinhub){
            if (!visited[j]){
                if (dist(i,j)<dist(aind,i)+dist(aind,j)-2*hubfroma){
                    nodesingrp++;
                    visited[j]=true;
                }
            }
        }
        //cout<<nodesingrp<<'\n';
        if (nodesingrp>n/2){return -hubdist;}
        nodesleft-=nodesingrp;
    }
    return hubdist;
}

Compilation message

towns.cpp: In function 'int hubDistance(int, int)':
towns.cpp:81:36: warning: conversion to 'int' from 'std::vector<int>::size_type {aka long unsigned int}' may alter its value [-Wconversion]
     int nodesleft = nodesinhub.size();
                     ~~~~~~~~~~~~~~~^~

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 28 ms 1016 KB Output is correct
2 Correct 19 ms 896 KB Output is correct
3 Correct 1 ms 256 KB Output is correct
4 Correct 26 ms 896 KB Output is correct
5 Correct 24 ms 896 KB Output is correct

Subtask #2
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 29 ms 1024 KB Output is correct
2 Correct 17 ms 896 KB Output is correct
3 Correct 24 ms 896 KB Output is correct
4 Correct 24 ms 896 KB Output is correct

Subtask #3
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 18 ms 896 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 13 ms 512 KB Output isn't correct

Subtask #5
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 13 ms 640 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 39.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 384 KB Output isn't correct
2 Halted 0 ms 0 KB -