結果

問題 No.2705 L to R Graph (Another ver.)
ユーザー hirayuu_ychirayuu_yc
提出日時 2024-02-11 09:27:41
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,790 ms / 3,000 ms
コード長 1,877 bytes
コンパイル時間 170 ms
コンパイル使用メモリ 82,252 KB
実行使用メモリ 371,656 KB
最終ジャッジ日時 2024-09-30 03:47:06
合計ジャッジ時間 66,006 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 41 ms
52,608 KB
testcase_01 AC 40 ms
52,992 KB
testcase_02 AC 45 ms
58,624 KB
testcase_03 AC 44 ms
59,648 KB
testcase_04 AC 45 ms
59,264 KB
testcase_05 AC 45 ms
59,776 KB
testcase_06 AC 45 ms
59,392 KB
testcase_07 AC 43 ms
58,496 KB
testcase_08 AC 1,315 ms
303,420 KB
testcase_09 AC 1,510 ms
337,436 KB
testcase_10 AC 1,096 ms
260,680 KB
testcase_11 AC 1,219 ms
287,604 KB
testcase_12 AC 898 ms
247,040 KB
testcase_13 AC 541 ms
195,348 KB
testcase_14 AC 1,442 ms
327,796 KB
testcase_15 AC 239 ms
110,348 KB
testcase_16 AC 891 ms
246,144 KB
testcase_17 AC 428 ms
171,360 KB
testcase_18 AC 1,365 ms
314,644 KB
testcase_19 AC 729 ms
235,148 KB
testcase_20 AC 492 ms
182,484 KB
testcase_21 AC 1,184 ms
278,488 KB
testcase_22 AC 1,156 ms
276,096 KB
testcase_23 AC 1,640 ms
358,080 KB
testcase_24 AC 1,661 ms
360,376 KB
testcase_25 AC 1,673 ms
354,476 KB
testcase_26 AC 1,682 ms
360,436 KB
testcase_27 AC 1,550 ms
346,172 KB
testcase_28 AC 1,529 ms
343,872 KB
testcase_29 AC 1,546 ms
339,564 KB
testcase_30 AC 1,569 ms
346,100 KB
testcase_31 AC 1,549 ms
342,048 KB
testcase_32 AC 1,552 ms
346,676 KB
testcase_33 AC 1,543 ms
343,732 KB
testcase_34 AC 1,652 ms
360,372 KB
testcase_35 AC 1,516 ms
343,076 KB
testcase_36 AC 1,652 ms
357,664 KB
testcase_37 AC 1,691 ms
365,888 KB
testcase_38 AC 1,534 ms
342,436 KB
testcase_39 AC 1,661 ms
360,440 KB
testcase_40 AC 1,625 ms
352,244 KB
testcase_41 AC 1,705 ms
370,716 KB
testcase_42 AC 1,617 ms
357,856 KB
testcase_43 AC 1,708 ms
371,284 KB
testcase_44 AC 1,694 ms
371,408 KB
testcase_45 AC 1,739 ms
371,420 KB
testcase_46 AC 1,736 ms
371,424 KB
testcase_47 AC 1,720 ms
371,656 KB
testcase_48 AC 1,708 ms
371,284 KB
testcase_49 AC 1,741 ms
371,400 KB
testcase_50 AC 1,730 ms
371,280 KB
testcase_51 AC 1,790 ms
371,532 KB
testcase_52 AC 1,737 ms
371,412 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

def cumsum(x,d=1):
    for i in range(d,len(x)):
        x[i]+=x[i-d]

N,P=map(int,input().split())
ans=0
f=[0,0]
fact=1
for i in range(2,N+1):
    f.append((pow(N,N-i,P)*fact)%P)
    fact*=N-i
    fact%=P
dist=[0]*N
cum=0
pat=0
for i in range(N-2,0,-1):
    cum+=f[i+2]
    pat+=cum
    cum%=P
    pat%=P
    dist[i]=pat
backet=150
pats=[0]*N*2
det=[[0]*N*2 for i in range(backet+1)]
for i in range(1,N+1):
    for j in range(N+1):
        if i*j>=N:
            break
        if j==0:
            pats[1]+=N-i+1
            pats[i*(j+1)]-=N-i+1
            continue
        if j>backet:
            cnt=0
            for k in range(i*j+1,i*(j+1),j):
                pats[k]+=1
                cnt+=1
            pats[i*(j+1)]-=cnt
        else:
            det[j][i*j+1]+=1
            cnt=(i*(j+1)-2)//j-i+1
            det[j][min(N*2-1,i*j+1+cnt*j)]-=1
            pats[i*(j+1)]-=cnt
for i in range(1,backet+1):
    cumsum(det[i],i)
    for j in range(N*2):
        pats[j]+=det[i][j]
cumsum(pats)
for i in range(N-1):
    pats[i]=N*(N+1)//2-pats[i]
    ans+=pats[i]*dist[i]
    ans%=P
cumsum(f)
div=[[] for i in range(N+1)]
cum=[[0] for i in range(N+1)]
for i in range(1,N+1):
    for j in range(i,N+1,i):
        div[j].append(i)
        cum[i].append(cum[i][-1]+f[j])
for i in range(1,N+1):
    gcd=[0]*len(div[i])
    for j in range(len(div[i])-1,-1,-1):
        gcd[j]+=N//div[i][j]
        for k in range(j):
            if div[i][j]%div[i][k]==0:
                gcd[k]-=gcd[j]
        pl=cum[div[i][j]][-1]-cum[div[i][j]][i//div[i][j]]
        plcnt=len(cum[div[i][j]])-1-i//div[i][j]
        mi=cum[div[i][j]][-1]-cum[div[i][j]][i//div[i][j]-1]+f[i-1]*(i//div[i][j]-1)
        micnt=plcnt+1+i//div[i][j]-1
        ans+=(pl+f[-1]*(micnt-plcnt)-mi)*gcd[j]
        if j==0:
            ans+=(pl+f[-1]*(micnt-plcnt)-mi)*N*(N-1)//2
        ans%=P
print((ans*N*(N-1))%P)
0