結果

問題 No.2501 Maximum Inversion Number
ユーザー 👑 emthrmemthrm
提出日時 2023-07-13 01:04:24
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 588 ms / 2,000 ms
コード長 2,832 bytes
コンパイル時間 1,466 ms
コンパイル使用メモリ 131,540 KB
実行使用メモリ 7,424 KB
最終ジャッジ日時 2024-09-15 13:55:30
合計ジャッジ時間 5,084 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <algorithm>
#include <cassert>
#include <cstdint>
#include <functional>
#include <iostream>
#include <iterator>
#include <numeric>
#include <vector>
std::int64_t NChoose2(const int n) { return std::int64_t{n} * (n - 1) / 2; }
std::int64_t Solve(const int m,
const std::vector<int>& l, const std::vector<int>& r) {
const int n = l.size();
assert(std::ssize(r) == n);
std::vector<int> order(n * 2); // L_i, R_i
std::iota(order.begin(), order.end(), 0);
std::ranges::stable_sort(
order, {},
[n, &l, &r](const int i) -> int { return i < n ? l[i] : r[i - n]; });
assert(order.back() >= n);
std::int64_t length = std::reduce(l.begin(), l.end(), INT64_C(0));
if (length > m) return -1;
std::int64_t ans = std::transform_reduce(
l.begin(), l.end(), INT64_C(0), std::plus<std::int64_t>(), NChoose2);
if (length == m) return NChoose2(m) - ans;
int optional_values = 0; //
for (int i = 0; i < n * 2; ++i) {
if (order[i] < n) {
const int index = order[i];
length -= l[index];
ans -= NChoose2(l[index]);
++optional_values;
} else {
const int index = order[i] - n;
length += r[index];
assert(length <= m);
ans += NChoose2(r[index]);
--optional_values;
}
if (i + 1 < n * 2) {
const int next_v = (order[i + 1] < n ?
l[order[i + 1]] : r[order[i + 1] - n]);
// t_0
if (length + std::int64_t{next_v} * optional_values >= m) {
assert(optional_values > 0);
const int t0 = (m - length + optional_values - 1) / optional_values - 1;
assert((order[i] < n ? l[order[i]] : r[order[i] - n]) <= t0 &&
t0 < next_v);
// t_0
const int num_of_t0 = (t0 + 1) * optional_values - (m - length);
assert(0 <= num_of_t0 && num_of_t0 < optional_values);
ans += NChoose2(t0) * num_of_t0;
ans += NChoose2(t0 + 1) * (optional_values - num_of_t0);
return NChoose2(m) - ans;
}
}
}
assert(length == std::reduce(r.begin(), r.end(), INT64_C(0)) && length < m);
return -1;
}
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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0