結果

問題 No.1734 Decreasing Elements
コンテスト
ユーザー SSRS
提出日時 2021-11-05 22:16:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 149 ms / 3,000 ms
コード長 1,673 bytes
コンパイル時間 2,175 ms
コンパイル使用メモリ 181,864 KB
実行使用メモリ 12,076 KB
最終ジャッジ日時 2024-11-06 13:09:30
合計ジャッジ時間 5,381 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
template <typename T>
struct dual_segment_tree{
	int N;
	vector<T> ST;
	dual_segment_tree(vector<T> A){
		int n = A.size();
		N = 1;
		while (N < n){
			N *= 2;
		}
		ST = vector<T>(N * 2 - 1, 0);
		for (int i = 0; i < n; i++){
			ST[N - 1 + i] = A[i];
		}
	}
	T operator [](int k){
		k += N - 1;
		T ans = ST[k];
		while (k > 0){
			k = (k - 1) / 2;
			ans += ST[k];
		}
		return ans;
	}
	void range_add(int L, int R, T x, int i, int l, int r){
		if (R <= l || r <= L){
			return;
		} else if (L <= l && r <= R){
			ST[i] += x;
			return;
		} else {
			int m = (l + r) / 2;
			range_add(L, R, x, i * 2 + 1, l, m);
			range_add(L, R, x, i * 2 + 2, m, r);
			return;
		}
	}
	void range_add(int L, int R, T x){
		range_add(L, R, x, 0, 0, N);
	}
	void all_add(T x){
		ST[0] += x;
	}
};
int main(){
  int N;
  cin >> N;
  vector<int> A(N);
  for (int i = 0; i < N; i++){
    cin >> A[i];
  }
  vector<int> B(200001);
  for (int i = 0; i <= 200000; i++){
    B[i] = i;
  }
  dual_segment_tree<int> ST(B);
  priority_queue<tuple<int, int, int>> pq;
  pq.push(make_tuple(200001, 0, 200001));
  int ans = 0;
  for (int i = 0; i < N; i++){
    if (ST[A[i]] > 0){
      ans++;
      int c = ST[A[i]];
      vector<tuple<int, int, int>> tmp;
      while (get<0>(pq.top()) > c){
        int L = get<1>(pq.top());
        int R = get<2>(pq.top());
        pq.pop();
        pq.push(make_tuple(c, L, L + c));
        tmp.push_back(make_tuple(R - L - c, L + c, R));
        ST.range_add(L + c, R, -c);
      }
      int cnt = tmp.size();
      for (int j = 0; j < cnt; j++){
        pq.push(tmp[j]);
      }
    }
  }
  cout << ans << endl;
}
0