結果

問題 No.2974 関数の芽
ユーザー SSRSSSRS
提出日時 2024-11-29 22:13:49
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 548 ms / 2,000 ms
コード長 3,269 bytes
コンパイル時間 2,682 ms
コンパイル使用メモリ 220,124 KB
実行使用メモリ 13,952 KB
最終ジャッジ日時 2024-11-29 22:13:59
合計ジャッジ時間 8,984 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 2 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 5 ms
5,248 KB
testcase_13 AC 36 ms
5,248 KB
testcase_14 AC 363 ms
5,504 KB
testcase_15 AC 548 ms
13,824 KB
testcase_16 AC 423 ms
13,824 KB
testcase_17 AC 501 ms
13,824 KB
testcase_18 AC 480 ms
13,824 KB
testcase_19 AC 517 ms
13,824 KB
testcase_20 AC 489 ms
13,824 KB
testcase_21 AC 479 ms
13,824 KB
testcase_22 AC 482 ms
13,952 KB
testcase_23 AC 496 ms
13,824 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
struct dual_segment_tree{
  int N;
  vector<long long> ST;
  dual_segment_tree(int n){
    N = 1;
    while (N < n){
      N *= 2;
    }
    ST.resize(N * 2 - 1, 0);
  }
  void range_add(int L, int R, long long x, int i, int l, int r){
    if (r <= L || R <= l){
    } else if (L <= l && r <= R){
      ST[i] += x;
    } 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);
    }
  }
  void range_add(int L, int R, long long x){
    range_add(L, R, x, 0, 0, N);
  }
  long long operator [](int k){
    k += N - 1;
    long long ans = ST[k];
    while (k > 0){
      k = (k - 1) / 2;
      ans += ST[k];
    }
    return ans;
  }
};
int main(){
  int Q;
  cin >> Q;
  vector<int> K(Q), L(Q), M(Q), N(Q), X(Q);
  for (int i = 0; i < Q; i++){
    cin >> K[i] >> L[i] >> M[i] >> N[i] >> X[i];
  }
  vector<int> X2 = X;
  sort(X2.begin(), X2.end());
  X2.erase(unique(X2.begin(), X2.end()), X2.end());
  int cnt = X2.size();
  dual_segment_tree ST_A(cnt), ST_B(cnt), ST_L(cnt), ST_R(cnt);
  for (int i = 0; i < Q; i++){
    if (K[i] > 0){
      int pl = partition_point(X2.begin(), X2.end(), [&](long long x){
        return K[i] * x + L[i] < 0;
      }) - X2.begin();
      int pr = partition_point(X2.begin(), X2.end(), [&](long long x){
        return K[i] * x + L[i] <= 0;
      }) - X2.begin();
      ST_A.range_add(pr, cnt, K[i]);
      ST_B.range_add(pr, cnt, L[i]);
      ST_L.range_add(pr, cnt, K[i]);
      ST_R.range_add(pl, cnt, K[i]);
    } else if (K[i] < 0){
      int pl = partition_point(X2.begin(), X2.end(), [&](long long x){
        return K[i] * x + L[i] > 0;
      }) - X2.begin();
      int pr = partition_point(X2.begin(), X2.end(), [&](long long x){
        return K[i] * x + L[i] >= 0;
      }) - X2.begin();
      ST_A.range_add(0, pr, K[i]);
      ST_B.range_add(0, pr, L[i]);
      ST_L.range_add(0, pr, K[i]);
      ST_R.range_add(0, pl, K[i]);
    } else {
      ST_B.range_add(0, cnt, max(L[i], 0));
    }
    if (M[i] > 0){
      int pl = partition_point(X2.begin(), X2.end(), [&](long long x){
        return M[i] * x + N[i] < 0;
      }) - X2.begin();
      int pr = partition_point(X2.begin(), X2.end(), [&](long long x){
        return M[i] * x + N[i] <= 0;
      }) - X2.begin();
      ST_A.range_add(pr, cnt, -M[i]);
      ST_B.range_add(pr, cnt, -N[i]);
      ST_L.range_add(pr, cnt, -M[i]);
      ST_R.range_add(pl, cnt, -M[i]);
    } else if (M[i] < 0){
      int pl = partition_point(X2.begin(), X2.end(), [&](long long x){
        return M[i] * x + N[i] > 0;
      }) - X2.begin();
      int pr = partition_point(X2.begin(), X2.end(), [&](long long x){
        return M[i] * x + N[i] >= 0;
      }) - X2.begin();
      ST_A.range_add(0, pr, -M[i]);
      ST_B.range_add(0, pr, -N[i]);
      ST_L.range_add(0, pr, -M[i]);
      ST_R.range_add(0, pl, -M[i]);
    } else {
      ST_B.range_add(0, cnt, -max(N[i], 0));
    }
    int p = lower_bound(X2.begin(), X2.end(), X[i]) - X2.begin();
    __int128_t y = (__int128_t) ST_A[p] * X[i] + ST_B[p];
    long long l = ST_L[p];
    long long r = ST_R[p];
    if (y == 0 && l == 0 && r == 0){
      cout << "Yes" << endl;
    } else {
      cout << "No" << endl;
    }
  }
}
0