結果

問題 No.2698 Many Asakatsu
ユーザー SSRSSSRS
提出日時 2024-03-22 23:12:11
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,832 bytes
コンパイル時間 1,891 ms
コンパイル使用メモリ 207,620 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-09-30 12:27:10
合計ジャッジ時間 4,633 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 1 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 1 ms
5,248 KB
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
const long long INF = 1000000000000000000;
int main(){
  int N, M;
  cin >> N >> M;
  vector<long long> A(N), B(N);
  for (int i = 0; i < N; i++){
    cin >> A[i] >> B[i];
  }
  int Q;
  long long X;
  cin >> Q >> X;
  vector<int> C(N);
  for (int i = 0; i < N; i++){
    cin >> C[i];
  }
  int p = 0;
  __int128_t x = X;
  auto d = [&](int i, long long R){
    if (R <= A[i]){
      return (A[i] - R) * M + B[i] * M * (M - 1) / 2;
    } else if (A[i] + B[i] * (M - 1) <= R){
      return R * M - (A[i] * 2 + B[i] * (M - 1)) * M / 2;
    } else {
      int p = (R - A[i]) / B[i];
      long long ans = 0;
      ans += (R * 2 - A[i] * 2 - B[i] * p) * (p + 1) / 2;
      ans += (A[i] * 2 + B[i] * (M + p) - R * 2) * (M - p - 1) / 2;
      return ans;
    }
  };
  struct node{
    int l, r;
    int idx;
    node* lch;
    node* rch;
    node(int l, int r, int idx): l(l), r(r), idx(idx), lch(nullptr), rch(nullptr){
    }
  };
  node* root = new node(0, 1000000000, 0);
  auto line_add = [&](auto line_add, int idx, node* v){
    long long la = d(v->idx, v->l);
    long long lb = d(idx, v->l);
    long long ra = d(v->idx, v->r);
    long long rb = d(idx, v->r);
    if (make_pair(la, v->idx) >= make_pair(lb, idx) && make_pair(ra, v->idx) >= make_pair(rb, idx)){
      return;
    } else if (make_pair(la, v->idx) <= make_pair(lb, idx) && make_pair(ra, v->idx) <= make_pair(rb, idx)){
      v->idx = idx;
    } else {
      int m = (v->l + v->r) / 2;
      long long ma = d(v->idx, m);
      long long mb = d(idx, m);
      if (make_pair(ma, v->idx) < make_pair(mb, idx)){
        swap(v->idx, idx);
        swap(la, lb);
        swap(ra, rb);
      }
      if (make_pair(la, v->idx) < make_pair(lb, idx)){
        if (v->lch == nullptr){
          v->lch = new node(v->l, m, idx);
        } else {
          line_add(line_add, idx, v->lch);
        }
      }
      if (make_pair(ra, v->idx) < make_pair(rb, idx)){
        if (v->rch == nullptr){
          v->rch = new node(m + 1, v->r, idx);
        } else {
          line_add(line_add, idx, v->rch);
        }
      }
    }
  };
  auto get_min = [&](int x){
    int idx = -1;
    node* v = root;
    while (true){
      if (v == nullptr){
        break;
      }
      if (idx == -1){
        idx = v->idx;
      } else if (make_pair(d(v->idx, x), v->idx) < make_pair(d(idx, x), idx)){
        idx = v->idx;
      }
      if (x <= (v->l + v->r) / 2){
        v = v->lch;
      } else {
        v = v->rch;
      }
    }
    return idx;
  };
  for (int i = 1; i < N; i++){
    line_add(line_add, i, root);
  }
  for (int i = 0; i < Q; i++){
    if (X >= 1000000000){
      cout << N;
    } else {
      int p = get_min(X);
      cout << p + 1;
      X += C[p];
    }
    if (i < Q - 1){
      cout << ' ';
    }
  }
  cout << endl;
}
0