結果

問題 No.2642 Don't cut line!
ユーザー ゼットゼット
提出日時 2024-02-19 23:10:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 922 ms / 4,000 ms
コード長 2,052 bytes
コンパイル時間 243 ms
コンパイル使用メモリ 81,700 KB
実行使用メモリ 145,036 KB
最終ジャッジ日時 2024-02-20 12:48:41
合計ジャッジ時間 24,773 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge16
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 41 ms
55,744 KB
testcase_01 AC 909 ms
145,036 KB
testcase_02 AC 900 ms
145,036 KB
testcase_03 AC 922 ms
144,652 KB
testcase_04 AC 900 ms
144,524 KB
testcase_05 AC 901 ms
144,524 KB
testcase_06 AC 671 ms
111,152 KB
testcase_07 AC 687 ms
111,728 KB
testcase_08 AC 674 ms
110,948 KB
testcase_09 AC 676 ms
111,796 KB
testcase_10 AC 682 ms
111,748 KB
testcase_11 AC 678 ms
110,920 KB
testcase_12 AC 690 ms
111,884 KB
testcase_13 AC 670 ms
111,492 KB
testcase_14 AC 669 ms
111,408 KB
testcase_15 AC 690 ms
112,268 KB
testcase_16 AC 726 ms
112,268 KB
testcase_17 AC 789 ms
135,612 KB
testcase_18 AC 851 ms
140,780 KB
testcase_19 AC 621 ms
123,448 KB
testcase_20 AC 610 ms
111,100 KB
testcase_21 AC 536 ms
105,112 KB
testcase_22 AC 536 ms
102,876 KB
testcase_23 AC 881 ms
144,000 KB
testcase_24 AC 506 ms
110,856 KB
testcase_25 AC 550 ms
111,064 KB
testcase_26 AC 691 ms
111,128 KB
testcase_27 AC 586 ms
120,592 KB
testcase_28 AC 824 ms
138,836 KB
testcase_29 AC 658 ms
111,048 KB
testcase_30 AC 586 ms
111,528 KB
testcase_31 AC 700 ms
129,268 KB
testcase_32 AC 614 ms
111,112 KB
testcase_33 AC 41 ms
55,744 KB
testcase_34 AC 41 ms
55,744 KB
testcase_35 AC 37 ms
53,588 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