結果

問題 No.1332 Range Nearest Query
ユーザー SSRSSSRS
提出日時 2021-01-08 23:11:56
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,685 bytes
コンパイル時間 1,977 ms
コンパイル使用メモリ 185,540 KB
実行使用メモリ 192,488 KB
最終ジャッジ日時 2024-04-28 05:05:59
合計ジャッジ時間 7,687 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

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