Codeforces Round #573 (Div. 1) A. Tokitsukaze and Discard Items

• Time : O(m)
• Space : O(m)

c++

#include <bits/stdc++.h>

#pragma optimization_level 3
#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math,O3")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx")
#pragma GCC optimize("Ofast")//Comment optimisations for interactive problems (use endl)
#pragma GCC target("avx,avx2,fma")
#pragma GCC optimization ("unroll-loops")

using namespace std;

struct PairHash {
    inline std::size_t operator()(const std::pair<int, int> &v) const { return v.first * 31 + v.second; }
};

// speed
#define Code ios_base::sync_with_stdio(false);
#define By ios::sync_with_stdio(0);
#define Sumfi cout.tie(NULL);

// alias
using ll = long long;
using ld = long double;

// constants
const ld PI = 3.14159265358979323846;  /* pi */
const ll INF = 1e18;
const ld EPS = 1e-9;
const ll MAX_N = 101010;
const ll mod = 998244353;

// typedef
typedef pair<ll, ll> pll;
typedef vector<pll> vpll;
typedef array<ll,3> all3;
typedef array<ll,5> all5;
typedef vector<all3> vall3;
typedef vector<all5> vall5;
typedef vector<ll> vll;
typedef vector<vll> vvll;
typedef vector<int> vi;
typedef vector<string> vs;
typedef vector<vs> vvs;
typedef unordered_set<ll> usll;
typedef unordered_set<pll, PairHash> uspll;
typedef unordered_map<ll, ll> umll;

// macros
#define rep(i,n) for(ll i=0;i<n;i++)
#define rrep(i,m,n) for(ll i=n;i>=m;i--)
#define all(a) begin(a), end(a)
#define rall(a) rbegin(a), rend(a)

ll __gcd(ll x, ll y) { return !y ? x : __gcd(y, x % y); }
ll __lcm(ll x, ll y) { return x * y / __gcd(x,y); }

using namespace std;

ll solve(ll n, vll& A, ll p) {
    ll res = 0, now = 1;
    queue<ll> Q;
    rep(i, A.size()) Q.push(A[i]);
    while(!Q.empty()) {
        ll gap = 0;
        while(!Q.empty() and Q.front() < now + gap + p) {
            ll until = now + gap + p;
            while(!Q.empty() and Q.front() < until) {
                Q.pop();
                gap++;
            }
            res++;
        }
        if(!Q.empty()) {
            now = now + gap + p;
            now += (Q.front() - now) / p * p;
        }
    }

    return res;
}

int main() {
    Code By Sumfi

    //cout.precision(12);

    ll tc = 1;
    //cin>>tc;
    for (ll i = 1; i <= tc; i++) {
        ll n,m,p;
        cin>>n>>m>>p;
        vll A(m);
        rep(i, m) cin>>A[i];
        cout<<solve(n,A,p)<<endl;
    }
    return 0;
}