結果
問題 | No.259 セグメントフィッシング+ |
ユーザー | Mao-beta |
提出日時 | 2024-03-21 02:04:36 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 2,749 bytes |
コンパイル時間 | 218 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 70,016 KB |
最終ジャッジ日時 | 2024-09-30 09:57:48 |
合計ジャッジ時間 | 4,197 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | RE | - |
testcase_01 | RE | - |
testcase_02 | RE | - |
testcase_03 | RE | - |
testcase_04 | RE | - |
testcase_05 | RE | - |
testcase_06 | RE | - |
testcase_07 | RE | - |
testcase_08 | RE | - |
testcase_09 | RE | - |
testcase_10 | RE | - |
testcase_11 | RE | - |
testcase_12 | RE | - |
testcase_13 | RE | - |
testcase_14 | RE | - |
testcase_15 | RE | - |
testcase_16 | RE | - |
testcase_17 | RE | - |
testcase_18 | RE | - |
testcase_19 | RE | - |
testcase_20 | RE | - |
testcase_21 | RE | - |
testcase_22 | RE | - |
testcase_23 | RE | - |
testcase_24 | RE | - |
testcase_25 | RE | - |
ソースコード
import sys import math import bisect from heapq import heapify, heappop, heappush from collections import deque, defaultdict, Counter from functools import lru_cache from itertools import accumulate, combinations, permutations, product sys.setrecursionlimit(1000000) MOD = 10 ** 9 + 7 MOD99 = 998244353 input = lambda: sys.stdin.readline().strip() NI = lambda: int(input()) NMI = lambda: map(int, input().split()) NLI = lambda: list(NMI()) SI = lambda: input() SMI = lambda: input().split() SLI = lambda: list(SMI()) EI = lambda m: [NLI() for _ in range(m)] class BIT(): """ BIT 0-index ACL for python add(p, x): p番目にxを加算 get(p): p番目を取得 sum0(r): [0:r)の和を取得 sum(l, r): [l:r)の和を取得 """ def __init__(self, N): self.n = N self.data = [0 for i in range(N)] def add(self, p, x): assert 0 <= p < self.n, "0<=p<n,p={0},n={1}".format(p, self.n) p += 1 while (p <= self.n): self.data[p - 1] += x p += p & -p def get(self, p): return self.sum(p, p + 1) def sum(self, l, r): assert (0 <= l and l <= r and r <= self.n), "0<=l<=r<=n,l={0},r={1},n={2}".format(l, r, self.n) return self.sum0(r) - self.sum0(l) def sum0(self, r): s = 0 while (r > 0): s += self.data[r - 1] r -= r & -r return s def debug(self): res = [self.get(p) for p in range(self.n)] return res def main(): N, Q = NMI() # N-1, N-2, ..., 1, 0, 0, 1, ..., N-1 # 魚はこの中を全部右に動き、端で消えて左から出てくるとする # Rの魚は右側、Lの魚は左側に入るが、そこからt巻き戻した位置に入れる # Cは2箇所作ったうえでtずらす bit = BIT(2*N) for t in range(Q): x, y, z = SMI() y, z = int(y), int(z) if x == "L": p = (N-1-y-t) % (2*N) bit.add(p, z) elif x == "R": p = (y+N-t) % (2*N) bit.add(p, z) else: ans = 0 ry, rz = N+y-t, N+z-t ly, lz = N-z-t, N-y-t # print(t, ly, lz, ry, rz) if ry // (2*N) == rz // (2*N): ans += bit.sum(ry % (2*N), rz % (2*N)) else: ans += bit.sum(0, rz % (2 * N)) ans += bit.sum(ry % (2 * N), 2*N) if ly // (2*N) == lz // (2*N): ans += bit.sum(ly % (2*N), lz % (2*N)) else: ans += bit.sum(0, lz % (2 * N)) ans += bit.sum(ly % (2 * N), 2*N) print(ans) # print(x, y, z, bit.debug()) if __name__ == "__main__": main()