Submission #775480

#	Time	Username	Problem	Language	Result	Execution time	Memory
775480		Charizard2021	Jakarta Skyscrapers (APIO15_skyscraper)	C++17	36 / 100	353 ms	262144 KiB

skyscraper

#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false);; cin.tie(NULL)
void selfMax(int& a, int b){
    a = max(a, b);
}
void selfMin(int& a, int b){
    a = min(a, b);
}
template<typename T> istream& operator>>(istream& in, vector<T>& a) {for(auto &x : a) in >> x; return in;};
template<typename T1, typename T2> istream& operator>>(istream& in, pair<T1, T2>& x) {return in >> x.first >> x.second;}

template<typename T1, typename T2> ostream& operator<<(ostream& out, const pair<T1, T2>& x) {return out << x.first << ' ' << x.second;}
template<typename T> ostream& operator<<(ostream& out, vector<T>& a) {for(auto &x : a) out << x << ' '; return out;};
template<typename T> ostream& operator<<(ostream& out, vector<vector<T> >& a) {for(auto &x : a) out << x << '\n'; return out;};
template<typename T1, typename T2> ostream& operator<<(ostream& out, vector<pair<T1, T2> >& a) {for(auto &x : a) out << x << '\n'; return out;};

/*

Use DSU dsu(N);

*/
struct DSU {
	vector<int> e;
	DSU(int N) { e = vector<int>(N, -1); }

	// get representive component (uses path compression)
	int get(int x) { return e[x] < 0 ? x : e[x] = get(e[x]); }

	bool same_set(int a, int b) { return get(a) == get(b); }

	int size(int x) { return -e[get(x)]; }

	bool unite(int x, int y) {  // union by size
		x = get(x), y = get(y);
		if (x == y) return false;
		if (e[x] > e[y]) swap(x, y);
		e[x] += e[y]; e[y] = x;
		return true;
	}
};
/*

Run with Bit<int> BIT

*/
template <class T> class BIT {
    private:
        int size;
        vector<T> bit;
        vector<T> arr;
    public:
        BIT(int size) : size(size), bit(size + 1), arr(size) {}
        void set(int ind, int val) { add(ind, val - arr[ind]); }
        void add(int ind, int val) {
            arr[ind] += val;
            ind++;
            for (; ind <= size; ind += ind & -ind) { bit[ind] += val; }
        }
        T pref_sum(int ind) {
            ind++;
            T total = 0;
            for (; ind > 0; ind -= ind & -ind) { total += bit[ind]; }
            return total;
        }
};
/*

Change Comb for specific Seg tree probs, but run with Seg<int> Seg;

*/
template<class T> struct Seg {
	const T ID = 1e18; T comb(T a, T b) { return min(a,b); }
	int n; vector<T> seg;
	void init(int _n) { n = _n; seg.assign(2*n,ID); }
	void pull(int p) { seg[p] = comb(seg[2*p],seg[2*p+1]); }
	void upd(int p, T val) {
		seg[p += n] = val; for (p /= 2; p; p /= 2) pull(p); }
	T query(int l, int r) {
		T ra = ID, rb = ID;
		for (l += n, r += n+1; l < r; l /= 2, r /= 2) {
			if (l&1) ra = comb(ra,seg[l++]);
			if (r&1) rb = comb(seg[--r],rb);
		}
		return comb(ra,rb);
	}
};
int main(){
    int n, m;
    cin >> n >> m;
    vector<pair<int, int> > dogs(m);
    for(int i = 0; i < m; i++){
        cin >> dogs[i];
    }
    map<pair<int, int>, int> mp;
    for(int i = 0; i < m; i++){
        mp[dogs[i]]++;
    }
    vector<pair<int,int> > adj[n];
    for(int i = 0; i < m; i++){
        if(0 <= dogs[i].first + dogs[i].second && dogs[i].first + dogs[i].second < n){
            adj[dogs[i].first].push_back(make_pair(dogs[i].second + dogs[i].first, 1));
            int k = 2;
            while(0 <= dogs[i].first + dogs[i].second * k && dogs[i].first + dogs[i].second * k < n && mp.count(make_pair(dogs[i].first + dogs[i].second * k, dogs[i].second)) == 0){
                adj[dogs[i].first].push_back(make_pair(dogs[i].first + dogs[i].second * k, k));
                k++;
            }
            if(0 <= dogs[i].first + dogs[i].second * k && dogs[i].first + dogs[i].second * k < n){
                adj[dogs[i].first].push_back(make_pair(dogs[i].first + dogs[i].second * k, k));
            }
        }
        if(0 <= dogs[i].first - dogs[i].second && dogs[i].first - dogs[i].second < n){
            adj[dogs[i].first].push_back(make_pair(dogs[i].first - dogs[i].second, 1));
            int k = 2;
            while(0 <= dogs[i].first - dogs[i].second * k && dogs[i].first - dogs[i].second * k < n && mp.count(make_pair(dogs[i].first - dogs[i].second * k, dogs[i].second)) == 0){
                adj[dogs[i].first].push_back(make_pair(dogs[i].first - dogs[i].second * k, k));
                k++;
            }
            if(0 <= dogs[i].first - dogs[i].second * k && dogs[i].first - dogs[i].second * k < n){
                adj[dogs[i].first].push_back(make_pair(dogs[i].first - dogs[i].second * k, k));
            }
        }
    }
    priority_queue<pair<int, int> > pq;
    vector<long long> distances(n, 1e18);
    if(!(0 <= dogs[0].first && dogs[0].first < n)){
        cout << -1 << "\n";
        return 0;
    }
    if(!(0 <= dogs[1].first && dogs[1].first < n)){
        cout << -1 << "\n";
        return 0;
    }
    distances[dogs[0].first] = 0;
    pq.push(make_pair(0LL, dogs[0].first));
    while(!pq.empty()){
        pair<long long, int> x2 = pq.top();
        pq.pop();
        if(distances[x2.second] != -x2.first){
            continue;
        }
        int u = x2.second;
        for(pair<int, int> x : adj[u]){
            int v = x.first;
            if(distances[v] > distances[u] + (long long)x.second){
                distances[v] = distances[u] + (long long)x.second;
                pq.push(make_pair(-distances[v], v));
            }
        }
    }
    if(distances[dogs[1].first] == (long long)1e18){
        cout << -1 << "\n";
    }
    else{
        cout << distances[dogs[1].first] << "\n";
    }
}

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