Submission #965534

#	Time	Username	Problem	Language	Result	Execution time	Memory
965534		MarwenElarbi	Bitaro's travel (JOI23_travel)	C++17	100 / 100	607 ms	7368 KiB

travel

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#define vi vector<int>
#define ve vector
#define ll long long
#define vl vector<ll>
#define vll vector<pair<ll,ll>>
#define onbit __builtin_popcount
#define ii pair<int,int>
#define vvi vector<vi>
#define vii vector<ii>
#define gii greater<ii>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define INF 1e18
#define eps 1e-7
#define eps1 1e-2
#define optimise ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
#define MAX_A 1e5+5
using namespace std;
using namespace __gnu_pbds;
template <class T>
using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const ll MOD = (1LL << 61) - 1;
const int nax = 1e5+5;
const int MAX_VAL = 1e6+1;
double PI=3.14159265359;
int arx[8]={1,0,0,-1,-1,-1, 1, 1};
int ary[8]={0,1,-1, 0, 1,-1,-1, 1};
void setIO(string s) {
    freopen((s + ".in").c_str(), "r", stdin);
    freopen((s + ".out").c_str(), "w", stdout);
}
vector<int> tab(nax);
int main(){
    optimise;
    //setIO("dec");
    int n;
    cin>>n;
    tab.resize(n);
    for (int i = 0; i < n; ++i)
    {
        cin>>tab[i];
    }
    int q;
    cin>>q;
    while(q--){
        int s;
        cin>>s;
        int l=lower_bound(tab.begin(),tab.end(),s)-tab.begin();
        int r=l;
        //cout<<tab[l]<<endl;
        long long ans=0;
        while(true){
            //cout <<l<<" "<<r<<" "<<ans<<endl;
            if(l==0){
                ans+=tab[n-1]-s;
                break;
            }else if(r==n){
                ans+=s-tab[0];
                break;
            }
            if(s-tab[l-1]<=tab[r]-s){
                //cout << "nabba"<<endl;
                int i=lower_bound(tab.begin(),tab.end(),2*s-tab[r])-tab.begin();
                //cout <<tab[r]<<endl;
                ans+=s-tab[i];
                l=i;
                s=tab[i];
            }else{
                int i=lower_bound(tab.begin(),tab.end(),2*s-tab[l-1])-tab.begin()-1;
                //cout <<"nabba"<<" "<<2*s-tab[l-1]<<endl;
                ans+=tab[i]-s;
                r=i+1;
                s=tab[i];
            }
        }
        cout <<ans<<endl;
    }   
}

Compilation message (stderr)

travel.cpp: In function 'void setIO(std::string)':
travel.cpp:36:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   36 |     freopen((s + ".in").c_str(), "r", stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
travel.cpp:37:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   37 |     freopen((s + ".out").c_str(), "w", stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 10 / 10
• Subtask 3: 30 / 30
• Subtask 4: 55 / 55