Submission #730295

#	Time	Username	Problem	Language	Result	Execution time	Memory
730295		ogibogi2004	Rail (IOI14_rail)	C++14	100 / 100	90 ms	772 KiB

rail

#include "rail.h"
#include<bits/stdc++.h>
using namespace std;

void findLocation(int N, int first, int location[], int stype[])
{
    memset(stype,0,sizeof(stype));
    memset(location,0,sizeof(location));
    location[0]=first;
    stype[0]=1;
    map<int,int> mtype;
    mtype[location[0]]=1;
    vector<pair<int,int> >distances;
    vector<int>dist0(N);
    vector<int>dist1(N);
    for(int i=1;i<N;i++)
    {
        distances.push_back({getDistance(0, i),i});
        dist0[i]=distances.back().first;
    }
    sort(distances.begin(),distances.end());
    location[distances[0].second]=location[0]+distances[0].first;
    stype[distances[0].second]=2;
    mtype[location[distances[0].second]]=2;
    set<int>cs;
    set<int>ds;
    int minc=0,maxd=distances[0].second,sec=maxd;
    cs.insert(location[0]);
    ds.insert(location[distances[0].second]);
    for(int i=1;i<N;i++)
    {
        if(i==maxd)continue;
        dist1[i]=getDistance(maxd,i);
    }
    for(int i=1;i<distances.size();i++)
    {
        int j=distances[i].second;
        if(dist0[j]==dist1[j]+dist0[sec])
        {
            //left of sec
            if(dist1[j]<dist0[sec])
            {
                //between 0 and sec
                location[j]=location[sec]-dist1[j];
                stype[j]=1;
                mtype[location[j]]=1;
            }
            else
            {
                //left of 0
                //can be C or D
                //if D, there should be C before it
                //C X 0 D
                //calc dist(minc,X) => if dist(minc,X)+location[sec]-location[minc]==dist1[minc], ok, it is D
                int t=getDistance(minc,j);
                int pos1=(t+location[sec]-dist1[j]+location[minc])/2;
                if(mtype[pos1]!=1)
                {
                    stype[j]=1;
                    location[j]=location[sec]-dist1[j];
                    minc=j;
                    mtype[location[j]]=1;
                }
                else
                {
                    stype[j]=2;
                    location[j]=location[minc]+t;
                    mtype[location[j]]=2;
                }
            }
        }
        else
        {
            //cout<<j<<" ?2\n";
            //right of sec
            //0 D X D
            //if X=C, then getDist(maxd,X)+location[maxd]-location[0]=dist0[X]
            int t=getDistance(j, maxd);
            int pos1=(-t+location[0]+dist0[j]+location[maxd])/2;
            //7-
            //cout<<pos1<<" "<<(-t+location[0]+dist0[j]+location[maxd])<<endl;
            if(mtype[pos1]!=2)
            {
                stype[j]=2;
                location[j]=location[0]+dist0[j];
                maxd=j;
                mtype[location[j]]=2;
            }
            else
            {
                stype[j]=1;
                location[j]=location[maxd]-t;
                mtype[location[j]]=1;
            }
        }
    }
    /*for(int i=0;i<N;i++)
    {
        cout<<i<<": "<<location[i]<<" "<<stype[i]<<endl;
    }*/
}

Compilation message (stderr)

rail.cpp: In function 'void findLocation(int, int, int*, int*)':
rail.cpp:7:27: warning: 'sizeof' on array function parameter 'stype' will return size of 'int*' [-Wsizeof-array-argument]
    7 |     memset(stype,0,sizeof(stype));
      |                          ~^~~~~~
rail.cpp:5:57: note: declared here
    5 | void findLocation(int N, int first, int location[], int stype[])
      |                                                     ~~~~^~~~~~~
rail.cpp:7:20: warning: argument to 'sizeof' in 'void* memset(void*, int, size_t)' call is the same expression as the destination; did you mean to dereference it? [-Wsizeof-pointer-memaccess]
    7 |     memset(stype,0,sizeof(stype));
      |                    ^~~~~~~~~~~~~
rail.cpp:8:30: warning: 'sizeof' on array function parameter 'location' will return size of 'int*' [-Wsizeof-array-argument]
    8 |     memset(location,0,sizeof(location));
      |                             ~^~~~~~~~~
rail.cpp:5:41: note: declared here
    5 | void findLocation(int N, int first, int location[], int stype[])
      |                                     ~~~~^~~~~~~~~~
rail.cpp:8:23: warning: argument to 'sizeof' in 'void* memset(void*, int, size_t)' call is the same expression as the destination; did you mean to dereference it? [-Wsizeof-pointer-memaccess]
    8 |     memset(location,0,sizeof(location));
      |                       ^~~~~~~~~~~~~~~~
rail.cpp:35:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   35 |     for(int i=1;i<distances.size();i++)
      |                 ~^~~~~~~~~~~~~~~~~

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