結果
問題 | No.2705 L to R Graph (Another ver.) |
ユーザー |
|
提出日時 | 2024-02-11 09:14:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,623 ms / 3,000 ms |
コード長 | 1,877 bytes |
コンパイル時間 | 260 ms |
コンパイル使用メモリ | 82,504 KB |
実行使用メモリ | 290,188 KB |
最終ジャッジ日時 | 2024-09-28 18:07:18 |
合計ジャッジ時間 | 59,465 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 50 |
ソースコード
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=100 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)