結果

問題 No.3187 Mingle
ユーザー titia
提出日時 2025-06-23 03:34:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,762 ms / 2,500 ms
コード長 1,885 bytes
コンパイル時間 367 ms
コンパイル使用メモリ 81,664 KB
実行使用メモリ 96,640 KB
最終ジャッジ日時 2025-08-30 07:44:14
合計ジャッジ時間 39,921 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline

# エラトステネスの篩を用いた素因数分解・約数列挙
MAX=3*10**5+10 # 使いたい最大値を指定

# 線形篩。エラトステネスの篩を線形にできる。
Sieve=[-1]*MAX
Primes=[]

for i in range(2,MAX):
    if Sieve[i]==-1:
        Primes.append(i)
        Sieve[i]=i

    for p in Primes:
        if p*i>=MAX or p>Sieve[i]:
            break
        else:
            Sieve[p*i]=p


# 素因数分解
def fact(x):
    D=dict()
    while x!=1:
        k=Sieve[x]
        if k in D:
            D[k]+=1
        else:
            D[k]=1
        x//=k
    return D

# 約数列挙
def faclist(x):
    LIST=[1]
    while x!=1:
        k=Sieve[x]
        count=0
        while x%k==0:
            count+=1
            x//=k

        LIST2=[]
        for l in LIST:
            for i in range(1,count+1):
                LIST2.append(l*k**i)
        LIST+=LIST2

    return LIST

N,mod=map(int,input().split())

seg_el=1<<((N*2+1).bit_length()) # Segment treeの台の要素数
SEG=[0]*(2*seg_el) # 1-indexedなので、要素数2*seg_el.Segment treeの初期値で初期化

def getvalue(n,seg_el): # 一点の値を取得
    i=n+seg_el
    ANS=0
    
    ANS=SEG[i]
    i>>=1# 子ノードへ
    
    while i!=0:
        ANS+=SEG[i]
        i>>=1

    return ANS

def updates(l,r,x): # 区間[l,r)を更新.
    L=l+seg_el
    R=r+seg_el

    while L<R:
        if L & 1:
            SEG[L]=(SEG[L]+x)%mod
            L+=1

        if R & 1:
            R-=1
            SEG[R]=(SEG[R]+x)%mod
        L>>=1
        R>>=1


for i in range(3,N+1):
    L=faclist(i)

    INV=pow(i,mod-2,mod)

    G=getvalue(i,seg_el)
    ANS=(G*INV+1)*pow(i-len(L),mod-2,mod)*i%mod

    #print(i,L,G,ANS)

    for x in L[1:]:
        updates(i+1,i+x,ANS)

    #print([getvalue(i,seg_el) for i in range(1,N+1)])

print(ANS)

    
    
0