update binary_indexed_tree

tree/binary_indexed_tree.cpp

@@ -37,6 +37,11 @@ template<typename T> struct BIT {
 		return sum(r) - sum(max(l - 1, 0));
 	}
 
+	inline T get(int idx) {
+		assert(0 < idx and idx <= N);
+		return sum(idx, idx);
+	}
+
 	void add(int idx, T val) {
 		assert(0 < idx && idx <= N);
 		while(idx <= N) {

tree/docs/binary_indexed_tree.md

@@ -12,9 +12,10 @@ documentation_of: ../binary_indexed_tree.cpp
 - `BIT(v)`：配列 `v` の要素で BIT を構築する．
 
 ## メンバ関数
-- `sum(idx)`：`idx` 番目までの要素の総和を返す．
-- `sum(l, r)`：`l` 番目から `r` 番目までの要素の総和を返す．
-- `add(idx, val)`：`idx` 番目の要素に `val` を加える．
+- `sum(idx)`：$[1, \mathrm{idx}]$ の要素の総和を返す．
+- `sum(l, r)`：$[l, r]$ の要素の総和を返す．
+- `get(idx)`：$\mathrm{idx}$ 番目の要素の値を返す．
+- `add(idx, val)`：$\mathrm{idx}$ 番目の要素に `val` を加える．
 
 `idx`, `l`, `r` は 1-indexed である．
 
@@ -23,4 +24,5 @@ documentation_of: ../binary_indexed_tree.cpp
 - `BIT(v)`：$O(n)$
 - `sum(idx)`：$O(\log n)$
 - `sum(l, r)`：$O(\log n)$
+- `get(idx)`：$O(\log n)$
 - `add(idx, val)`：$O(\log n)$