結果

問題 No.1502 Many Simple Additions
ユーザー googol_S0
提出日時 2021-05-07 22:08:06
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 515 ms / 2,000 ms
コード長 1,654 bytes
コンパイル時間 276 ms
コンパイル使用メモリ 81,864 KB
実行使用メモリ 129,808 KB
最終ジャッジ日時 2024-09-15 18:12:52
合計ジャッジ時間 7,624 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input=sys.stdin.buffer.readline
sys.setrecursionlimit(10**6)
N,M,K=map(int,input().split())
G=[[] for i in range(N)]
for i in range(M):
  x,y,z=map(int,input().split())
  x-=1
  y-=1
  G[x].append((y,z))
  G[y].append((x,z))
A=[]
ANS=1
ANS2=1
mod=10**9+7
F=[0]*N
X=[]
Y=[[] for i in range(N)]
V=K-1
from collections import *
INF=10**17
for i in range(N):
  if F[i]:
    continue
  for j in range(len(X)-1,-1,-1):
    Y[X[j]].clear()
    del X[j]
  Q=deque([(i,(1,0))])
  while len(Q):
    for _ in range(len(Q)):
      v=Q.popleft()
      X.append(v[0])
      F[v[0]]=1
      Y[v[0]].append(v[1])
      for j in range(len(G[v[0]])-1,-1,-1):
        Q.append((G[v[0]][j][0],(-v[1][0],G[v[0]][j][1]-v[1][1])))
        del G[v[0]][j]
  X=list(set(X))
  L,R=1,K
  L2,R2=1,V
  for j in range(len(X)):
    j=X[j]
    Z=[INF,INF]
    for k in range(len(Y[j])):
      if Y[j][k][0]==1:
        if Z[0]==INF:
          Z[0]=Y[j][k][1]
        else:
          if Z[0]!=Y[j][k][1]:
            L=K+1
            L2=K
      else:
        if Z[1]==INF:
          Z[1]=Y[j][k][1]
        else:
          if Z[1]!=Y[j][k][1]:
            L=K+1
            L2=K
    if Z[0]<INF and Z[1]<INF:
      if (Z[0]^Z[1])&1:
        L=K+1
        L2=K
      else:
        x=(Z[1]-Z[0])>>1
        L=max(L,x)
        L2=max(L2,x)
        R=min(R,x)
        R2=min(R2,x)
    if Z[0]<INF:
      R=min(R,K-Z[0])
      R2=min(R2,V-Z[0])
      L=max(L,-Z[0]+1)
      L2=max(L2,-Z[0]+1)
    if Z[1]<INF:
      R=min(R,Z[1]-1)
      R2=min(R2,Z[1]-1)
      L=max(L,Z[1]-K)
      L2=max(L2,Z[1]-V)
  ANS=ANS*max(0,R-L+1)%mod
  ANS2=ANS2*max(0,R2-L2+1)%mod
print((ANS-ANS2)%mod)
0