結果

問題 No.1332 Range Nearest Query
ユーザー SSRSSSRS
提出日時 2021-01-08 23:15:34
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 2,815 bytes
コンパイル時間 3,375 ms
コンパイル使用メモリ 203,352 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-16 17:23:18
合計ジャッジ時間 15,178 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
testcase_45 RE -
testcase_46 RE -
testcase_47 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target("avx2")
#pragma GCC optimize("unroll-loops")
#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
using namespace std;
const int INF = 1000000000;
template <typename T>
struct binary_indexed_tree{
	int N;
	vector<T> BIT;
	binary_indexed_tree(int n){
		N = 1;
		while (N < n){
			N *= 2;
		}
		BIT = vector<T>(N + 1, 0);
	}
	void add(int i, T x){
		i++;
		while (i <= N){
			BIT[i] += x;
			i += i & -i;
		}
	}
	T sum(int i){
		T ans = 0;
		while (i > 0){
			ans += BIT[i];
			i -= i & -i;
		}
		return ans;
	}
	T sum(int L, int R){
		return sum(R) - sum(L);
	}
	T all(){
		return BIT[N];
	}
};
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int N;
  cin >> N;
  vector<int> X(N);
  for (int i = 0; i < N; i++){
    cin >> X[i];
  }
  vector<int> X2 = X;
  sort(X2.begin(), X2.end());
  int Q;
  cin >> Q;
  assert(N <= 10000 && Q <= 10000);
  vector<int> l(Q), r(Q), x(Q);
  for (int i = 0; i < Q; i++){
    cin >> l[i] >> r[i] >> x[i];
    l[i]--;
  }
  vector<int> tv(Q, INF), fv(Q, -1);
  while (true){
    vector<int> mid(Q, -1);
    bool update = false;
    for (int i = 0; i < Q; i++){
      if (tv[i] - fv[i] > 1){
        mid[i] = (tv[i] + fv[i] + 2) / 2 - 1;
        update = true;
      }
    }
    if (!update){
      break;
    }
    vector<vector<int>> padd(N);
    for (int i = 0; i < N; i++){
      int p = lower_bound(X2.begin(), X2.end(), X[i]) - X2.begin();
      padd[i].push_back(p);
    }
    vector<vector<tuple<int, int, int>>> add(N);
    vector<vector<tuple<int, int, int>>> sub(N);
    for (int i = 0; i < Q; i++){
      if (mid[i] != -1){
        int L = x[i] - mid[i];
        int R = x[i] + mid[i];
        int Lp = lower_bound(X2.begin(), X2.end(), L) - X2.begin();
        int Rp = upper_bound(X2.begin(), X2.end(), R) - X2.begin();
        if (l[i] > 0){
          sub[l[i] - 1].push_back(make_tuple(Lp, Rp, i));
        }
        add[r[i] - 1].push_back(make_tuple(Lp, Rp, i));
      }
    }
    binary_indexed_tree<int> BIT(N);
    vector<int> cnt(Q, 0);
    for (int i = 0; i < N; i++){
      for (int j : padd[i]){
        BIT.add(j, 1);
      }
      int addcnt = add[i].size();
      for (int j = 0; j < addcnt; j++){
        int L = get<0>(add[i][j]);
        int R = get<1>(add[i][j]);
        int id = get<2>(add[i][j]);
        cnt[id] += BIT.sum(L, R);
      }
      int subcnt = sub[i].size();
      for (int j = 0; j < subcnt; j++){
        int L = get<0>(sub[i][j]);
        int R = get<1>(sub[i][j]);
        int id = get<2>(sub[i][j]);
        cnt[id] -= BIT.sum(L, R);
      }
    }
    for (int i = 0; i < Q; i++){
      if (mid[i] != -1){
        if (cnt[i] == 0){
          fv[i] = mid[i];
        } else {
          tv[i] = mid[i];
        }
      }
    }
  }
  for (int i = 0; i < Q; i++){
    cout << tv[i] << "\n";
  }
}
0