結果

問題 No.956 Number of Unbalanced
コンテスト
ユーザー FF256grhy
提出日時 2019-12-19 01:11:54
言語 C++14
(gcc 13.3.0 + boost 1.87.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;
}
0