#767972 (C++17) No.1965 Heavier

提出ソース

結果

問題	No.1965 Heavier
コンテスト
ユーザー	first_vil
提出日時	2022-06-03 22:00:48
言語	C++17 (gcc 13.3.0 + boost 1.87.0)
結果	AC
実行時間	195 ms / 2,000 ms
コード長	5,485 bytes
コンパイル時間	2,468 ms
コンパイル使用メモリ	204,940 KB
最終ジャッジ日時	2025-01-29 17:41:51
ジャッジサーバーID （参考情報）	judge2 / judge2

このコードへのチャレンジ
（要ログイン）

ファイルパターン	結果
sample	AC * 3
other	AC * 26

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
template<class T> using V = vector<T>;
using VI = V<int>;
using VL = V<ll>;
using VS = V<string>;
template<class T> using PQ = priority_queue<T, V<T>, greater<T>>;
using graph = V<VI>;
template<class T> using w_graph = V<V<pair<int, T>>>;
#define FOR(i,a,n) for(int i=(a);i<(n);++i)
#define eFOR(i,a,n) for(int i=(a);i<=(n);++i)
#define rFOR(i,a,n) for(int i=(n)-1;i>=(a);--i)
#define erFOR(i,a,n) for(int i=(n);i>=(a);--i)
#define all(a) a.begin(),a.end()
#define rall(a) a.rbegin(),a.rend()
#define inside(h,w,y,x) (unsigned(y)<h&&unsigned(x)<w)
#ifdef _DEBUG
#define line cout << "-----------------------------\n"
#define stop system("pause")
#endif
constexpr ll INF = 1000000000;
constexpr ll LLINF = 1LL << 61;
constexpr ll mod = 1000000007;
constexpr ll MOD = 998244353;
constexpr ld eps = 1e-10;
constexpr int dy[]{ -1,0,1,0 }, dx[]{ 0,1,0,-1 };
template<class T> inline bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; }return false; }
template<class T> inline bool chmin(T& a, const T& b) { if (a > b) { a = b; return true; }return false; }
inline void init() { cin.tie(nullptr); cout.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(15); }
template<class T> inline istream& operator>>(istream& is, V<T>& v) { for (auto& a : v)is >> a; return is; }
template<class T, class U> inline istream& operator>>(istream& is, pair<T, U>& p) { is >> p.first >> p.second; return is; }
template<class T> inline V<T> vec(size_t a) { return V<T>(a); }
template<class T> inline V<T> defvec(T def, size_t a) { return V<T>(a, def); }
template<class T, class... Ts> inline auto vec(size_t a, Ts... ts) { return V<decltype(vec<T>(ts...))>(a, vec<T>(ts...)); }
template<class T, class... Ts> inline auto defvec(T def, size_t a, Ts... ts) { return V<decltype(defvec<T>(def, ts...))>(a, defvec<T>(def, ts...)); }
template<class T> inline void print(const T& a) { cout << a << "\n"; }
template<class T, class... Ts> inline void print(const T& a, const Ts&... ts) { cout << a << " "; print(ts...); }
template<class T> inline void print(const V<T>& v) { for (int i = 0; i < v.size(); ++i)cout << v[i] << (i == v.size() - 1 ? "\n" : " "); }
template<class T> inline void print(const V<V<T>>& v) { for (auto& a : v)print(a); }
template<class T> inline constexpr const T cumsum(const V<T>& a, int l, int r) { return 0 <= l && l <= r && r < a.size() ? a[r] - (l == 0 ? 0 : a[l - 1]) : 0; }//[l,r]
template<class T> inline constexpr const T min(const V<T>& v) { return *min_element(all(v)); }
template<class T> inline constexpr const T max(const V<T>& v) { return *max_element(all(v)); }
template<class T> inline V<T>& operator++(V<T>& v) { for (T& a : v)++a; return v; }
template<class T> inline V<T>& operator--(V<T>& v) { for (T& a : v)--a; return v; }

template<class Op> class SegmentTree {
	using T = typename Op::T;
	int len, n;
	vector<T> dat;
public:
	SegmentTree(const int _n) : n(_n) {
		for (len = 1; len < n; len <<= 1);
		dat.resize(len << 1, Op::unit);
	}
	SegmentTree(const vector<T>& v) : n(v.size()) {
		for (len = 1; len < n; len <<= 1);
		dat.resize(len << 1, Op::unit);
		for (int i = 0; i < n; ++i)dat[i + len] = v[i];
		for (int i = len - 1; 0 < i; --i)
			dat[i] = Op::merge(dat[i << 1], dat[i << 1 | 1]);
	}
	inline void update(int idx, const T a) {
		idx += len;
		dat[idx] = Op::update(dat[idx], a);
		for (idx >>= 1; 0 < idx; idx >>= 1)
			dat[idx] = Op::merge(dat[idx << 1], dat[idx << 1 | 1]);
	}
	inline T value(int l, int r) {
		T vl = Op::unit, vr = Op::unit;
		for (l += len, r += len; l < r; l >>= 1, r >>= 1) {
			if (l & 1)vl = Op::merge(vl, dat[l++]);
			if (r & 1)vr = Op::merge(dat[--r], vr);
		}
		return Op::merge(vl, vr);
	}
	inline T operator[](int idx) { return dat[idx + len]; }
};
template<class Type> struct node {
	using T = Type;
	inline static T unit = INF;
	inline static T merge(T vl, T vr) { return min(vl, vr); }
	inline static T update(T vl, T vr) { return min(vl, vr); }
};

template<class T> class compress {
    vector<T> xs;
public:
    compress() {}
    compress(const vector<T>& vs) {
        add(vs);
    }
    inline void add(const vector<T>& vs) {
        copy(all(vs), back_inserter(xs));
    }
    inline void add(const T& x) {
        xs.emplace_back(x);
    }
    inline void build() {
        sort(all(xs));
        xs.erase(unique(all(xs)), xs.end());
    }
    inline int idx(const T& x) const {
        return lower_bound(all(xs), x) - xs.begin();
    }
    inline T operator[](int k) const {
        return xs[k];
    }
    inline int size() const {
        return xs.size();
    }
};

#include <atcoder/modint>
using mint = atcoder::modint998244353;
int main() {
    init();

    int n; cin >> n;
    VI a(n), b(n); cin >> a >> b;

    //a[j] - a[i] <= abs(b[i] - b[j]);
    //a[j] - a[i] <= b[i] - b[j] || a[j] - a[i] <= b[j] - b[i];
    //a[j] + b[j] <= b[i] + a[i] || a[j] - b[j] <= a[i] - b[i];

    ll ans = 0;
    compress<int> cp, cm;
    FOR(i, 0, n) {
        cp.add(a[i] + b[i]);
        cm.add(a[i] - b[i]);
    }
    cp.build(), cm.build();
    SegmentTree<node<int>> p(cp.size()), m(cm.size());
    int r = n;
    rFOR(i, 0, n) {
        chmin(r, p.value(0, cp.idx(a[i] + b[i]) + 1));
        chmin(r, m.value(0, cm.idx(a[i] - b[i]) + 1));
        p.update(cp.idx(a[i] + b[i]), i);
        m.update(cm.idx(a[i] - b[i]), i);
        ans += r - i - 1;
    }
    print(ans);

    return 0;
}

yukicoder

結果

ソースコード