Submission #610805

#	Time	Username	Problem	Language	Result	Execution time	Memory
610805		NicolaAbusaad2014	Jakarta Skyscrapers (APIO15_skyscraper)	C++14	100 / 100	706 ms	56408 KiB

skyscraper

/**
 * Prof.Nicola
**/
#include <bits/stdc++.h>

using namespace std;
#define endl "\n"
#define mp make_pair
#define F first
#define S second
#define REP(i,l,r) for(int i=(l);i<(r);i++)
#define PER(i,l,r) for(int i=(r)-1;i>=(l);i--)
#define bitl __builtin_clz
#define bitr __builtin_ctz
#define bits __builtin_popcount
const long dx[4]={1,0,-1,0},dz[4]={0,1,0,-1};
const long double pi=3.14159265359;
const long long mod=1e9+7;
long long p(long long x){while(x&(x-1)){x=x&(x-1);}return x;}
long long squared(long long x){return (x*x)%mod;}
long long power(long long x,long long p){if(p==0){return 1;}if(p%2==1){return (power(x,p-1)*x)%mod;}return squared(power(x,p/2));}
long long inv(long long x){return power(x,mod-2);}
long long log(long long x,long long z){if(x<z){return 0;}return 1+log(x/z,z);}
template<class T>void re(T&x){cin>>x;}
template<class T1,class T2> void re(pair<T1,T2>&x){re(x.first);re(x.second);}
template<class T>void re(vector<T>&x){for(long i=0;i<x.size();i++){re(x[i]);}}
template<class T>void re(deque<T>&x){for(long i=0;i<x.size();i++){re(x[i]);}}
template<class T>void out(T x){cout<<x<<" ";}
template<class T1,class T2>void out(pair<T1,T2>x){out(x.first);out(x.second);cout<<endl;}
template<class T>void out(vector<T>x,long l=0,long r=0){if(!r){r=x.size();}for(long i=l;i<r;i++){out(x[i]);}cout<<endl;}
template<class T>void out(deque<T>x,long l=0,long r=0){if(!r){r=x.size();}for(long i=l;i<r;i++){out(x[i]);}cout<<endl;}
template<class T>void out(set<T>x){while(!x.empty()){out(*x.begin());x.erase(*x.begin());}cout<<endl;}
template<class T1,class T2>void out(map<T1,T2>x){while(!x.empty()){out(*x.begin());x.erase(x.begin()->first);}cout<<endl;}
template<class T>void out(queue<T>x){while(!x.empty()){out(x.front());x.pop();}cout<<endl;}
template<class T>void out(priority_queue<T>x){while(!x.empty()){out(x.top());x.pop();}cout<<endl;}
template<class T>void out(stack<T>x){while(!x.empty()){out(x.top());x.pop();}cout<<endl;}
template<class T>T cross(complex<T>x,complex<T>z){return (conj(x)*z).imag();}
template<class T>T dot(complex<T>x,complex<T>z){return (conj(x)*z).real();}
set<long long>::iterator T;
long long vLE(long long x,vector<long long>&VT,long l=0,long r=-1){if(r==-1){r=VT.size()-1;}if(VT[l]>x){return -1;}long long z=p(r-l);while(z){if(l+z<=r&&VT[l+z]<=x){l+=z;}z/=2;}return l;}
long long vL(long long x,vector<long long>&VT,long l=0,long r=-1){if(r==-1){r=VT.size()-1;}if(VT[l]>=x){return -1;}long long z=p(r-l);while(z){if(l+z<=r&&VT[l+z]<x){l+=z;}z/=2;}return l;}
long long vGE(long long x,vector<long long>&VT,long l=0,long r=-1){if(r==-1){r=VT.size()-1;}if(VT[l]>=x){return l;}l=vL(x,VT,l,r);if(l==r){return -1;}return l+1;}
long long vG(long long x,vector<long long>&VT,long l=0,long r=-1){if(r==-1){r=VT.size()-1;}if(VT[l]>x){return l;}l=vLE(x,VT,l,r);if(l==r){return -1;}return l+1;}
long long sLE(long long x,set<long long>&ST){if(ST.count(x)){return x;}if(*ST.begin()>x){return -1;}ST.insert(x);T=ST.find(x);T--;ST.erase(x);return *T;}
long long sL(long long x,set<long long>&ST){if(*ST.begin()>=x){return -1;}if(ST.count(x)){T=ST.find(x);T--;return *T;}ST.insert(x);T=ST.find(x);T--;ST.erase(x);return *T;}
long long sGE(long long x,set<long long>&ST){if(ST.count(x)){return x;}if(*ST.rbegin()<x){return -1;}ST.insert(x);T=ST.find(x);T++;ST.erase(x);return *T;}
long long sG(long long x,set<long long>&ST){if(*ST.rbegin()<=x){return -1;}if(ST.count(x)){T=ST.find(x);T++;return *T;}ST.insert(x);T=ST.find(x);T++;ST.erase(x);return *T;}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int n,m,s,cost,pos,jump,ans,inf=1e9;
    cin>>n>>m;
    s=sqrt(n)+1;
    vector<pair<int,int> >v(m);
    vector<bool>ok(n);
    re(v);
    vector<vector<int> >possible(n),cnt(n),l(n),r(n);
    vector<vector<bool> >vis(n);
    vector<int>path(n,inf);
    REP(i,0,m){
        if(v[i].S>s){
            for(int j=v[i].F%v[i].S;j<n;j+=v[i].S){
                cnt[j].push_back(v[i].S);
                if(j+v[i].S<n){
                    r[j].push_back(cnt[j+v[i].S].size());
                }
                else{
                    r[j].push_back(0);
                }
                if(j-v[i].S>=0){
                    l[j].push_back(cnt[j-v[i].S].size()-1);
                }
                else{
                    l[j].push_back(0);
                }
            }
            possible[v[i].F].push_back(cnt[v[i].F].size()+s);
            continue;
        }
        possible[v[i].F].push_back(v[i].S);
    }
    REP(i,0,n){
        vis[i].resize(s+1+cnt[i].size());
    }
    deque<pair<int,pair<int,int> > >pq;
    pq.push_back(mp(0,mp(v[0].F,possible[v[0].F][0])));
    while(!pq.empty()){
        cost=pq.front().F;
        pos=pq.front().S.F;
        jump=pq.front().S.S;
        pq.pop_front();
        if(vis[pos][jump]){
            continue;
        }
        vis[pos][jump]=true;
        if(ok[pos]){
            if(jump<=s){
                if(pos+jump<n){
                    pq.push_back(mp(cost+1,mp(pos+jump,jump)));
                }
                if(pos-jump>=0){
                    pq.push_back(mp(cost+1,mp(pos-jump,jump)));
                }
                continue;
            }
            if(pos+cnt[pos][jump-s-1]<n){
                pq.push_back(mp(cost+1,mp(pos+cnt[pos][jump-s-1],r[pos][jump-s-1]+s+1)));
            }
            if(pos-cnt[pos][jump-s-1]>=0){
                pq.push_back(mp(cost+1,mp(pos-cnt[pos][jump-s-1],l[pos][jump-s-1]+s+1)));
            }
            continue;
        }
        path[pos]=cost;
        if(jump<=s){
            if(pos+jump<n){
                pq.push_back(mp(cost+1,mp(pos+jump,jump)));
            }
            if(pos-jump>=0){
                pq.push_back(mp(cost+1,mp(pos-jump,jump)));
            }
        }
        else{
            if(pos+cnt[pos][jump-s-1]<n){
                pq.push_back(mp(cost+1,mp(pos+cnt[pos][jump-s-1],r[pos][jump-s-1]+s+1)));
            }
            if(pos-cnt[pos][jump-s-1]>=0){
                pq.push_back(mp(cost+1,mp(pos-cnt[pos][jump-s-1],l[pos][jump-s-1]+s+1)));
            }
        }
        ok[pos]=true;
        REP(i,0,possible[pos].size()){
            pq.push_front(mp(cost,mp(pos,possible[pos][i])));
        }
    }
    ans=path[v[1].F];
    if(ans==inf){
        cout<<-1<<endl;
        return 0;
    }
    cout<<ans<<endl;
    return 0;
}

Compilation message (stderr)

skyscraper.cpp: In function 'int main()':
skyscraper.cpp:11:35: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   11 | #define REP(i,l,r) for(int i=(l);i<(r);i++)
      |                                   ^
skyscraper.cpp:133:9: note: in expansion of macro 'REP'
  133 |         REP(i,0,possible[pos].size()){
      |         ^~~
skyscraper.cpp: In instantiation of 'void re(std::vector<_Tp>&) [with T = std::pair<int, int>]':
skyscraper.cpp:57:9:   required from here
skyscraper.cpp:26:53: warning: comparison of integer expressions of different signedness: 'long int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   26 | template<class T>void re(vector<T>&x){for(long i=0;i<x.size();i++){re(x[i]);}}
      |                                                    ~^~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 12 / 12
• Subtask 3: 14 / 14
• Subtask 4: 21 / 21
• Subtask 5: 43 / 43