結果

問題 No.2161 Black Market
コンテスト
ユーザー Kude
提出日時 2022-12-12 23:13:52
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 104 ms / 7,000 ms
コード長 2,667 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,880 ms
コンパイル使用メモリ 244,860 KB
最終ジャッジ日時 2025-02-09 10:38:36
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
namespace {
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-function"
#include<atcoder/all>
#pragma GCC diagnostic pop
using namespace std;
using namespace atcoder;
#define rep(i,n) for(int i = 0; i < (int)(n); i++)
#define rrep(i,n) for(int i = (int)(n) - 1; i >= 0; i--)
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } else return false; }
template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return true; } else return false; }
using ll = long long;
using P = pair<int,int>;
using VI = vector<int>;
using VVI = vector<VI>;
using VL = vector<ll>;
using VVL = vector<VL>;

template <class S>
struct value_compression : vector<S> {
  bool built = false;
  using VS = vector<S>;
  using VS::VS;
  value_compression(vector<S> v) : vector<S>(move(v)) {}
  void build() {
    sort(this->begin(), this->end());
    this->erase(unique(this->begin(), this->end()), this->end());
    built = true;
  }
  template <class T>
  void convert(T first, T last) {
    assert(built);
    for (; first != last; ++first) *first = (*this)(*first);
  }
  int operator()(S x) {
    assert(built);
    return lower_bound(this->begin(), this->end(), x) - this->begin();
  }
  void clear() {
    this->clear();
    built = false;
  }
};

} int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  int n, k, l, p;
  cin >> n >> k >> l >> p;
  vector<vector<pair<ll, ll>>> ps1{{{0, 0}}}, ps2{{{0, 0}}};
  rep(_, n) {
    int a, b;
    cin >> a >> b;
    int sz = ps1.size();
    ps1.emplace_back();
    rrep(i, sz) {
      for(auto [x, y]: ps1[i]) {
        ps1[i+1].emplace_back(x + a, y + b);
      }
    }
    swap(ps1, ps2);
  }
  int sz1 = ps1.size(), sz2 = ps2.size();
  value_compression<ll> vc;
  rep(i, sz1) for(auto [x, y]: ps1[i]) vc.emplace_back(y);
  vc.build();
  rep(i, sz1) for(auto& [_, y]: ps1[i]) y = vc(y);
  rep(i, sz1) sort(all(ps1[i]));
  rep(i, sz2) sort(rall(ps2[i]));
  vector<pair<ll, ll>> d;
  int included = 0;
  ll ans = 0;
  rrep(k2, sz2) {
    int k1 = min(sz1 - 1, k - k2) + 1;
    if (k1 <= 0) continue;
    while(included < k1) {
      int dsz = d.size();
      d.insert(d.end(), all(ps1[included]));
      included++;
      inplace_merge(d.begin(), d.begin() + dsz, d.end());
    }
    int dsz = d.size();
    int ptr = 0;
    int m = vc.size();
    fenwick_tree<int> ft(m);
    for(auto [x, y]: ps2[k2]) {
      while(ptr < dsz && d[ptr].first <= l - x) {
        auto [_, yi] = d[ptr++];
        ft.add(yi, 1);
      }
      ans += ft.sum(vc(p - y), m);
    }
  }
  cout << ans << '\n';
}
0