結果

問題 No.59 鉄道の旅
コンテスト
ユーザー nebukuro09
提出日時 2016-11-21 16:58:03
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 86 ms / 5,000 ms
コード長 1,744 bytes
コンパイル時間 945 ms
コンパイル使用メモリ 111,632 KB
実行使用メモリ 12,940 KB
最終ジャッジ日時 2024-06-12 05:09:40
合計ジャッジ時間 1,807 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.stdio;
import std.array;
import std.string;
import std.conv;
import std.algorithm;
import std.typecons;
import std.range;
import std.random;
import std.math;
import std.container;
import std.numeric;


class SegmentTree {
  int[] table;
  int N, table_size;

  this(int n) {
    N = n;
    table_size = 1;
    while (table_size < n) table_size *= 2;
    table_size *= 2;
    table = new int[](table_size);
  }

  void add(int pos, int num) {
    add_rec(0, 0, table_size/2-1, pos, num);
  }

  void add_rec(int i, int left, int right, int pos, int num) {
    table[i] += num;
    if (left == right)
      return;
    auto mid = (left + right) / 2;
    if (pos <= mid)
      add_rec(i*2+1, left, mid, pos, num);
    else
      add_rec(i*2+2, mid+1, right, pos, num);
  }

  int sum(int pl, int pr) {
    return sum_rec(0, pl, pr, 0, table_size/2-1);
  }

  int sum_rec(int i, int pl, int pr, int left, int right) {
    if (pl > right || pr < left)
      return 0;
    else if (pl <= left && right <= pr)
      return table[i];
    else 
      return sum_rec(i*2+1, pl, pr, left, (left+right)/2) +
             sum_rec(i*2+2, pl, pr, (left+right)/2+1, right);
  }
}

void main() {
  auto W_MAX = 1000000;
  auto sg = new SegmentTree(W_MAX);
  auto input = readln.split.map!(to!int);
  auto N = input[0];
  auto K = input[1];
  foreach(_; 0..N) {
    auto W = readln.chomp.to!int;
    if (W > 0 && sg.sum(W, W_MAX) < K)
      sg.add(W, 1);
    else if (W < 0 && sg.sum(-W, -W) > 0)
      sg.add(-W, -1);
  }
  writeln(sg.sum(0, W_MAX));
}


void test_sg() {
  auto sg = new SegmentTree(100);
  sg.add(50, 20);
  sg.add(60, 3);
  writeln(sg.sum(0, 100));
  writeln(sg.sum(0, 40));
  writeln(sg.sum(50, 60));
  writeln(sg.sum(51, 60));
}
0