結果

問題 No.1226 I hate Robot Arms
ユーザー tyawanmusityawanmusi
提出日時 2020-08-30 02:39:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 968 ms / 2,000 ms
コード長 5,038 bytes
コンパイル時間 578 ms
コンパイル使用メモリ 82,236 KB
実行使用メモリ 120,880 KB
最終ジャッジ日時 2024-06-10 10:07:06
合計ジャッジ時間 20,335 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 41 ms
53,632 KB
testcase_01 AC 43 ms
54,144 KB
testcase_02 AC 450 ms
86,260 KB
testcase_03 AC 476 ms
90,716 KB
testcase_04 AC 557 ms
99,996 KB
testcase_05 AC 537 ms
99,804 KB
testcase_06 AC 869 ms
117,464 KB
testcase_07 AC 448 ms
86,168 KB
testcase_08 AC 355 ms
91,776 KB
testcase_09 AC 647 ms
96,560 KB
testcase_10 AC 306 ms
81,560 KB
testcase_11 AC 568 ms
102,308 KB
testcase_12 AC 425 ms
90,256 KB
testcase_13 AC 422 ms
95,060 KB
testcase_14 AC 703 ms
99,648 KB
testcase_15 AC 287 ms
89,352 KB
testcase_16 AC 833 ms
117,168 KB
testcase_17 AC 448 ms
87,544 KB
testcase_18 AC 383 ms
86,652 KB
testcase_19 AC 577 ms
102,872 KB
testcase_20 AC 570 ms
100,600 KB
testcase_21 AC 617 ms
104,344 KB
testcase_22 AC 859 ms
118,144 KB
testcase_23 AC 881 ms
120,880 KB
testcase_24 AC 893 ms
118,944 KB
testcase_25 AC 911 ms
118,112 KB
testcase_26 AC 881 ms
120,568 KB
testcase_27 AC 928 ms
118,492 KB
testcase_28 AC 968 ms
117,788 KB
testcase_29 AC 908 ms
118,108 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import cos, sin, radians

# 入力高速化
import sys
input = lambda: sys.stdin.readline().rstrip()

