結果

問題 No.956 Number of Unbalanced
ユーザー climpetclimpet
提出日時 2019-12-20 02:31:26
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 756 ms / 2,000 ms
コード長 1,749 bytes
コンパイル時間 582 ms
コンパイル使用メモリ 58,200 KB
実行使用メモリ 7,936 KB
最終ジャッジ日時 2024-07-07 03:39:01
合計ジャッジ時間 4,130 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;

namespace{
typedef long long LL;

template <class Tp>
struct fenwick_tree{
	typedef unsigned size_type;
	std::vector<Tp> bit;
	int minidx;

	fenwick_tree(){}
	explicit fenwick_tree(size_type n, int minidx_ = 1){
		init(n, minidx_);
	}
	void init(size_type n, int minidx_ = 1){
		bit.assign(n + 1, Tp());
		minidx = minidx_;
	}
	Tp sum(int i_) const{
		if(i_ < minidx){ return Tp(); }
		size_type i = i_ - minidx + 1;
		Tp s = Tp();
		while(i){
			s += bit[i];
			i -= i & -i;
		}
		return s;
	}
	void add(int i_, Tp x){
		if(i_ < minidx){ return; }
		size_type i = i_ - minidx + 1;
		size_type sz = bit.size();
		while(i < sz){
			bit[i] += x;
			i += i & -i;
		}
	}
};

void main2(){
	int n;
	scanf("%d", &n);
	vector<int> as(n);
	for(int i = 0; i < n; ++i){
		scanf("%d", &as[i]);
	}
	int mval = *max_element(as.begin(), as.end());
	vector<vector<int> > postbl(mval + 1);
	for(int i = 0; i < n; ++i){
		postbl[as[i]].push_back(i);
	}

	fenwick_tree<int> ft;
	vector<int> hs;
	hs.reserve(n + 1);
	LL ans = 0;
	for(int x = 0; x <= mval; ++x){
		const vector<int> &poss = postbl[x];
		int sz = poss.size();
		for(int l = 0; l < sz; ){
			int r;
			for(r = l + 1; r < sz && poss[r] - poss[r - 1] <= sz; ++r);
			int beg = max(poss[l] - sz + 1, 0);
			int en = min(poss[r - 1] + sz, n);
			hs.assign(1, 0);
			for(int i = beg; i < en; ++i){
				hs.push_back(hs.back() + (x == as[i] ? 1 : -1));
			}
			int minh = *min_element(hs.begin(), hs.end());
			int maxh = *max_element(hs.begin(), hs.end());
			ft.init(maxh - minh + 2, minh);
			for(int y : hs){
				ans += ft.sum(y - 1);
				ft.add(y, 1);
			}
			l = r;
		}
	}
	printf("%lld\n", ans);
}

}

int main(){ main2(); }
0