結果
問題 |
No.2304 Distinct Elements
|
ユーザー |
![]() |
提出日時 | 2023-05-12 22:26:57 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 3,029 bytes |
コンパイル時間 | 1,370 ms |
コンパイル使用メモリ | 127,988 KB |
最終ジャッジ日時 | 2025-02-12 23:13:00 |
ジャッジサーバーID (参考情報) |
judge3 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 46 WA * 12 |
ソースコード
#ifndef LOCAL #define FAST_IO #endif // ============ #include <algorithm> #include <array> #include <bitset> #include <cassert> #include <cmath> #include <iomanip> #include <iostream> #include <list> #include <map> #include <numeric> #include <queue> #include <random> #include <set> #include <stack> #include <string> #include <tuple> #include <unordered_map> #include <unordered_set> #include <utility> #include <vector> #define OVERRIDE(a, b, c, d, ...) d #define REP2(i, n) for (i32 i = 0; i < (i32)(n); ++i) #define REP3(i, m, n) for (i32 i = (i32)(m); i < (i32)(n); ++i) #define REP(...) OVERRIDE(__VA_ARGS__, REP3, REP2)(__VA_ARGS__) #define PER(i, n) for (i32 i = (i32)(n) - 1; i >= 0; --i) #define ALL(x) begin(x), end(x) using namespace std; using u32 = unsigned int; using u64 = unsigned long long; using i32 = signed int; using i64 = signed long long; using f64 = double; using f80 = long double; template <typename T> using Vec = vector<T>; template <typename T> bool chmin(T &x, const T &y) { if (x > y) { x = y; return true; } return false; } template <typename T> bool chmax(T &x, const T &y) { if (x < y) { x = y; return true; } return false; } template <typename T> Vec<tuple<i32, i32, T>> runlength(const Vec<T> &a) { if (a.empty()) { return Vec<tuple<i32, i32, T>>(); } Vec<tuple<i32, i32, T>> ret; i32 prv = 0; REP(i, 1, a.size()) { if (a[i - 1] != a[i]) { ret.emplace_back(prv, i, a[i - 1]); prv = i; } } ret.emplace_back(prv, (i32)a.size(), a.back()); return ret; } #ifdef INT128 using u128 = __uint128_t; using i128 = __int128_t; istream &operator>>(istream &is, i128 &x) { i64 v; is >> v; x = v; return is; } ostream &operator<<(ostream &os, i128 x) { os << (i64)x; return os; } istream &operator>>(istream &is, u128 &x) { u64 v; is >> v; x = v; return is; } ostream &operator<<(ostream &os, u128 x) { os << (u64)x; return os; } #endif [[maybe_unused]] constexpr i32 INF = 1000000100; [[maybe_unused]] constexpr i64 INF64 = 3000000000000000100; struct SetUpIO { SetUpIO() { #ifdef FAST_IO ios::sync_with_stdio(false); cin.tie(nullptr); #endif cout << fixed << setprecision(15); } } set_up_io; // ============ #ifdef DEBUGF #else #define DBG(x) (void)0 #endif int main() { i32 n; cin >> n; Vec<i32> a(n); REP(i, n) { cin >> a[i]; } sort(ALL(a)); REP(i, n) { a[i] -= i; } Vec<i32> b(n - 1); REP(i, n - 1) { b[i] = a[i] - a[i + 1]; } i64 ans = 0; i32 cur = 0; while (cur < n - 1) { while (cur < n - 1 && b[cur] < 0) { ++cur; } Vec<i32> c; while (cur < n - 1 && b[cur] >= 0) { c.push_back(b[cur++]); } REP(i, c.size()) { ans += (i64)min(i + 1, (i32)c.size() - i) * c[i]; } } cout << ans << '\n'; }