結果

問題 No.1746 Sqrt Integer Segments
ユーザー tktk_snsntktk_snsn
提出日時 2022-06-12 22:23:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 957 ms / 2,000 ms
コード長 1,074 bytes
コンパイル時間 440 ms
コンパイル使用メモリ 81,840 KB
実行使用メモリ 261,272 KB
最終ジャッジ日時 2023-10-24 17:01:24
合計ジャッジ時間 26,045 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 602 ms
138,596 KB
testcase_01 AC 625 ms
138,844 KB
testcase_02 AC 783 ms
178,476 KB
testcase_03 AC 811 ms
215,512 KB
testcase_04 AC 771 ms
167,668 KB
testcase_05 AC 874 ms
201,368 KB
testcase_06 AC 757 ms
156,328 KB
testcase_07 AC 656 ms
141,968 KB
testcase_08 AC 798 ms
178,648 KB
testcase_09 AC 869 ms
200,828 KB
testcase_10 AC 744 ms
211,748 KB
testcase_11 AC 855 ms
191,124 KB
testcase_12 AC 951 ms
261,272 KB
testcase_13 AC 943 ms
201,680 KB
testcase_14 AC 887 ms
201,540 KB
testcase_15 AC 957 ms
255,984 KB
testcase_16 AC 913 ms
201,224 KB
testcase_17 AC 836 ms
236,496 KB
testcase_18 AC 668 ms
141,424 KB
testcase_19 AC 675 ms
142,996 KB
testcase_20 AC 710 ms
142,364 KB
testcase_21 AC 730 ms
143,148 KB
testcase_22 AC 699 ms
143,424 KB
testcase_23 AC 666 ms
140,624 KB
testcase_24 AC 820 ms
150,344 KB
testcase_25 AC 845 ms
149,524 KB
testcase_26 AC 761 ms
148,292 KB
testcase_27 AC 821 ms
154,724 KB
testcase_28 AC 829 ms
154,412 KB
testcase_29 AC 774 ms
153,960 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import Counter
from random import randint
U = 10 ** 6
mask = (1 << 64) - 1


def prime_sieve(N):
    #sieve[i] : iの最小の素因数
    sieve = [0] * (N + 1)
    prime = []
    for i in range(2, N + 1):
        if sieve[i] == 0:
            sieve[i] = i
            prime.append(i)
        for p in prime:
            if p > sieve[i] or i * p > N:
                break
            sieve[i * p] = p
    return sieve


def xorshift64(seed=123456789):
    x = seed
    while 1:
        x = x ^ (x << 13) & mask
        x = x ^ (x >> 7) & mask
        x = x ^ (x << 17) & mask
        yield x


def main(N, A):
    # rnd = xorshift64()
    h = [randint(0, mask) for _ in range(U+1)]
    sieve = prime_sieve(U)

    xor = [0] * (N + 1)
    for i, a in enumerate(A, 1):
        while a > 1:
            p = sieve[a]
            xor[i] ^= h[p]
            a //= p
        xor[i] ^= xor[i-1]

    ans = 0
    for v in Counter(xor).values():
        ans += v * (v - 1) // 2
    return ans


N = int(input())
A = list(map(int, input().split()))
print(main(N, A))
0