結果

問題 No.1226 I hate Robot Arms
ユーザー Kiri8128Kiri8128
提出日時 2020-09-13 11:30:38
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,557 ms / 2,000 ms
コード長 4,860 bytes
コンパイル時間 362 ms
コンパイル使用メモリ 86,964 KB
実行使用メモリ 126,892 KB
最終ジャッジ日時 2023-09-02 13:29:33
合計ジャッジ時間 40,749 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 74 ms
71,928 KB
testcase_01 AC 77 ms
71,920 KB
testcase_02 AC 732 ms
92,496 KB
testcase_03 AC 856 ms
96,444 KB
testcase_04 AC 948 ms
108,092 KB
testcase_05 AC 888 ms
105,776 KB
testcase_06 AC 1,511 ms
125,368 KB
testcase_07 AC 705 ms
92,812 KB
testcase_08 AC 559 ms
100,028 KB
testcase_09 AC 1,022 ms
105,272 KB
testcase_10 AC 481 ms
85,936 KB
testcase_11 AC 1,002 ms
108,908 KB
testcase_12 AC 773 ms
96,668 KB
testcase_13 AC 644 ms
100,708 KB
testcase_14 AC 1,184 ms
108,416 KB
testcase_15 AC 445 ms
99,984 KB
testcase_16 AC 1,462 ms
124,028 KB
testcase_17 AC 737 ms
94,172 KB
testcase_18 AC 584 ms
90,068 KB
testcase_19 AC 935 ms
109,204 KB
testcase_20 AC 1,013 ms
109,236 KB
testcase_21 AC 1,147 ms
112,472 KB
testcase_22 AC 1,524 ms
126,280 KB
testcase_23 AC 1,492 ms
125,624 KB
testcase_24 AC 1,481 ms
125,708 KB
testcase_25 AC 1,557 ms
126,892 KB
testcase_26 AC 1,527 ms
126,288 KB
testcase_27 AC 1,427 ms
124,196 KB
testcase_28 AC 1,409 ms
123,944 KB
testcase_29 AC 1,380 ms
125,660 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda: sys.stdin.readline().rstrip()
class SegmentTree():
    def __init__(self, init, unitX, unitA, f, g, h):
        self.f = f # (X, X) -> X
        self.g = g # (X, A, size) -> X
        self.h = h # (A, A) -> A
        self.unitX = unitX
        self.unitA = unitA
        self.f = f
        if type(init) == int:
            self.n = init
            self.n = 1 << (self.n - 1).bit_length()
            self.X = [unitX] * (self.n * 2)
            self.size = [1] * (self.n * 2)
        else:
            self.n = len(init)
            self.n = 1 << (self.n - 1).bit_length()
            self.X = [unitX] * self.n + init + [unitX] * (self.n - len(init))
            self.size = [0] * self.n + [1] * len(init) + [0] * (self.n - len(init))
            for i in range(self.n-1, 0, -1):
                self.X[i] = self.f(self.X[i*2], self.X[i*2|1])
    
        for i in range(self.n - 1, 0, -1):
            self.size[i] = self.size[i*2] + self.size[i*2|1]
        
        self.A = [unitA] * (self.n * 2)
        
    def update(self, i, x):
        i += self.n
        self.propagate_above(i)
        self.X[i] = x
        self.A[i] = unitA
        self.calc_above(i)
    
    def calc(self, i):
        return self.g(self.X[i], self.A[i], self.size[i])
    
    def calc_above(self, i):
        i >>= 1
        while i:
            self.X[i] = self.f(self.calc(i*2), self.calc(i*2|1))
            i >>= 1
    
    def propagate(self, i):
        self.X[i] = self.g(self.X[i], self.A[i], self.size[i])
        self.A[i*2] = self.h(self.A[i*2], self.A[i])
        self.A[i*2|1] = self.h(self.A[i*2|1], self.A[i])
        self.A[i] = self.unitA
        
    def propagate_above(self, i):
        H = i.bit_length()
        for h in range(H, 0, -1):
            self.propagate(i >> h)
    
    def propagate_all(self):
        for i in range(1, self.n):
            self.propagate(i)
    
    def getrange(self, l, r):
        l += self.n
        r += self.n
        l0, r0 = l // (l & -l), r // (r & -r) - 1
        self.propagate_above(l0)
        self.propagate_above(r0)
        
        al = self.unitX
        ar = self.unitX
        while l < r:
            if l & 1:
                al = self.f(al, self.calc(l))
                l += 1
            if r & 1:
                r -= 1
                ar = self.f(self.calc(r), ar)
            l >>= 1
            r >>= 1
        return self.f(al, ar)
    
    def getvalue(self, i):
        i += self.n
        self.propagate_above(i)
        return self.calc(i)
    
    def operate_range(self, l, r, a):
        l += self.n
        r += self.n
        l0, r0 = l // (l & -l), r // (r & -r) - 1
        self.propagate_above(l0)
        self.propagate_above(r0)
        while l < r:
            if l & 1:
                self.A[l] = self.h(self.A[l], a)
                l += 1
            if r & 1:
                r -= 1
                self.A[r] = self.h(self.A[r], a)
            l >>= 1
            r >>= 1
        
        self.calc_above(l0)
        self.calc_above(r0)
    
    def debug(self):
        print("self.n =", self.n)
        deX = []
        deA = []
        deS = []
        a, b = self.n, self.n * 2
        while b:
            deX.append(self.X[a:b])
            deA.append(self.A[a:b])
            deS.append(self.size[a:b])
            a, b = a//2, a
        print("--- debug ---")
        for d in deX[::-1]:
            print(d)
        print("--- ---")
        for d in deA[::-1]:
            print(d)
        print("--- ---")
        for d in deS[::-1]:
            print(d)
        print("--- ---")
    
    def debug(self):
        for i in range(min(self.n - 1, 10)):
            print(i, self.getvalue(i))
        for i in range(min(self.n, 11)):
            print(i, self.getrange(0, i))

from math import tan, sin, cos, pi, sqrt, atan2
f = lambda x, y: (x[0] + y[0], x[1] + y[1])
g = lambda x, a, s: (a[0] * x[0] - a[1] * x[1], a[1] * x[0] + a[0] * x[1])
h = lambda a, b: (a[0] * b[0] - a[1] * b[1], a[0] * b[1] + a[1] * b[0])

unitA = (1, 0) # * (a + bi), + (c + di)
unitX = (0, 0) # x, y
z = pi / 180
N, Q = map(int, input().split())
st = SegmentTree([(1, 0) for _ in range(N)], unitX, unitA, f, g, h)
for _ in range(Q):
    q = [int(a) for a in input().split()]
    if q[0] == 0:
        i, s = q[1:]
        s *= z
        i -= 1
        x, y = st.getvalue(i)
        r = sqrt(x**2 + y**2)
        t = atan2(y, x)
        if i == 0:
            u = 0
        else:
            x2, y2 = st.getvalue(i-1)
            u = atan2(y2, x2)
        st.operate_range(i, N, (cos((s - t + u)), sin((s - t + u))))
    elif q[0] == 1:
        i, s = q[1:]
        i -= 1
        x, y = st.getvalue(i)
        r = sqrt(x**2 + y**2)
        x *= s / r
        y *= s / r
        st.update(i, (x, y))
    else:
        i = q[1]
        print(*st.getrange(0, i))
0