結果

問題 No.2075 GCD Subsequence
ユーザー taiga0629kyoprotaiga0629kyopro
提出日時 2022-09-16 18:15:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,240 ms / 4,000 ms
コード長 3,456 bytes
コンパイル時間 163 ms
コンパイル使用メモリ 82,284 KB
実行使用メモリ 158,188 KB
最終ジャッジ日時 2024-12-21 15:09:40
合計ジャッジ時間 29,996 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 528 ms
95,744 KB
testcase_01 AC 537 ms
95,744 KB
testcase_02 AC 516 ms
95,828 KB
testcase_03 AC 526 ms
95,872 KB
testcase_04 AC 538 ms
95,828 KB
testcase_05 AC 533 ms
95,744 KB
testcase_06 AC 535 ms
95,872 KB
testcase_07 AC 537 ms
95,872 KB
testcase_08 AC 1,009 ms
141,092 KB
testcase_09 AC 1,233 ms
158,188 KB
testcase_10 AC 1,028 ms
142,304 KB
testcase_11 AC 1,145 ms
149,748 KB
testcase_12 AC 1,078 ms
145,236 KB
testcase_13 AC 948 ms
137,304 KB
testcase_14 AC 1,151 ms
150,184 KB
testcase_15 AC 990 ms
139,968 KB
testcase_16 AC 1,011 ms
140,840 KB
testcase_17 AC 1,240 ms
157,988 KB
testcase_18 AC 986 ms
128,512 KB
testcase_19 AC 1,022 ms
128,512 KB
testcase_20 AC 980 ms
128,760 KB
testcase_21 AC 1,000 ms
128,512 KB
testcase_22 AC 1,051 ms
128,640 KB
testcase_23 AC 983 ms
128,844 KB
testcase_24 AC 998 ms
128,512 KB
testcase_25 AC 996 ms
128,768 KB
testcase_26 AC 996 ms
128,888 KB
testcase_27 AC 1,025 ms
128,904 KB
testcase_28 AC 891 ms
128,896 KB
testcase_29 AC 536 ms
95,744 KB
testcase_30 AC 533 ms
95,808 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 


from random import shuffle,randrange
rd=randrange

class primes():
    def __init__(self, n):
        self.prime_num = n
        self.min_prime = [-1] * (self.prime_num + 1)  # 2以上の自然数に対して最小の素因数を表す
        self.min_prime[0] = 0
        self.min_prime[1] = 1
        i = 2
        self.prime = []
        self.memo_prifac = {}
        self.memodiv=dict()
        while i <= self.prime_num:
            if self.min_prime[i] == -1:
                self.min_prime[i] = i
                self.prime.append(i)
            for j in self.prime:
                if i * j > self.prime_num or j > self.min_prime[i]: break
                self.min_prime[j * i] = j
            i += 1

    def prifac(self, n):
        # 素因数分解した結果を返す
        if n in self.memo_prifac:
            return self.memo_prifac[n]
        res = {}
        x = n
        while x > 1:
            p = self.min_prime[x]
            if p in res:
                res[p] += 1
            else:
                res[p] = 1
            x //= p

        # self.memo_prifac[n] = res  #場合によってはこの行を消すと高速化

        return res

    def divisors(self, n):
        # 約数列挙 メモした方がいいかも
        if n in self.memodiv:return self.memodiv[n]
        if n== 1: return [1]
        prf = self.prifac(n)
        keys = [key for key in prf]

        def divsearch(i):
            if i == len(keys) - 1:
                return [keys[i] ** j for j in range(prf[keys[i]] + 1)]
            else:
                res = []
                subres = divsearch(i + 1)
                p = keys[i]
                for j in range(prf[p] + 1):
                    for node in subres:
                        res.append(node * p ** j)
                return res

        self.memodiv[n]=divsearch(0)
        return self.memodiv[n]

pri=primes(10**6+100)
u=[0]*(10**6+100)
u[1]=1
for x in range(2,10**6+10):
    u[x]=1
    d=pri.prifac(x)
    for p in d:
        if d[p]>=2:u[x]=0
        u[x]*=-1


from math import gcd
mod=998244353
def naive(n,p):
    ans=0
    for bit in range(1,2**n):
        x=[]
        for i in range(n):
            if (bit>>i)&1:x.append(p[i])
        k=len(x)
        flag=1
        for i in range(k-1):
            if gcd(x[i],x[i+1])==1:flag=0
        ans+=flag
    return ans%mod

def sol1(n,P):
    p=[0]+P[:]
    ma=max(p)
    dp=[0]*(ma+1)
    for i in range(1,n+1):
        res=0
        for j in range(1,ma+1):
            if gcd(p[i],j)==1:res+=dp[j]
        dp[p[i]]+=sum(dp)+1-res
        dp[p[i]]%=mod
    return sum(dp)%mod

def sol2(n,P):
    p=[0]+P[:]
    for i in range(1,n+1):
        res=1
        for q in pri.prifac(p[i]):res*=q
        p[i]=res
    ma=max(p)
    dp=[0]*(ma+1)
    g=[0]*(ma+1)
    sumdp=0
    for i in range(1,n+1):
        f1=0
        div=[]
        for m in pri.divisors(p[i]):div.append(m)
        for m in div:
            f1+=u[m]*g[m]
        delta=sumdp+1-f1
        dp[p[i]]+=delta
        dp[p[i]]%=mod
        sumdp+=delta
        sumdp%=mod
        for m in div:
            g[m]+=delta
            g[m]%=mod
    return sum(dp)%mod

n=int(input())
a=list(map(int,input().split()))
print(sol2(n,a))


cnt=0
while 0:
    cnt+=1
    print(cnt)
    n=randrange(1,19)
    p=[rd(1,100) for i in range(n)]
    ansn=naive(n,p)
    ans1=sol1(n,p)
    ans2=sol2(n,p)
    if ans1!=ansn:
        print(n)
        print(*p)
        exit()
0