diff --git a/atcoder/dsu.py b/atcoder/dsu.py index ad73361..18ee814 100644 --- a/atcoder/dsu.py +++ b/atcoder/dsu.py @@ -59,25 +59,3 @@ def groups(self) -> typing.List[typing.List[int]]: result[leader_buf[i]].append(i) return list(filter(lambda r: r, result)) - - -# https://atcoder.jp/contests/practice2/tasks/practice2_a -def main() -> None: - import sys - - n, q = map(int, sys.stdin.readline().split()) - dsu = DSU(n) - - for _ in range(q): - t, u, v = map(int, sys.stdin.readline().split()) - if t == 0: - dsu.merge(u, v) - if t == 1: - if dsu.same(u, v): - print(1) - else: - print(0) - - -if __name__ == '__main__': - main() diff --git a/atcoder/fenwicktree.py b/atcoder/fenwicktree.py index c5250a9..a49e4e2 100644 --- a/atcoder/fenwicktree.py +++ b/atcoder/fenwicktree.py @@ -28,26 +28,3 @@ def _sum(self, r: int) -> typing.Any: r -= r & -r return s - - -# https://atcoder.jp/contests/practice2/tasks/practice2_b -def main() -> None: - import sys - - n, q = map(int, sys.stdin.readline().split()) - fenwick_tree = FenwickTree(n) - - a = map(int, sys.stdin.readline().split()) - for i, ai in enumerate(a): - fenwick_tree.add(i, ai) - - for _ in range(q): - t, x, y = map(int, sys.stdin.readline().split()) - if t == 0: - fenwick_tree.add(x, y) - if t == 1: - print(fenwick_tree.sum(x, y)) - - -if __name__ == '__main__': - main() diff --git a/example/dsu_practice.py b/example/dsu_practice.py new file mode 100644 index 0000000..069b6d2 --- /dev/null +++ b/example/dsu_practice.py @@ -0,0 +1,24 @@ +# https://atcoder.jp/contests/practice2/tasks/practice2_a + +from atcoder.dsu import DSU + + +def main() -> None: + import sys + + n, q = map(int, sys.stdin.readline().split()) + dsu = DSU(n) + + for _ in range(q): + t, u, v = map(int, sys.stdin.readline().split()) + if t == 0: + dsu.merge(u, v) + if t == 1: + if dsu.same(u, v): + print(1) + else: + print(0) + + +if __name__ == '__main__': + main() diff --git a/example/fenwick_practice.py b/example/fenwick_practice.py new file mode 100644 index 0000000..263a454 --- /dev/null +++ b/example/fenwick_practice.py @@ -0,0 +1,25 @@ +# https://atcoder.jp/contests/practice2/tasks/practice2_b + +from atcoder.fenwicktree import FenwickTree + + +def main() -> None: + import sys + + n, q = map(int, sys.stdin.readline().split()) + fenwick_tree = FenwickTree(n) + + a = map(int, sys.stdin.readline().split()) + for i, ai in enumerate(a): + fenwick_tree.add(i, ai) + + for _ in range(q): + t, x, y = map(int, sys.stdin.readline().split()) + if t == 0: + fenwick_tree.add(x, y) + if t == 1: + print(fenwick_tree.sum(x, y)) + + +if __name__ == '__main__': + main()