結果

問題 No.568 じゃんじゃん　落とす　委員会
ユーザー pekempey
提出日時 2017-09-08 23:21:32
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,650 bytes
コンパイル時間 815 ms
コンパイル使用メモリ 81,008 KB
最終ジャッジ日時 2025-01-05 02:47:34
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24 WA * 2
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:43:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   43 |     scanf("%d %d %d", &x[i], &a[i], &b[i]);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>

using namespace std;

template<class T>
struct FenwickTree {
  std::vector<T> dat;

  FenwickTree(int n) : dat(n + 1) {}

  void add(int k, T v) {
    for (k++; k > 0; k &= k - 1) {
      dat[k] += v;
    }
  }

  T sum(int k) {
    T ret = 0;
    for (k++; k < dat.size(); k += k & -k) {
      ret += dat[k];
    }
    return ret;
  }
};

int main() {
  int n, m;
  cin >> n >> m;

  vector<int> p(n);
  vector<int> x(n), a(n), b(n);

  FenwickTree<int> ft1(1e5 + 10);
  FenwickTree<int> ft2(1e5 + 10);

  int two = 0;
  int three = 0;
  for (int i = 0; i < n; i++) {
    p[i] = i;
    scanf("%d %d %d", &x[i], &a[i], &b[i]);
    if (x[i] == 1) {
      ft1.add(b[i], 1);
    }
    if (x[i] == 2) {
      two++;
      ft2.add(b[i], 1);
    }
    if (x[i] == 3) {
      two++;
      three++;
    }
  }

  sort(p.begin(), p.end(), [&](int i, int j) {
    return a[i] > a[j];
  });

  int ans = 1e9;
  int i = 0;
  while (i <= n) {
    int ok = 0;
    int ng = 1e5 + 5;
    while (ng - ok > 1) {
      int mid = (ok + ng) / 2;
      if (ft1.sum(mid) + two >= m) {
        ok = mid;
      } else {
        ng = mid;
      }
    }
    ans = min(ans, ft2.sum(ok) + three);

    if (i == n) break;
    
    int j = i;
    while (i < n && a[p[i]] == a[p[j]]) {
      int ii = p[i];
      x[ii]++;
      if (x[ii] == 3) {
        ft2.add(b[ii], -1);
        three++;
      } else if (x[ii] == 2) {
        ft2.add(b[ii], 1);
        ft1.add(b[ii], -1);
        two++;
      } else if (x[ii] == 1) {
        ft1.add(b[ii], 1);
      }
      i++;
    }
  }

  cout << ans << endl;
}
0