結果

問題 No.1013 〇マス進む
ユーザー NoneNone
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0