Submission #891833

# Submission time Handle Problem Language Result Execution time Memory
891833 2023-12-24T07:33:37 Z abcvuitunggio Towns (IOI15_towns) C++17
61 / 100
16 ms 1116 KB
#include "towns.h"
#include <bits/stdc++.h>
using namespace std;
int n,dist[111][111],cnt,u,v,R,sz[2],lim;
mt19937 rd(0);
int d(int u, int v){
    if (u==v)
        return 0;
    if (u>v)
        swap(u,v);
    if (dist[u][v])
        return dist[u][v];
    cnt++;
    return dist[u][v]=getDistance(u,v);
}
int calc(vector <int> ve){
    vector <int> l,r;
    int j=ve.back();
    l.push_back(j);
    ve.pop_back();
    int pos=0;
    for (int i:ve){
        if (max(l.size(),r.size())+ve.size()-pos+1<=n/2)
            return R;
        if (cnt==lim)
            return (max(l.size(),r.size())+(ve.size()-pos+2)/2>n/2?-R:R);
        pos++;
        int x=(d(i,u)+d(i,v)-d(u,v))/2,y=(d(j,u)+d(j,v)-d(u,v))/2;
        if (d(i,j)<x+y){
            l.push_back(i);
            if (l.size()>n/2)
                return -R;
        }
        else
            r.push_back(i);
    }
    return calc(r);
}
int hubDistance(int N, int sub){
    cnt=0,n=N,lim=(sub&1?n*(sub==5?10:n-1):n*7+1)/2;
    memset(dist,0,sizeof(dist));
    pair <int, int> mx={0,-1};
    for (int i=0;i<N;i++)
        mx=max(mx,{d(0,i),i});
    u=mx.second;
    mx={0,-1};
    for (int i=0;i<N;i++)
        mx=max(mx,{d(u,i),i});
    v=mx.second,R=1e9,sz[0]=sz[1]=0;
	for (int i=0;i<N;i++)
        R=min(R,abs(d(i,u)-d(i,v)));
    vector <int> a,b,tmp;
    for (int i=0;i<N;i++){
        int val=d(i,u)-d(i,v);
        if (val<-R)
            sz[0]++;
        if (val==-R)
            a.push_back(i);
        if (val==R)
            b.push_back(i);
        if (val>R)
            sz[1]++;
    }
    if (!R)
        b.clear();
    R=(R+d(u,v))/2;
    if (sub<3)
        return R;
    if (max(sz[0],sz[1])>N/2)
        return -R;
    if (min(sz[1]+b.size(),sz[0]+a.size())>N/2)
        return -R;
    if (max(a.size(),sz[1]+b.size())<=N/2)
        return R;
    if (max(b.size(),sz[0]+a.size())<=N/2)
        return R;
    return calc(b.size()+sz[1]<=N/2?a:b);
}

Compilation message

towns.cpp: In function 'int d(int, int)':
towns.cpp:6:18: warning: declaration of 'v' shadows a global declaration [-Wshadow]
    6 | int d(int u, int v){
      |              ~~~~^
towns.cpp:4:28: note: shadowed declaration is here
    4 | int n,dist[111][111],cnt,u,v,R,sz[2],lim;
      |                            ^
towns.cpp:6:11: warning: declaration of 'u' shadows a global declaration [-Wshadow]
    6 | int d(int u, int v){
      |       ~~~~^
towns.cpp:4:26: note: shadowed declaration is here
    4 | int n,dist[111][111],cnt,u,v,R,sz[2],lim;
      |                          ^
towns.cpp: In function 'int calc(std::vector<int>)':
towns.cpp:23:51: warning: comparison of integer expressions of different signedness: 'long unsigned int' and 'int' [-Wsign-compare]
   23 |         if (max(l.size(),r.size())+ve.size()-pos+1<=n/2)
      |             ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~
towns.cpp:26:63: warning: comparison of integer expressions of different signedness: 'long unsigned int' and 'int' [-Wsign-compare]
   26 |             return (max(l.size(),r.size())+(ve.size()-pos+2)/2>n/2?-R:R);
      |                     ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~
towns.cpp:31:25: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   31 |             if (l.size()>n/2)
      |                 ~~~~~~~~^~~~
towns.cpp: In function 'int hubDistance(int, int)':
towns.cpp:71:43: warning: comparison of integer expressions of different signedness: 'const long unsigned int' and 'int' [-Wsign-compare]
   71 |     if (min(sz[1]+b.size(),sz[0]+a.size())>N/2)
      |         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~
towns.cpp:73:37: warning: comparison of integer expressions of different signedness: 'const long unsigned int' and 'int' [-Wsign-compare]
   73 |     if (max(a.size(),sz[1]+b.size())<=N/2)
      |         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~
towns.cpp:75:37: warning: comparison of integer expressions of different signedness: 'const long unsigned int' and 'int' [-Wsign-compare]
   75 |     if (max(b.size(),sz[0]+a.size())<=N/2)
      |         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~
towns.cpp:77:31: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   77 |     return calc(b.size()+sz[1]<=N/2?a:b);
      |                 ~~~~~~~~~~~~~~^~~~~
# Verdict Execution time Memory Grader output
1 Correct 10 ms 1112 KB Output is correct
2 Correct 9 ms 860 KB Output is correct
3 Correct 0 ms 440 KB Output is correct
4 Correct 16 ms 1072 KB Output is correct
5 Correct 12 ms 860 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 10 ms 1112 KB Output is correct
2 Correct 9 ms 860 KB Output is correct
3 Correct 16 ms 860 KB Output is correct
4 Correct 12 ms 856 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 9 ms 1112 KB Output is correct
2 Correct 14 ms 856 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 13 ms 968 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 12 ms 1116 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 9 ms 348 KB Output is correct
2 Correct 12 ms 860 KB Output is correct
3 Correct 16 ms 856 KB Output is correct
# Verdict Execution time Memory Grader output
1 Incorrect 12 ms 348 KB Output isn't correct
2 Halted 0 ms 0 KB -