結果

問題 No.2501 Maximum Inversion Number
ユーザー 👑 emthrmemthrm
提出日時 2023-07-13 16:09:28
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,694 bytes
コンパイル時間 986 ms
コンパイル使用メモリ 102,796 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-11 19:40:51
合計ジャッジ時間 4,450 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 87 ms
5,248 KB
testcase_02 AC 195 ms
5,248 KB
testcase_03 AC 116 ms
5,248 KB
testcase_04 AC 93 ms
5,248 KB
testcase_05 AC 96 ms
5,248 KB
testcase_06 AC 103 ms
5,248 KB
testcase_07 RE -
testcase_08 AC 113 ms
5,248 KB
testcase_09 AC 91 ms
5,248 KB
testcase_10 AC 106 ms
5,248 KB
testcase_11 AC 78 ms
5,248 KB
testcase_12 AC 79 ms
5,248 KB
testcase_13 RE -
testcase_14 AC 130 ms
5,248 KB
testcase_15 RE -
testcase_16 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cassert>
#include <cstdint>
#include <iostream>
#include <iterator>
#include <numeric>
#include <vector>

std::int64_t NChoose2(const int n) { return std::int64_t{n} * (n - 1) / 2; }

// <WA> #890216 ベース
// 二分探索の下界の初期値が間違っている。
std::int64_t Solve(
    int m, const std::vector<int>& l, const std::vector<int>& r) {
  const int n = l.size();
  assert(std::ssize(r) == n);

  if (std::reduce(l.begin(), l.end(), INT64_C(0)) > m) return -1;

  int t0 = m + 1;
  for (int lbound = 1; lbound + 1 < t0;) {  // 本来は 0
    const int t = std::midpoint(lbound, t0);
    std::int64_t length = 0;
    for (int i = 0; i < n; ++i) {
      length += std::clamp(t, l[i], r[i]);
    }
    (length >= m ? t0 : lbound) = t;
  }
  if (t0 > m) return -1;

  std::int64_t ans = NChoose2(m);
  int num_of_t0 = 0;
  for (int i = 0; i < n; ++i) {
    const int c_i = std::clamp(t0, l[i], r[i]);
    m -= c_i;
    ans -= NChoose2(c_i);
    if (l[i] < t0 && t0 <= r[i]) ++num_of_t0;
  }
  assert(m <= 0 && -m <= num_of_t0);
  ans -= (NChoose2(t0 - 1) - NChoose2(t0)) * -m;
  return ans;
}

int main() {
  constexpr int kMaxT = 200000, kMaxN = 200000, kMaxM = 1000000000;

  int t;
  std::cin >> t;
  assert(1 <= t && t <= kMaxT);

  while (t--) {
    int n, m;
    std::cin >> n >> m;
    assert(1 <= n && n <= kMaxN && 1 <= m && m <= kMaxM);
    std::vector<int> l(n);
    for (int i = 0; i < n; ++i) {
      std::cin >> l[i];
    }
    std::vector<int> r(n);
    for (int i = 0; i < n; ++i) {
      std::cin >> r[i];
      assert(0 <= l[i] && l[i] <= r[i] && r[i] <= kMaxM);
    }
    std::cout << Solve(m, l, r) << '\n';
  }
  return 0;
}
0