結果

問題 No.728 ギブ and テイク
コンテスト
ユーザー たこし
提出日時 2018-09-28 17:12:39
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 541 ms / 3,000 ms
コード長 3,070 bytes
記録
コンパイル時間 1,910 ms
コンパイル使用メモリ 207,548 KB
最終ジャッジ日時 2025-01-06 13:51:24
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#include <cstdio>
#include <string>

using namespace std;

#define INF 100000000
#define YJ 1145141919
#define INF_INT_MAX 2147483647
#define INF_LL_MAX 9223372036854775807
#define EPS 1e-10
#define MOD 1000000007
#define Pi acos(-1)
#define LL long long
#define ULL unsigned long long
#define LD long double

class Node {
public:
    Node(LL v = unitValue) : value(v) {

    }

    Node operator + (const Node& monoid) const {
        return Node(this->getValue() + monoid.getValue());
    }

    static Node unit() {
        return Node(unitValue);
    }

    void setValue(LL l) {
        this->value = l;
    }

    LL getValue() const {
        return this->value;
    }

private:
    LL value;
    static const LL unitValue = 0;
};

template <class T>
class SegmentTree {
public:
    SegmentTree(int size) {
        init(size);
    }

    void init(int size) {
        vec.clear();
        k = 1;
        while(k < size) {
            k *= 2;
        }
        for(int i = 0; i < 2*k; i++) {
            vec.push_back(T::unit());
        }
    }

    void update(int i, T x) { //iをxに変更する
        i += k-1;
        vec[i] = x;
        while(i > 0) {
            i = (i-1)/2;
            vec[i] = vec[i*2+1] + vec[i*2+2];
        }
    }

    T find(int l, int r) {
        return _find(l, r, 0, 0, k);
    }

private:
    vector<T> vec;
    int k;

    T _find(int l, int r, int k, int x, int y) {
        if (r <= x || y <= l) { //重なってない場合
            return T::unit();
        } else if (l <= x && y <= r) { //完全にかぶってる場合
            return vec[k];
        } else { //中途半端にかぶってる場合
            int mid = (x+y)/2;
            T lNode = _find(l, r, 2*k+1, x, mid);
            T rNode = _find(l, r, 2*k+2, mid, y);
            return lNode + rNode;
        }
    }
};

#define int long long

const int MAX_N = 300005;

int N;

class Child {
public:
  int a, l, r;
  int index;
  bool operator < (const Child& c) const {
    return a+r < c.a+c.r;
  }

  

}children[MAX_N];

int A[MAX_N];

signed main()
{
  unique_ptr<SegmentTree<Node> > segmentTree = unique_ptr<SegmentTree<Node> > (new SegmentTree<Node>(MAX_N));
  
  cin >> N;
  for(int i = 0; i < N; i++) {
    cin >> children[i].a;
    A[i] = children[i].a;
    segmentTree->update(i, 1);
  }

  for(int i = 0; i < N; i++) {
    cin >> children[i].l >> children[i].r;
    children[i].index = i;
  }

  int ans = 0;

  vector<pair<int, int> > vec;
  for(int i = 0; i < N; i++) {
    vec.push_back(make_pair(children[i].a + children[i].r, i+1));
    vec.push_back(make_pair(children[i].a, -(i+1)));
  }

  sort(begin(vec), end(vec));

  for(auto it : vec) {
    int index = it.second;
    if(index > 0) {
      index--;
      segmentTree->update(index, 0);
    } else {
      index = -1*index;
      index--;
      const Child& child = children[index];
      int l = lower_bound(A, A+index, child.a - child.l) - A;
      ans += segmentTree->find(l, index).getValue();
    }
  }

  cout << ans << endl;

  return 0;
}
0