結果
問題 | No.1013 〇マス進む |
ユーザー | None |
提出日時 | 2021-10-29 14:42:46 |
言語 | PyPy3 (7.3.13) |
結果 |
AC
|
実行時間 | 592 ms / 2,000 ms |
コード長 | 1,892 bytes |
コンパイル時間 | 574 ms |
コンパイル使用メモリ | 86,932 KB |
実行使用メモリ | 256,848 KB |
最終ジャッジ日時 | 2023-07-29 06:14:46 |
合計ジャッジ時間 | 25,068 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 65 ms
71,600 KB |
testcase_01 | AC | 67 ms
71,120 KB |
testcase_02 | AC | 60 ms
71,276 KB |
testcase_03 | AC | 67 ms
76,092 KB |
testcase_04 | AC | 64 ms
75,904 KB |
testcase_05 | AC | 70 ms
76,100 KB |
testcase_06 | AC | 70 ms
76,068 KB |
testcase_07 | AC | 72 ms
76,148 KB |
testcase_08 | AC | 70 ms
75,788 KB |
testcase_09 | AC | 72 ms
76,276 KB |
testcase_10 | AC | 71 ms
76,144 KB |
testcase_11 | AC | 74 ms
75,852 KB |
testcase_12 | AC | 74 ms
76,380 KB |
testcase_13 | AC | 89 ms
81,652 KB |
testcase_14 | AC | 260 ms
153,976 KB |
testcase_15 | AC | 538 ms
208,024 KB |
testcase_16 | AC | 581 ms
195,644 KB |
testcase_17 | AC | 230 ms
141,168 KB |
testcase_18 | AC | 276 ms
158,344 KB |
testcase_19 | AC | 184 ms
122,320 KB |
testcase_20 | AC | 485 ms
234,660 KB |
testcase_21 | AC | 331 ms
138,784 KB |
testcase_22 | AC | 279 ms
159,972 KB |
testcase_23 | AC | 135 ms
103,996 KB |
testcase_24 | AC | 493 ms
240,440 KB |
testcase_25 | AC | 470 ms
237,408 KB |
testcase_26 | AC | 131 ms
104,788 KB |
testcase_27 | AC | 204 ms
135,384 KB |
testcase_28 | AC | 130 ms
102,884 KB |
testcase_29 | AC | 453 ms
231,304 KB |
testcase_30 | AC | 179 ms
123,936 KB |
testcase_31 | AC | 306 ms
174,900 KB |
testcase_32 | AC | 232 ms
150,080 KB |
testcase_33 | AC | 243 ms
152,384 KB |
testcase_34 | AC | 385 ms
203,244 KB |
testcase_35 | AC | 377 ms
196,152 KB |
testcase_36 | AC | 93 ms
82,604 KB |
testcase_37 | AC | 93 ms
82,528 KB |
testcase_38 | AC | 412 ms
214,760 KB |
testcase_39 | AC | 170 ms
119,012 KB |
testcase_40 | AC | 387 ms
204,756 KB |
testcase_41 | AC | 137 ms
105,612 KB |
testcase_42 | AC | 225 ms
145,756 KB |
testcase_43 | AC | 172 ms
121,940 KB |
testcase_44 | AC | 243 ms
154,224 KB |
testcase_45 | AC | 218 ms
143,144 KB |
testcase_46 | AC | 139 ms
106,836 KB |
testcase_47 | AC | 227 ms
144,908 KB |
testcase_48 | AC | 273 ms
160,840 KB |
testcase_49 | AC | 381 ms
205,312 KB |
testcase_50 | AC | 323 ms
180,120 KB |
testcase_51 | AC | 296 ms
168,832 KB |
testcase_52 | AC | 113 ms
92,144 KB |
testcase_53 | AC | 121 ms
96,620 KB |
testcase_54 | AC | 382 ms
198,332 KB |
testcase_55 | AC | 145 ms
107,784 KB |
testcase_56 | AC | 489 ms
234,444 KB |
testcase_57 | AC | 354 ms
176,932 KB |
testcase_58 | AC | 508 ms
252,004 KB |
testcase_59 | AC | 592 ms
253,332 KB |
testcase_60 | AC | 572 ms
256,848 KB |
testcase_61 | AC | 495 ms
244,028 KB |
testcase_62 | AC | 484 ms
240,788 KB |
testcase_63 | AC | 63 ms
71,288 KB |
testcase_64 | AC | 65 ms
71,440 KB |
ソースコード
class Doubling: def __init__(self, A, K_max=10**18 + 1): """ :param A: 初期条件(写像 A:i->j を定義) :param K_max: K_max まで参照する可能性がある doubling[i][k] := dp[2^i][k] ダブリングを考える際、初期値は[0,1,2,....,N-1] (=dp[i][k] の添え字 k)と考えるべき """ self.k_max = K_max.bit_length() self.n = len(A) self.doubling = [[-1] * self.n for _ in range(self.k_max)] self.doubling[0]=A for i in range(1, self.k_max): self.doubling[i] = self.merge(self.doubling[i-1],self.doubling[i-1]) def merge(self,dp_i,dp_j): """ :return dp[i+j] = merge(dp[i],dp[j]) """ dp_ij=[0]*self.n for k in range(self.n): if dp_j[k]!=-1: dp_ij[k]=dp_i[(dp_j[k]+k)%N] + dp_j[k] return dp_ij def apply_all(self, K): """ :param K: K回進む 0~N-1の全ての要素を同時にK回進める ダブリングを考える際、初期値は[0,1,2,....,N-1] (=dp[i][k] の添え字 k)と考えるべき """ res=P for k in range(K.bit_length()): if K&(1<<k): res = self.merge(self.doubling[k],res) return res ################################################################################ def example(): global input example=iter( """ 5 2 1 2 3 4 5 """ .strip().split("\n")) input=lambda:next(example) """ 1->3->4->1->3->4 """ ################################################################################ import sys input = sys.stdin.readline # example() N, K = map(int, input().split()) P = list(map(int, input().split())) doubling = Doubling(P, K_max = 10**18 + 1) res=doubling.apply_all(K-1) for i,r in enumerate(res): print(i+r+1)