글 읽기 - 런타임 에러 해결 부탁드립니다... ㅠ

s1004101 4년 전 0

코드가 좀 더럽습니다만...

구름ide에서 질문에 나온 반례들 다 통과했는데

백준에서는 런타임 에러가 뜹니다.

무엇이 문제인지 모르겠습니다. 도와주세요.. .ㅠ

# 이거 나중에 보면 코드 뭔 뜻인지 모를 듯
import sys
# sys.stdin = open('problem.txt', 'r')


def solve():
	# construct graph
	nbd = [set() for _ in range(2*n)]
	for i in range(n-1):
		if first_floor[i] + first_floor[i+1] <= w:
			nbd[i].add(i+1); nbd[i+1].add(i)
		if second_floor[i] + second_floor[i+1] <= w:
			nbd[i+n].add(i+n+1); nbd[i+n+1].add(i+n)
		if first_floor[i] + second_floor[i] <= w:
			nbd[i].add(i+n); nbd[i+n].add(i)
	if first_floor[0] + first_floor[n-1] <= w:
		nbd[0].add(n-1); nbd[n-1].add(0)
	if second_floor[0] + second_floor[n-1] <= w:
		nbd[n].add(2*n-1); nbd[2*n-1].add(n)
	if first_floor[n-1] + second_floor[n-1] <= w:
		nbd[n-1].add(2*n-1); nbd[2*n-1].add(n-1)
	# construct sets
	num_edge = [set([]) for _ in range(3)] # number of edge attached to each point
	for i in range(2*n):
		if len(nbd[i]) != 0:
			num_edge[len(nbd[i])-1].add(i)
	
	# begin solving
	rst = 2*n
	while num_edge[0] or num_edge[1] or num_edge[2]:
		# find now_point and friend_point
		rst -= 1
		if not num_edge[0]:
			if not num_edge[1]:
				now_point = num_edge[2].pop()
			else:
				now_point = num_edge[1].pop()
		else:
			now_point = num_edge[0].pop()
		friend_point = nbd[now_point].pop()
		num_edge[len(nbd[friend_point])-1].discard(friend_point)
		# reflect change of removing points
		# 1. neighborhood of now_point
		for _nbd in nbd[now_point]:
			l = len(nbd[_nbd])
			if l == 1:
				num_edge[0].discard(_nbd)
			elif l == 2 or l == 3:
				num_edge[l-1].discard(_nbd)
				num_edge[l-2].add(_nbd)
			nbd[_nbd].discard(now_point)
		# 2. neighbodhood of friend_point
		for _nbd in nbd[friend_point]:
			l = len(nbd[_nbd])
			if l == 1:
				num_edge[0].discard(_nbd)
			elif l == 2 or l == 3:
				num_edge[l-1].discard(_nbd)
				num_edge[l-2].add(_nbd)
			nbd[_nbd].discard(friend_point)
		# 3. now_point, friend_point self
		nbd[now_point].clear(); nbd[friend_point].clear()
		for tmp in num_edge:
			tmp.discard(now_point); tmp.discard(friend_point)
	# print
	print(rst)


for _ in range(int(sys.stdin.readline())):
	n, w = map(int, sys.stdin.readline().split())
	first_floor = list(map(int, sys.stdin.readline().split()))
	second_floor = list(map(int, sys.stdin.readline().split()))
	solve()

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

문제

문제

출처

ICPC

런타임 에러 해결 부탁드립니다... ㅠ

s1004101 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말