結果
問題 | No.1300 Sum of Inversions |
ユーザー | jupiro |
提出日時 | 2020-12-02 16:54:13 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 4,090 bytes |
コンパイル時間 | 1,557 ms |
コンパイル使用メモリ | 140,688 KB |
実行使用メモリ | 15,768 KB |
最終ジャッジ日時 | 2024-09-13 10:32:13 |
合計ジャッジ時間 | 8,297 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,940 KB |
testcase_03 | AC | 110 ms
12,756 KB |
testcase_04 | AC | 110 ms
12,500 KB |
testcase_05 | AC | 88 ms
10,936 KB |
testcase_06 | AC | 126 ms
14,084 KB |
testcase_07 | AC | 120 ms
13,480 KB |
testcase_08 | AC | 133 ms
14,528 KB |
testcase_09 | AC | 133 ms
14,580 KB |
testcase_10 | AC | 72 ms
9,560 KB |
testcase_11 | AC | 73 ms
9,800 KB |
testcase_12 | AC | 115 ms
12,568 KB |
testcase_13 | AC | 109 ms
12,260 KB |
testcase_14 | AC | 156 ms
15,604 KB |
testcase_15 | AC | 135 ms
14,432 KB |
testcase_16 | AC | 113 ms
13,128 KB |
testcase_17 | AC | 71 ms
9,408 KB |
testcase_18 | AC | 81 ms
10,340 KB |
testcase_19 | AC | 95 ms
11,664 KB |
testcase_20 | AC | 100 ms
11,836 KB |
testcase_21 | AC | 97 ms
11,832 KB |
testcase_22 | AC | 90 ms
10,920 KB |
testcase_23 | AC | 128 ms
14,060 KB |
testcase_24 | AC | 93 ms
11,324 KB |
testcase_25 | AC | 77 ms
10,028 KB |
testcase_26 | AC | 77 ms
9,944 KB |
testcase_27 | AC | 85 ms
10,700 KB |
testcase_28 | AC | 138 ms
14,908 KB |
testcase_29 | AC | 97 ms
11,700 KB |
testcase_30 | AC | 134 ms
14,440 KB |
testcase_31 | AC | 88 ms
11,112 KB |
testcase_32 | AC | 95 ms
11,316 KB |
testcase_33 | AC | 63 ms
15,756 KB |
testcase_34 | AC | 75 ms
15,640 KB |
testcase_35 | AC | 79 ms
15,768 KB |
testcase_36 | WA | - |
ソースコード
#include <iostream> #include <string> #include <sstream> #include <stack> #include <algorithm> #include <cmath> #include <queue> #include <bitset> #include <iomanip> #include <limits> #include <chrono> #include <random> #include <array> #include <unordered_map> #include <functional> #include <complex> #include <numeric> #include <cctype> #include <map> #include <set> #include <cstdlib> #include <bitset> #include <tuple> #include <assert.h> #include <deque> #include <utility> #include <fstream> using namespace std; typedef long long ll; using ull = unsigned long long; template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; } template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; } //template<typename T> T gcd(T a, T b) { a = abs(a), b = abs(b); while (b > 0) { tie(a, b) = make_pair(b, a % b); }return a; } //mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count()); constexpr long long INF = 1LL << 60; constexpr int inf = 1000000007; //constexpr long long mod = 1000000007LL; constexpr long long mod = 998244353; template<typename T> struct BinaryIndexedTree { int n; vector<T> bit; BinaryIndexedTree() :n(0) {} BinaryIndexedTree(int _n) :n(_n) { bit = vector<T>(n + 1); } void add1(int idx, T val) { for (int i = idx; i <= n; i += i & -i) bit[i] += val; } T sum1(int idx) { T res = 0; for (int i = idx; i > 0; i -= i & -i) res += bit[i]; return res; } //0-indexed void add(int idx, T val) { add1(idx + 1, val); } //0-indexed [left, right) T sum(int left, int right) { return sum1(right) - sum1(left); } int lower_bound(T x) { int res = 0; int k = 1; while (2 * k <= n) k <<= 1; for (; k > 0; k >>= 1) { if (res + k <= n and bit[res + k] < x) { x -= bit[res + k]; res += k; } } return res; } }; template<typename T> struct Compress { vector<T> v; Compress() {} Compress(vector<T> _v) :v(_v) { build(); } void add(T x) { v.emplace_back(x); } void build() { sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end()); } void build(vector<T> _v) { v = _v; sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end()); } int get(T x) { return lower_bound(v.begin(), v.end(), x) - v.begin(); } T& operator[](int i) { return v[i]; } int size() { return (int)v.size(); } }; struct mint { long long x; mint(long long x = 0) :x((x% mod + mod) % mod) {} mint& operator+=(const mint a) { if ((x += a.x) >= mod) x -= mod; return *this; } mint& operator-=(const mint a) { if ((x += mod - a.x) >= mod) x -= mod; return *this; } mint& operator*=(const mint a) { (x *= a.x) %= mod; return *this; } mint operator+(const mint a) const { mint res(*this); return res += a; } mint operator-(const mint a) const { mint res(*this); return res -= a; } mint operator*(const mint a) const { mint res(*this); return res *= a; } mint pow(ll t) const { if (!t) return 1; mint a = pow(t >> 1); a *= a; if (t & 1) a *= *this; return a; } // for prime mod mint inv() const { return pow(mod - 2); } mint& operator/=(const mint a) { return (*this) *= a.inv(); } mint operator/(const mint a) const { mint res(*this); return res /= a; } }; int main() { cin.tie(nullptr); ios::sync_with_stdio(false); int n; cin >> n; vector<ll> a(n); for (int i = 0; i < n; i++) cin >> a[i]; Compress comp(a); vector<ll> lsum(n), lcnt(n), rsum(n), rcnt(n); { BinaryIndexedTree<ll> bsum(n), bcnt(n); for (int i = 0; i < n; i++) { int t = comp.get(a[i]); lcnt[i] = bcnt.sum(t + 1, n); lsum[i] = bsum.sum(t + 1, n); bcnt.add(t, 1); bsum.add(t, a[i]); } } { BinaryIndexedTree<ll> bsum(n), bcnt(n); for (int i = n - 1; i >= 0; i--) { int t = comp.get(a[i]); rcnt[i] = bcnt.sum(0, t); rsum[i] = bsum.sum(0, t); bcnt.add(t, 1); bsum.add(t, a[i]); } } mint res = 0; for (int i = 0; i < n; i++) { res += lsum[i] * rcnt[i]; res += lcnt[i] * rsum[i]; res += mint(a[i]) * lcnt[i] * rcnt[i]; } cout << res.x << "\n"; }