結果
問題 | No.2501 Maximum Inversion Number |
ユーザー | suisen |
提出日時 | 2023-07-12 19:09:20 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 112 ms / 2,000 ms |
コード長 | 1,850 bytes |
コンパイル時間 | 873 ms |
コンパイル使用メモリ | 80,024 KB |
実行使用メモリ | 6,384 KB |
最終ジャッジ日時 | 2023-10-13 18:04:37 |
合計ジャッジ時間 | 4,506 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,368 KB |
testcase_01 | AC | 34 ms
4,348 KB |
testcase_02 | AC | 55 ms
4,348 KB |
testcase_03 | AC | 39 ms
4,348 KB |
testcase_04 | AC | 41 ms
6,384 KB |
testcase_05 | AC | 40 ms
6,276 KB |
testcase_06 | AC | 42 ms
6,208 KB |
testcase_07 | AC | 35 ms
4,348 KB |
testcase_08 | AC | 40 ms
4,356 KB |
testcase_09 | AC | 23 ms
4,628 KB |
testcase_10 | AC | 39 ms
4,352 KB |
testcase_11 | AC | 35 ms
4,352 KB |
testcase_12 | AC | 34 ms
4,348 KB |
testcase_13 | AC | 1 ms
4,348 KB |
testcase_14 | AC | 45 ms
6,168 KB |
testcase_15 | AC | 34 ms
4,352 KB |
testcase_16 | AC | 112 ms
4,352 KB |
ソースコード
#include <algorithm> #include <cassert> #include <iostream> #include <numeric> #include <vector> int64_t solve(const size_t n, const uint64_t m, const std::vector<uint64_t>& l, const std::vector<uint64_t>& r) { const uint64_t sum_l = std::accumulate(l.begin(), l.end(), uint64_t(0)); const uint64_t sum_r = std::accumulate(r.begin(), r.end(), uint64_t(0)); if (m < sum_l or m > sum_r) { return -1; } if (m == sum_l) { uint64_t x = 0; for (uint64_t e : l) x += e * e; return (m * m - x) / 2; } const auto [t0, sum_t0] = [&] { auto is_ok = [&](uint64_t t) { uint64_t sum = 0; for (size_t i = 0; i < n; ++i) { sum += std::clamp(t, l[i], r[i]); } return std::make_pair(sum < m, sum); }; // binary search uint64_t tl = 0, tr = 1000000010; assert(is_ok(tl).first and not is_ok(tr).first); while (::abs(tl - tr) > 1) { uint64_t t = (tl + tr) / 2; (is_ok(t).first ? tl : tr) = t; } return std::make_pair(tl, is_ok(tl).second); }(); uint64_t d = m - sum_t0; uint64_t x = 0; for (size_t i = 0; i < n; ++i) { uint64_t c = std::clamp(t0, l[i], r[i]); if (d > 0 and l[i] <= t0 and t0 < r[i]) { ++c, --d; } x += c * c; } assert(d == 0); return (m * m - x) / 2; } int main() { std::ios::sync_with_stdio(false); std::cin.tie(nullptr); size_t t; std::cin >> t; while (t --> 0) { size_t n; uint64_t m; std::cin >> n >> m; std::vector<uint64_t> l(n), r(n); for (auto& e : l) std::cin >> e; for (auto& e : r) std::cin >> e; std::cout << solve(n, m, l, r) << '\n'; } return 0; }