結果
問題 |
No.2382 Amidakuji M
|
ユーザー |
|
提出日時 | 2023-07-14 22:13:05 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 76 ms / 2,000 ms |
コード長 | 3,169 bytes |
コンパイル時間 | 1,941 ms |
コンパイル使用メモリ | 201,780 KB |
最終ジャッジ日時 | 2025-02-15 14:10:18 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
#ifndef hari64 #include <bits/stdc++.h> // #pragma GCC target("avx2") // #pragma GCC optimize("O3") // #pragma GCC optimize("unroll-loops") #define debug(...) #else #include "util/viewer.hpp" #define debug(...) viewer::_debug(__LINE__, #__VA_ARGS__, __VA_ARGS__) #endif // clang-format off using namespace std; using ll = long long; using ld = long double; using pii = pair<int, int>; using pll = pair<ll, ll>; template <typename T> using vc = vector<T>; template <typename T> using vvc = vector<vc<T>>; template <typename T> using vvvc = vector<vvc<T>>; using vi = vc<int>; using vl = vc<ll>; using vpi = vc<pii>; using vpl = vc<pll>; #define ALL(x) begin(x), end(x) #define RALL(x) (x).rbegin(), (x).rend() constexpr int INF = 1001001001; constexpr long long INFll = 1001001001001001001; template <class T> bool chmax(T& a, const T& b) { return a < b ? a = b, 1 : 0; } template <class T> bool chmin(T& a, const T& b) { return a > b ? a = b, 1 : 0; } // clang-format on template <class T> struct fenwick_tree { fenwick_tree() : _n(0) {} explicit fenwick_tree(int n) : _n(n), data(n) {} explicit fenwick_tree(vector<T>& As) : _n(As.size()), data(As) {} // A[idx]+=x void add(int idx, T x) { assert(0 <= idx && idx < _n); idx++; while (idx <= _n) { data[idx - 1] += x; idx += idx & -idx; } } // Σ_[l,r) T sum(int l, int r) const { assert(0 <= l && l <= r && r <= _n); return _sum(r) - _sum(l); } // Σ_[0,r) T sum(int r) const { assert(0 <= r && r <= _n); return _sum(r); } // A[idx] O(logN) T get(int idx) const { assert(0 <= idx && idx < _n); return sum(idx, idx + 1); } // debug vector<T> state() const { vector<T> ret(_n); for (int i = 0; i < _n; i++) ret[i] = get(i); return ret; } inline T operator[](int idx) const { return get(idx); } private: int _n; vector<T> data; T _sum(int r) const { T s = 0; while (r > 0) s += data[r - 1], r -= r & -r; return s; } }; template <class T> // https://onlinejudge.u-aizu.ac.jp/problems/ALDS1_5_D long long inversion_number(const vector<T>& As) { vector<T> Bs = As, Xs(As.size()); sort(Bs.begin(), Bs.end()); Bs.erase(unique(Bs.begin(), Bs.end()), Bs.end()); assert(Bs.size() == As.size()); for (int i = 0; i < (int)As.size(); i++) Xs[i] = lower_bound(Bs.begin(), Bs.end(), As[i]) - Bs.begin(); int N = Xs.size(); long long ret = 0; fenwick_tree<T> f(N); for (int j = 0; j < N; j++) { ret += j - f.sum(Xs[j]); f.add(Xs[j], 1); } return ret; } int main() { cin.tie(0); ios::sync_with_stdio(false); ll N, M; cin >> N >> M; vl Ps(N); for (auto& P : Ps) cin >> P; ll inv = inversion_number(Ps); if (inv % M == 0) { cout << inv << endl; } else if ((M - inv % M) % 2 == 0) { cout << inv + (M - inv % M) << endl; } else if (M % 2 == 0) { cout << -1 << endl; } else { cout << inv + (M - inv % M) + M << endl; } return 0; }