글 읽기 - [파이썬] 엣지케이스가 있을까요?

(追記) (追記ここまで)

[파이썬] 엣지케이스가 있을까요?

13911번 - 집 구하기

k550706 2년 전 0

더미노드에 연결하는 테크닉을 합쳐서 처리했는데

93%에서 WA입니다

혹시 어디가 틀린지 알수 있을까요?

감사합니다.

import heapq,sys
input=lambda:sys.stdin.readline().rstrip()
v,e=map(int,input().split())
INF=float('inf')
g=[[] for _ in range(v+3)]
for i in range(e):
 a,b,c=map(int,input().split())
 g[a].append((c,b))
 g[b].append((c,a))
macdonald_dist=list(map(int,input().split()))[1]
macdonald=set(map(int,input().split()))
for i in macdonald:
 g[v+1].append((0,i))
starbucks_dist=list(map(int,input().split()))[1]
starbucks=set(map(int,input().split()))
for i in starbucks:
 g[v+2].append((0,i))
home=set(i for i in range(1,v+1))-macdonald-starbucks
def shortcut(lst,limit):
 d=[INF for _ in range(v+1)]+[0,0]
 pq=[(0,lst)]
 while pq:
 cur_d,cur_n=heapq.heappop(pq)
 if d[cur_n]<cur_d:continue
 for nxt_d,nxt_n in g[cur_n]:
 temp=d[cur_n]+nxt_d
 if temp>limit:continue
 if d[nxt_n]>temp:
 d[nxt_n]=temp
 heapq.heappush(pq,(temp,nxt_n))
 return d
a=shortcut(v+1,starbucks_dist)
b=shortcut(v+2,macdonald_dist)
ans=INF
for i in home:
 ans=min(ans,a[i]+b[i])
print([ans,-1][ans==INF])

cinador 2년 전 1

37 ~ 38번째 줄에 함수에 들어가는 limit 변수가 서로 바뀌었습니다. 둘만 바꿔주면 통과됩니다.

k550706 2년 전 1

하.. 감사합니다

왜 반대로 쓴게 93퍼까지 맞는거죠.....

댓글을 작성하려면 로그인해야 합니다.

(追記) (追記ここまで)

Baekjoon Online Judge

채점 현황

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말

사업자 등록 번호: 541-88-00682
대표자명: 최백준
주소: 서울시 서초구 서초대로74길 29 서초파라곤 412호
전화번호: 02-521-0487 (이메일로 연락 주세요)
이메일: contacts@startlink.io
통신판매신고번호: 제 2017-서울서초-2193 호

한국어 | English (Beta)

문제

문제

출처

ICPC

[파이썬] 엣지케이스가 있을까요?

k550706 2년 전 0

cinador 2년 전 1

k550706 2년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말