結果

問題 No.2642 Don't cut line!
ユーザー ゼットゼット
提出日時 2024-02-19 23:10:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 768 ms / 4,000 ms
コード長 2,052 bytes
コンパイル時間 208 ms
コンパイル使用メモリ 82,324 KB
実行使用メモリ 145,644 KB
最終ジャッジ日時 2024-09-29 03:39:26
合計ジャッジ時間 19,895 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
55,188 KB
testcase_01 AC 768 ms
145,604 KB
testcase_02 AC 712 ms
145,644 KB
testcase_03 AC 726 ms
145,276 KB
testcase_04 AC 707 ms
144,900 KB
testcase_05 AC 704 ms
145,088 KB
testcase_06 AC 537 ms
111,716 KB
testcase_07 AC 538 ms
112,288 KB
testcase_08 AC 530 ms
111,352 KB
testcase_09 AC 560 ms
112,220 KB
testcase_10 AC 541 ms
112,164 KB
testcase_11 AC 526 ms
111,460 KB
testcase_12 AC 553 ms
112,492 KB
testcase_13 AC 533 ms
112,048 KB
testcase_14 AC 534 ms
111,820 KB
testcase_15 AC 541 ms
112,344 KB
testcase_16 AC 552 ms
112,728 KB
testcase_17 AC 636 ms
136,120 KB
testcase_18 AC 730 ms
141,544 KB
testcase_19 AC 512 ms
124,260 KB
testcase_20 AC 515 ms
111,872 KB
testcase_21 AC 424 ms
105,576 KB
testcase_22 AC 425 ms
103,232 KB
testcase_23 AC 709 ms
144,488 KB
testcase_24 AC 395 ms
111,396 KB
testcase_25 AC 442 ms
111,516 KB
testcase_26 AC 550 ms
111,660 KB
testcase_27 AC 469 ms
121,280 KB
testcase_28 AC 663 ms
139,496 KB
testcase_29 AC 537 ms
111,380 KB
testcase_30 AC 471 ms
111,948 KB
testcase_31 AC 574 ms
129,844 KB
testcase_32 AC 497 ms
111,708 KB
testcase_33 AC 37 ms
55,192 KB
testcase_34 AC 35 ms
54,840 KB
testcase_35 AC 32 ms
53,800 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class unif:
  def __init__(self,n):
    self.pare=[-1]*n
    self.size=[1]*n
  def root(self,x):
    while self.pare[x]!=-1:
      x=self.pare[x]
    return x
  def unite(self,u,v):
    rootu=self.root(u)
    rootv=self.root(v)
    if rootu!=rootv:
      if self.size[rootu]>=self.size[rootv]:
        self.pare[rootv]=rootu
        self.size[rootu]+=self.size[rootv]
      else:
        self.pare[rootu]=rootv
        self.size[rootv]+=self.size[rootu]
  def same(self,s,t):
    return self.root(s)==self.root(t)
import pypyjit
pypyjit.set_param('max_unroll_recursion=-1')
N,M,C=map(int,input().split())
G=[[] for i in range(N)]
k=0
L=[]
for i in range(M):
  a,b,w,p=map(int,input().split())
  L.append((w,a-1,b-1,p))
L.sort()
cost=0
R=[]
Z=unif(N)
result=0
for i in range(M):
  w,a,b,p=L[i][:]
  if Z.same(a,b)==False:
    k+=1
    G[a].append((b,w))
    G[b].append((a,w))
    result=max(result,p)
    cost+=w
    Z.unite(a,b)
  else:
    R.append((w,a,b,p))
if k!=N-1:
  print(-1)
  exit()
if cost>C:
  print(-1)
  exit()
dist=[-1]*N
dist[0]=0
from collections import deque
S=deque()
p=[[0]*N for i in range(20)]
S.append(0)
v=[[0]*N for i in range(20)]
while S:
  x=S.popleft()
  for B in G[x]:
    y,w=B[:]
    if dist[y]>=0:
      continue
    dist[y]=dist[x]+1
    S.append(y)
    p[0][y]=x
    v[0][y]=w
for k in range(1,20):
  for i in range(N):
    p[k][i]=p[k-1][p[k-1][i]]
    v[k][i]=max(v[k-1][i],v[k-1][p[k-1][i]])
def LCA(x,y):
  a,b=x,y
  if dist[a]<dist[b]:
    a,b=b,a
  u=dist[a]-dist[b]
  for k in range(20):
    if (u>>k)&1:
      a=p[k][a]
  if a==b:
    return a
  for k in range(19,-1,-1):
    if p[k][a]!=p[k][b]:
      a=p[k][a]
      b=p[k][b]
  a=p[0][a]
  return a
for B in R:
  w,a,b,f=B[:]
  d=0
  z=LCA(a,b)
  e1=dist[a]-dist[z]
  e2=dist[b]-dist[z]
  pos=a
  for k in range(19,-1,-1):
    if (e1>>k)&1:
      d=max(d,v[k][pos])
      pos=p[k][pos]
  pos=b
  for k in range(19,-1,-1):
    if (e2>>k)&1:
      d=max(d,v[k][pos])
      pos=p[k][pos]
  cost2=cost-d+w
  if cost2<=C:
    result=max(result,f)
print(result)
0