結果

問題 No.2161 Black Market
ユーザー tnakao0123
提出日時 2022-12-15 15:23:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 518 ms / 7,000 ms
コード長 3,168 bytes
コンパイル時間 922 ms
コンパイル使用メモリ 63,184 KB
実行使用メモリ 10,496 KB
最終ジャッジ日時 2024-11-14 06:01:38
合計ジャッジ時間 4,114 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #

/* -*- coding: utf-8 -*-
 *
 * 2161.cc:  No.2161 Black Market - yukicoder
 */

#include<cstdio>
#include<vector>
#include<algorithm>
 
using namespace std;

/* constant */

const int MAX_N = 34;
const int MAX_M = MAX_N / 2;
const int MBITS = 1 << MAX_M;

/* typedef */

typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> vvi;

template <typename T>
struct BIT {
  int n;
  vector<T> bits;
  
  BIT() {}
  BIT(int _n) { init(_n); }

  void init(int _n) {
    n = _n;
    bits.assign(n + 1, 0);
  }

  T sum(int x) {
    x = min(x, n);
    T s = 0;
    while (x > 0) {
      s += bits[x];
      x -= (x & -x);
    }
    return s;
  }

  void add(int x, T v) {
    if (x <= 0) return;
    while (x <= n) {
      bits[x] += v;
      x += (x & -x);
    }
  }

  int lower_bound(T v) {
    int	k = 1;
    while ((k << 1) <= n) k <<=	1;
    int	x = 0;
    for	(; k > 0; k >>= 1)
      if (x + k <= n && bits[x + k] < v) {
        x += k;
        v -= bits[x];
      }
    return x + 1;
  }
};

/* global variables */

int as[MAX_N], bs[MAX_N], bnums[MBITS];
ll ass0[MBITS], ass1[MBITS], bss0[MBITS], bss1[MBITS];
ll uas[MBITS];
BIT<int> bit;

/* subroutines */

bool ltbss0(const int a, const int b) { return bss0[a] < bss0[b]; }
bool ltbss1(const int a, const int b) { return bss1[a] < bss1[b]; }

/* main */

int main() {
  bnums[0] = 0;
  for (int bits = 1, msb = 1; bits < MBITS; bits++) {
    if ((msb << 1) <= bits) msb <<= 1;
    bnums[bits] = bnums[bits ^ msb] + 1;
  }
  
  int n, k, l, p;
  scanf("%d%d%d%d", &n, &k, &l, &p);
  for (int i = 0; i < n; i++) scanf("%d%d", as + i, bs + i);

  int m0 = n / 2, m1 = n - m0;
  int mbits0 = 1 << m0, mbits1 = 1 << m1;

  vvi bvs0(m0 + 1), bvs1(m1 + 1);
  for (int bits = 0; bits < mbits0; bits++) bvs0[bnums[bits]].push_back(bits);
  for (int bits = 0; bits < mbits1; bits++) bvs1[bnums[bits]].push_back(bits);

  for (int bits = 1, msb = 1, msi = 0; bits < mbits0; bits++) {
    if ((msb << 1) <= bits) msb <<= 1, msi++;
    ass0[bits] = ass0[bits ^ msb] + as[msi];
    bss0[bits] = bss0[bits ^ msb] + bs[msi];
  }

  for (int bits = 1, msb = 1, msi = 0; bits < mbits1; bits++) {
    if ((msb << 1) <= bits) msb <<= 1, msi++;
    ass1[bits] = ass1[bits ^ msb] + as[m0 + msi];
    bss1[bits] = bss1[bits ^ msb] + bs[m0 + msi];
    uas[bits] = ass1[bits];
  }

  for (int i = 0; i <= m0; i++) sort(bvs0[i].begin(), bvs0[i].end(), ltbss0);
  for (int i = 0; i <= m1; i++) sort(bvs1[i].begin(), bvs1[i].end(), ltbss1);

  sort(uas, uas + mbits1);
  int un = unique(uas, uas + mbits1) - uas;

  ll sum = 0;
  for (int i0 = 0; i0 <= m0; i0++)
    for (int i1 = 0; i1 <= m1 && i0 + i1 <= k; i1++) {
      vi &bv0 = bvs0[i0], &bv1 = bvs1[i1];
      if (bv0.empty() || bv1.empty()) continue;

      int s0 = bv0.size(), s1 = bv1.size();
      bit.init(un);

      for (int i = 0, j = s1 - 1; i < s0; i++) {
	ll bi = bss0[bv0[i]], ai = ass0[bv0[i]];

	while (j >= 0 && bss1[bv1[j]] + bi >= p) {
	  int ui = lower_bound(uas, uas + un, ass1[bv1[j]]) - uas;
	  bit.add(ui + 1, 1);
	  j--;
	}

	int ui = upper_bound(uas, uas + un, l - ai) - uas;
	sum += bit.sum(ui);
      }
    }

  printf("%lld\n", sum);
  return 0;
}
0