結果
問題 | No.789 範囲の合計 |
ユーザー | InTheBloom |
提出日時 | 2024-04-30 02:49:21 |
言語 | D (dmd 2.106.1) |
結果 |
AC
|
実行時間 | 168 ms / 1,000 ms |
コード長 | 5,720 bytes |
コンパイル時間 | 4,715 ms |
コンパイル使用メモリ | 209,792 KB |
実行使用メモリ | 13,312 KB |
最終ジャッジ日時 | 2024-04-30 02:49:35 |
合計ジャッジ時間 | 7,597 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 1 ms
5,376 KB |
testcase_02 | AC | 153 ms
7,808 KB |
testcase_03 | AC | 115 ms
5,376 KB |
testcase_04 | AC | 168 ms
13,312 KB |
testcase_05 | AC | 146 ms
7,884 KB |
testcase_06 | AC | 142 ms
7,808 KB |
testcase_07 | AC | 94 ms
5,376 KB |
testcase_08 | AC | 136 ms
8,576 KB |
testcase_09 | AC | 127 ms
12,544 KB |
testcase_10 | AC | 152 ms
7,844 KB |
testcase_11 | AC | 112 ms
7,808 KB |
testcase_12 | AC | 112 ms
7,936 KB |
testcase_13 | AC | 1 ms
5,376 KB |
testcase_14 | AC | 1 ms
5,376 KB |
ソースコード
void read (T...) (string S, ref T args) { import std.conv : to; import std.array : split; auto buf = S.split; foreach (i, ref arg; args) { arg = buf[i].to!(typeof(arg)); } } void main () { // yosupojudge_Point_Add_Range_Sum(); yukicoder_789(); } void yosupojudge_Point_Add_Range_Sum () { import std; // yosupo judge: Point Add Range Sum (https://judge.yosupo.jp/problem/point_add_range_sum) long len = 10^^9; auto seg = new DynamicSegmentTree!(long, (long a, long b) => a + b, () => 0L)(len); int N, Q; readln.read(N, Q); auto a = readln.split.to!(int[]); foreach (i; 0..N) seg.set(i, a[i]); // seg.dump(); // writeln(); foreach (i; 0..Q) { int t, l, r; readln.read(t, l, r); if (t == 0) { seg.set(l, seg.get(l) + r); // seg.dump(); // writeln(); } if (t == 1) { writeln(seg.prod(l, r)); } } } void yukicoder_789 () { import std; // yukicoder No.789 範囲の合計 long len = 10^^9; auto seg = new DynamicSegmentTree!(long, (long a, long b) => a + b, () => 0L)(len); int N = readln.chomp.to!int; // seg.dump(); // writeln(); long ans = 0; foreach (i; 0..N) { int t, l, r; readln.read(t, l, r); if (t == 0) { seg.set(l, seg.get(l) + r); } if (t == 1) { ans += seg.prod(l, r + 1); } } writeln(ans); } import std.traits : ReturnType, isCallable, Parameters; import std.meta : AliasSeq; class DynamicSegmentTree (T, alias op, alias e) { // TODO: assertのメッセージを表示 static assert(isCallable!(op)); static assert(isCallable!(e)); static assert(is (ReturnType!(op) == T)); static assert(is (ReturnType!(e) == T)); static assert(is (Parameters!(op) == AliasSeq!(T, T))); static assert(is (Parameters!(e) == AliasSeq!())); // 内部が1-indexedで動的な完全二分セグメント木 import std.format : format; public: this (long N_) in { assert(1 <= N_, format("Dynamic SegmentTree: N = %s does not satisfy constraints. N must be in range of [1, %s]", 4 * 10L^^18)); } do { length = N_; // N_以上の2冪に設定 if ((N_ & (-N_)) == N_) { N = N_; } else { // msb+1 bitを立てる foreach_reverse (i; 0..63) { if (0 < (N_ & (1L << i))) { N = 1L << (i + 1); break; } } } } void set (long idx, T val) { idx++; internal_set(root, idx, val, 1, N + 1); } T get (long idx) { idx++; return internal_get(root, idx, 1, N + 1); } T prod (long l, long r) { l++, r++; return internal_prod(root, l, r, 1, N + 1); } void dump () { dfs(root); } private: struct node { long index; T value, product; node *left = null, right = null; } void node_update (node *n) { n.product = op( op((n.left == null ? e() : n.left.product), n.value), (n.right == null ? e() : n.right.product) ); } node *root = null; long N = 0; long length = 0; void dfs (const node *cur) { if (cur == null) return; dfs(cur.left); dfs(cur.right); import std.stdio; writeln("index: ", cur.index - 1, " value: ", cur.value, " product: ", cur.product); } // [l, r) : 今見ている部分木が管理する範囲 node *internal_set (ref node *cur, long idx, T val, long l, long r) { if (cur == null) { return cur = new node(idx, val, val, null, null); } if (cur.index == idx) { cur.value = val; node_update(cur); return cur; } // 既に部分木管理ノードが存在するときの処理 import std.algorithm : swap; long mid = (l + r) / 2; if (idx < mid) { // 今いる人を押しのける if (cur.index < idx) { swap(cur.value, val); swap(cur.index, idx); } cur.left = internal_set(cur.left, idx, val, l, mid); } else { if (idx < cur.index) { swap(cur.value, val); swap(cur.index, idx); } cur.right = internal_set(cur.right, idx, val, mid, r); } node_update(cur); return cur; } T internal_get (const node *cur, long idx, long l, long r) { if (cur == null) return e(); if (cur.index == idx) return cur.value; long mid = (l + r) / 2; if (idx < mid) return internal_get(cur.left, idx, l, mid); return internal_get(cur.right, idx, mid, r); } // [a, b) = 要求区間 T internal_prod (const node *cur, long a, long b, long l, long r) { if (cur == null || b <= l || r <= a) return e(); if (a <= l && r <= b) return cur.product; long mid = (l + r) / 2; T res = internal_prod(cur.left, a, b, l, mid); if (a <= cur.index && cur.index < b) res = op(res, cur.value); res = op(res, internal_prod(cur.right, a, b, mid, r)); return res; } }