結果

問題 No.1627 三角形の成立
ユーザー googol_S0
提出日時 2021-07-23 22:26:19
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,114 bytes
コンパイル時間 439 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 97,536 KB
最終ジャッジ日時 2024-07-18 18:16:38
合計ジャッジ時間 3,734 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 3 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod=10**9+7
def cmb(n,r):
  if r<0 or r>n:
    return 0
  return (g1[n]*g2[r]*g2[n-r])%mod
 
N=500000
g1=[1]*(N+3)
for i in range(2,N+3):
  g1[i]=g1[i-1]*i%mod
g2=[0]*len(g1)
g2[-1]=pow(g1[-1],mod-2,mod)
for i in range(N+1,-1,-1):
  g2[i]=g2[i+1]*(i+1)%mod
inv=[0]*(N+3)
for i in range(1,N+3):
  inv[i]=g2[i]*g1[i-1]%mod
N,M=map(int,input().split())
X=[0]*(N+M+2)
Y=[0]*(N+M+2)
Z=[0]*(N+M+2)
Y[N]+=M
for i in range(-N,N+1):
  if i==0:
    Y[M]+=N
    continue
  i=abs(i)
  y=(N-1)//i+1
  if y>=M:
    z=(M-1)*i
    if y:
      Y[y-1]+=max(i*min(y,M)-N,0)
    Y[M]+=N-z
    X[0]+=i*2
    X[M]-=i*2
  else:
    z=(y-1)*i
    if y:
      Y[y-1]+=i*y-N
    Y[y]+=(M-y+1)*(N-z)
    X[0]+=i*2
    X[y]-=i*2
N,M=M,N
for i in range(-N,N+1):
  if abs(i)<=1:
    continue
  i=abs(i)
  y=(N-1)//i+1
  if y>=M:
    z=(M-1)*i
    if y:
      Y[y-1]+=max(i*min(y,M)-N,0)
    Y[M]+=N-z
    X[0]+=i*2
    X[M]-=i*2
  else:
    z=(y-1)*i
    if y:
      Y[y-1]+=i*y-N
    Y[y]+=(M-y+1)*(N-z)
    X[0]+=i*2
    X[y]-=i*2
L=N*M
ANS=L*(L-1)*(L-2)//6
for i in range(N+M):
  X[i+1]+=X[i]
  ANS-=cmb(i+1,3)*(X[i+1]+Y[i+1])
print(ANS%mod)
0