結果
問題 | No.956 Number of Unbalanced |
ユーザー | FF256grhy |
提出日時 | 2019-12-19 01:11:54 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 764 ms / 2,000 ms |
コード長 | 4,024 bytes |
コンパイル時間 | 2,041 ms |
コンパイル使用メモリ | 181,336 KB |
実行使用メモリ | 12,080 KB |
最終ジャッジ日時 | 2024-07-06 23:14:10 |
合計ジャッジ時間 | 4,799 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 4 ms
6,812 KB |
testcase_01 | AC | 3 ms
6,944 KB |
testcase_02 | AC | 4 ms
6,944 KB |
testcase_03 | AC | 3 ms
6,940 KB |
testcase_04 | AC | 3 ms
6,944 KB |
testcase_05 | AC | 3 ms
6,944 KB |
testcase_06 | AC | 41 ms
7,248 KB |
testcase_07 | AC | 37 ms
7,040 KB |
testcase_08 | AC | 46 ms
7,168 KB |
testcase_09 | AC | 36 ms
6,944 KB |
testcase_10 | AC | 35 ms
6,940 KB |
testcase_11 | AC | 764 ms
12,080 KB |
testcase_12 | AC | 47 ms
7,168 KB |
testcase_13 | AC | 41 ms
11,036 KB |
testcase_14 | AC | 40 ms
7,924 KB |
testcase_15 | AC | 50 ms
11,116 KB |
testcase_16 | AC | 37 ms
7,680 KB |
testcase_17 | AC | 45 ms
11,012 KB |
testcase_18 | AC | 40 ms
7,936 KB |
testcase_19 | AC | 57 ms
11,092 KB |
testcase_20 | AC | 40 ms
7,936 KB |
testcase_21 | AC | 40 ms
10,936 KB |
testcase_22 | AC | 47 ms
10,996 KB |
testcase_23 | AC | 35 ms
7,648 KB |
testcase_24 | AC | 57 ms
11,092 KB |
testcase_25 | AC | 40 ms
6,944 KB |
testcase_26 | AC | 117 ms
6,940 KB |
testcase_27 | AC | 39 ms
6,940 KB |
testcase_28 | AC | 39 ms
6,944 KB |
testcase_29 | AC | 39 ms
7,936 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; using LL = long long int; #define incID(i, l, r) for(int i = (l) ; i < (r); ++i) #define incII(i, l, r) for(int i = (l) ; i <= (r); ++i) #define decID(i, l, r) for(int i = (r) - 1; i >= (l); --i) #define decII(i, l, r) for(int i = (r) ; i >= (l); --i) #define inc(i, n) incID(i, 0, n) #define inc1(i, n) incII(i, 1, n) #define dec(i, n) decID(i, 0, n) #define dec1(i, n) decII(i, 1, n) #define inID(v, l, r) ((l) <= (v) && (v) < (r)) #define inII(v, l, r) ((l) <= (v) && (v) <= (r)) #define PB push_back #define EB emplace_back #define MP make_pair #define FI first #define SE second #define ALL(v) v.begin(), v.end() #define RALL(v) v.rbegin(), v.rend() auto setmin = [](auto & a, auto b) { return (b < a ? a = b, true : false); }; auto setmax = [](auto & a, auto b) { return (b > a ? a = b, true : false); }; auto setmineq = [](auto & a, auto b) { return (b <= a ? a = b, true : false); }; auto setmaxeq = [](auto & a, auto b) { return (b >= a ? a = b, true : false); }; auto fl = [](auto a, auto b) { assert(b != 0); return a / b - (a % b != 0 && ((a >= 0) != (b >= 0)) ? 1 : 0); }; auto ce = [](auto a, auto b) { assert(b != 0); return a / b + (a % b != 0 && ((a >= 0) == (b >= 0)) ? 1 : 0); }; auto mo = [](auto a, auto b) { assert(b != 0); a %= b; if(a < 0) { a += abs(b); } return a; }; LL gcd(LL a, LL b) { return (b == 0 ? a : gcd(b, a % b)); } LL lcm(LL a, LL b) { return a / gcd(a, b) * b; } #define bit(b, i) (((b) >> (i)) & 1) #define SI(v) static_cast<int>(v.size()) #define RF(e, v) for(auto & e: v) #define until(e) while(! (e)) #define if_not(e) if(! (e)) #define ef else if #define UR assert(false) // ---- ---- template<typename T> class SegmentTree { private: int n, s; vector<T> a; function<T(T &, T &)> f; T e; bool ok; public: SegmentTree() { n = 0; } SegmentTree(int nn, function<T(T &, T &)> ff, T ee) { init(nn, ff, ee); } void init(int nn, function<T(T &, T &)> ff, T ee) { n = nn; f = ff; e = ee; s = 1; while(s < n) { s *= 2; } a = vector<T>(2 * s, e); ok = true; } void shift(int & p) { assert(inID(p, 0, n)); p += s; } void apply(int p, function<void(T &)> g) { shift(p); g(a[p]); while(p > 1) { p /= 2; a[p] = f(a[2 * p], a[2 * p + 1]); } } T fold_ID(int l, int r) { assert(ok); assert(inII(l, 0, n)); l += s; assert(inII(r, 0, n)); r += s; r--; T x = e, y = e; while(l <= r) { if(l % 2 == 1) { x = f(x, a[l]); l++; } if(r % 2 == 0) { y = f(a[r], y); r--; } l /= 2; r /= 2; } return f(x, y); } T fold_II(int l, int r) { return fold_ID(l + 0, r + 1); } T fold_CI(int l, int r) { return fold_ID(l + 1, r + 1); } T fold_CD(int l, int r) { return fold_ID(l + 1, r + 0); } const T & operator[](int p) { shift(p); return a[p]; } T & ref(int p) { shift(p); ok = false; return a[p]; } void update() { dec(i, s) { a[i] = f(a[2 * i], a[2 * i + 1]); } ok = true; } }; #define OP(s) [&](auto & A, auto & B) { return s; } #define AP(s) [&](auto & A) { s; } // ---- const int M = 100000; int main() { int n; cin >> n; vector<int> a(n); inc(i, n) { cin >> a[i]; a[i]--; } vector<vector<int>> b(M); inc(i, n) { b[a[i]].PB(i); } LL ans = 0; RF(e, b) { sort(ALL(e)); LL s = SI(e); if(s * s <= 2 * n) { inc(i, s) { inc(j, i + 1) { int c = i - j + 1; int d = e[i] - e[j] + 1; LL z = 2 * c - d; if(z < 0) { continue; } LL x = abs((inID(i + 1, 0, s) ? e[i + 1] : n) - e[i]); LL y = abs((inID(j - 1, 0, s) ? e[j - 1] : -1) - e[j]); if(z < min(x, y)) { ans += z * (z + 1) / 2; } else { setmin(x, z); setmin(y, z); LL w = max(x + y - 1 - z, 0LL); ans += x * y - w * (w + 1) / 2; } } } } else { vector<int> v(n, -1); RF(f, e) { v[f] = +1; } SegmentTree<LL> st(2 * n + 1, OP(A + B), 0); LL sum = n; inc(i, n) { st.apply(sum, AP(A++)); sum += v[i]; ans += st.fold_ID(0, sum); } } } cout << ans << endl; return 0; }