結果

問題 No.1746 Sqrt Integer Segments
ユーザー shotoyooshotoyoo
提出日時 2021-11-18 20:39:54
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 670 ms / 2,000 ms
コード長 1,936 bytes
コンパイル時間 172 ms
コンパイル使用メモリ 82,044 KB
実行使用メモリ 148,292 KB
最終ジャッジ日時 2024-09-13 09:26:22
合計ジャッジ時間 14,699 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 176 ms
101,952 KB
testcase_01 AC 178 ms
102,336 KB
testcase_02 AC 522 ms
128,392 KB
testcase_03 AC 419 ms
114,604 KB
testcase_04 AC 466 ms
119,068 KB
testcase_05 AC 668 ms
147,740 KB
testcase_06 AC 373 ms
113,664 KB
testcase_07 AC 288 ms
106,676 KB
testcase_08 AC 517 ms
129,204 KB
testcase_09 AC 640 ms
144,932 KB
testcase_10 AC 349 ms
112,828 KB
testcase_11 AC 634 ms
143,824 KB
testcase_12 AC 635 ms
146,144 KB
testcase_13 AC 670 ms
146,736 KB
testcase_14 AC 651 ms
145,896 KB
testcase_15 AC 660 ms
148,292 KB
testcase_16 AC 656 ms
147,840 KB
testcase_17 AC 474 ms
120,064 KB
testcase_18 AC 237 ms
123,904 KB
testcase_19 AC 240 ms
124,680 KB
testcase_20 AC 236 ms
124,288 KB
testcase_21 AC 277 ms
116,540 KB
testcase_22 AC 282 ms
116,668 KB
testcase_23 AC 242 ms
109,116 KB
testcase_24 AC 372 ms
129,004 KB
testcase_25 AC 376 ms
129,336 KB
testcase_26 AC 379 ms
129,128 KB
testcase_27 AC 466 ms
129,680 KB
testcase_28 AC 467 ms
129,296 KB
testcase_29 AC 473 ms
129,420 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

write = lambda x: sys.stdout.write(x+"\n"); writef = lambda x: print("{:.12f}".format(x))
debug = lambda x: sys.stderr.write(x+"\n")
YES="Yes"; NO="No"; pans = lambda v: print(YES if v else NO)
LI = lambda : list(map(int, input().split()))
# sys.setrecursionlimit(3*10**5+10)

def hurui(n):
    """線形篩
    pl: 素数のリスト
    mpf: iを割り切る最小の素因数
    """
    pl = []
    mpf = [None]*(n+1)
    for d in range(2,n+1):
        if mpf[d] is None:
            mpf[d] = d
            pl.append(d)
        for p in pl:
            if p*d>n or p>mpf[d]:
                break
            mpf[p*d] = p
    return pl, mpf
from collections import defaultdict
def factor(num):
    d = defaultdict(int)
    if num==1:
        d.update({1:1})
        return d
    while num>1:
        d[mpf[num]] += 1
        num //= mpf[num]
    return d
def fs(num):
    f = factor(num)
    ans = [1]
    for k,v in f.items():
        tmp = []
        for i in range(len(ans)):
            val = 1
            for _ in range(v):
                val *= k
                ans.append(ans[i]*val)
    return ans
pl, mpf = hurui(10**6+10)
n = int(input())
a = list(map(int, input().split()))
M = 10**9+7
# M = (1<<63) - 1
M = 72057594037927931
# M = 92709568269121
vs = {}
d = {}
d[0] = 1
val = 0
s = set()
ans = 0
import random
b = random.randint(123,M-2)
def sub(k):
    if k in vs:
        return vs[k]
    vs[k] = random.randint(1,M-1)
#     vs[k] = pow(b, k, M)
    return vs[k]
for i in range(n):
    f = factor(a[i])
    for k,v in f.items():
        if k==1:
            continue
        if v%2==1:
            tmp = sub(k)
            if k in s:
                s.remove(k)
                val ^= tmp
            else:
                s.add(k)
                val ^= tmp
            val %= M
    ans += d.get(val, 0)
    d.setdefault(val, 0)
    d[val] += 1
print(ans)
0