提出 #73071899

---

ソースコード 拡げる

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
#define fastio cin.tie(0)->sync_with_stdio(0)
#define all(x) (x).begin(),(x).end()
#define rall(x) (x).rbegin(),(x).rend()
#define compress(v) sort(all(v)), v.erase(unique(all(v)), v.end())
#define sz(x) (int)(x).size()
using namespace std;
typedef long long ll;

int main() {
    fastio; int N, D; cin >> N >> D;
    vector<int> v(N);
    for (auto& i : v) cin >> i;
    ll ans = 0;
    int L = 0;
    set<int> s;
    for (int R = 0; R < N; R++) {
        while (1) {
            auto it = s.lower_bound(v[R]);
            bool chk = false;
            if (it != s.end() && *it < v[R]+D) chk = true;
            if (it != s.begin() && *prev(it) > v[R]-D) chk = true;
            if (!chk) break;
            s.erase(v[L]); L++;
        }
        s.insert(v[R]);
        ans += R-L+1;
    }
    cout << ans << "\n";
    return 0;
}

提出情報

ジャッジ結果