# SegmentTree
# 通常のセグ木にも遅延セグ木にもなる優れモノ（当社比）
# "使いこなせていない", "定数倍がやばい" という理由で適宜セグ木の中身を直接変更しています
class SegmentTree:

  def __init__(self, n, p, unit, f, g, h):
    num = 2**((n-1).bit_length())
    seg = [unit]*(num*2)
    self.lazy = [None]*(num*2)
    for i in range(n):
      seg[num+i] = p[i]
    for i in range(num-1, 0, -1):
      seg[i] = f(seg[i << 1], seg[(i << 1)+1])
    self.num = num
    self.seg = seg
    self.unit = unit
    self.flag = False
    self.f = f
    self.g = g
    self.h = h

  def gindex(self, l, r):
    l += self.num
    r += self.num
    lm = (l//(l & -l)) >> 1
    rm = (r//(r & -r)) >> 1
    mm = max(lm, rm)
    r -= 1
    while l < r:
      if r <= rm:
        yield r
      if l <= lm:
        yield l
      l >>= 1
      r >>= 1
    while l:
      if l <= mm:
        yield l
      l >>= 1

  def propagates(self, ids):
    num = self.num
    g = self.g
    h = self.h
    for i in reversed(ids):
      v = self.lazy[i]
      if v is None:
        continue
      # ここ！！！！！！！！！！！！！
      # ここ！！！！！！！！！！！！！

      newv = v

      # ここ！！！！！！！！！！！！！
      # ここ！！！！！！！！！！！！！
      if (i << 1) < num:
        self.lazy[i << 1] = h(self.lazy[i << 1], newv)
        self.lazy[(i << 1)+1] = h(self.lazy[(i << 1)+1], newv)
      self.seg[i << 1] = g(self.seg[i << 1], newv)
      self.seg[(i << 1)+1] = g(self.seg[(i << 1)+1], newv)
      self.lazy[i] = None

  def query(self, l, r):
    f = self.f
    if self.flag:
      *ids, = self.gindex(l, r)
      self.propagates(ids)
    ansl = ansr = self.unit
    l += self.num
    r += self.num-1
    if l == r:
      return self.seg[l]
    while l < r:
      if l & 1:
        ansl = f(ansl, self.seg[l])
        l += 1
      if r & 1 == 0:
        ansr = f(self.seg[r], ansr)
        r -= 1
      l >>= 1
      r >>= 1
    if l == r:
      ansl = f(ansl, self.seg[l])
    return f(ansl, ansr)

  def update1(self, i, x):
    if self.flag:
      *ids, = self.gindex(i, i+1)
      self.propagates(ids)
    i += self.num
    f = self.f
    # 変更済み
    self.seg[i] = (self.seg[i][0]*x,self.seg[i][1]*x)
    # 変更済み
    while i:
      i >>= 1
      self.seg[i] = f(self.seg[i << 1], self.seg[(i << 1)+1])

  def update2(self, l, r, x):
    self.flag = True
    *ids, = self.gindex(l, r)
    self.propagates(ids)
    num = self.num
    f = self.f
    g = self.g
    h = self.h
    l += num
    r += num-1
    if l == r:
      self.seg[l] = g(self.seg[l], x)
      for i in ids:
        self.seg[i] = f(self.seg[i << 1], self.seg[(i << 1)+1])
      return
    while l < r:
      if l & 1:
        if l < num:
          self.lazy[l] = h(self.lazy[l], x)
        self.seg[l] = g(self.seg[l], x)
        l += 1
      if r & 1 == 0:
        if r < num:
          self.lazy[r] = h(self.lazy[r], x)
        self.seg[r] = g(self.seg[r], x)
        r -= 1
      l >>= 1
      r >>= 1
      # 変更済み
      x = x
      # 変更済み
    if l == r:
      self.lazy[l] = h(self.lazy[l], x)
      self.seg[l] = g(self.seg[l], x)
    for i in ids:
      self.seg[i] = f(self.seg[i << 1], self.seg[(i << 1)+1])
  
  def update(self, i, x):
    if type(i) is int:
      self.update1(i, x)
    else:
      self.update2(i[0], i[1], x)

# ベクトルの和
def f(x, y):
  return (x[0]+y[0], x[1]+y[1])
# ベクトルの回転
def g(x, y):
  a = rotate[y]
  return (x[0]*a[0][0] + x[1]*a[0][1], x[0]*a[1][0] + x[1]*a[1][1])
# 伝播部分において、回転行列を掛け合わせるのではなく、角度を足し合わせています
def h(x,y):
  if x is None:
    return y
  return (x+y) % 360

# 入力
n, q = map(int, input().split())

# 回転行列を前計算
num = 2**((n-1).bit_length())
rotate = [None]*360
for i in range(360):
  rotate[i]=(
    (cos(radians(i)), -sin(radians(i))),
    (sin(radians(i)), cos(radians(i)))
  )

# ご自身の遅延セグ木をお使いください
seg = SegmentTree(n, [(1, 0)]*n, (0, 0), f, g, h)

# 角度,長さ を保持
theta = [0]*n
length = [1]*n

for _ in range(q):
  query = tuple(map(int, input().split()))

  # クエリ0 : 区間 [i, n] のベクトルを回転
  # 0-indexd,開区間のため [i-1, n)
  if query[0] == 0:
    _, i, x=query

    seg.update((i-1, n), (x-theta[i-1]) % 360)
    theta[i-1] = x

  # クエリ1 : 腕 i の長さを変更
  # 0-indexdのため i-1 
  elif query[0] == 1:
    _, i, x = query

    seg.update(i-1, x/length[i-1])
    length[i-1] = x
  
  # クエリ2 : 区間 [1,i] のベクトルの和
  # 0-indexd,開区間のため [0, i)
  else:
    _, i = query

    x, y = seg.query(0, i)

    # たまに指数表記しやがるので指定してやります
    print('{:.9f}'.format(x), '{:.9f}'.format(y))
0