글 읽기 - 파이썬으로 푸는데 왜 이런 차이가 나는지 모르겠습니다.

grapecw 4년 전 0

아래는 제가 입력해서 무사히 정답입니다가 나온 코드입니다.

하지만 위에 이 3줄을 빼면 시간 초과가 뜨더군요.

import sys

input = sys.stdin.readline
print = sys.stdout.write

독학으로 공부하는 중이라서 지식이 일천해서 그런데 왜 입출력을 오버라이딩 하면 시간 초과가 안 나는지 모르겠습니다. 두 입출력 방식의 차이가 어떻길래 이런 결과가 뜨는지 알려 주실 수 있으신가요?

import sys

input = sys.stdin.readline
print = sys.stdout.write

def main() :
	n,m = input().split()

	n = int(n)
	m = int(m)

	parent = [i for i in range(n + 1)]
	level = [1 for i in range(n + 1)]

	for i in range(m) :
		line = input().split()
		for j in range(3) :
			line[j]= int(line[j])
		if(line[0]==0) :
			merge(line[1],line[2],level,parent)
		elif (line[0]==1):
			u = find(line[1],parent)
			v = find(line[2],parent)
			if(u==v) :
				print("yes\n")
			else :
				print("no\n")

def find(u,parent) :
	if (u==parent[u]) :
		return u;
	parent[u] = find(parent[u],parent);
	return parent[u];

def merge(u, v,level,parent) :
	u = find(u,parent)
	v = find(v,parent)
	if (u == v) :
		return
	if (level[u] > level[v]) :
		temp =u
		u=v
		v=temp
	parent[u] = v
	if (level[u] == level[v]) :
		level[v]+=1

main()

herdson 4년 전 1

https://m.blog.naver.com/PostV...

이 글에 간략하게 설명이 되어있군요. 참고가 되셨으면 합니다.

grapecw 4년 전 0

감사합니다. input 내에서도 여러가지 다른 처리를 하기 때문에 오래 걸렸던 거군요

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

문제

문제

출처

ICPC

파이썬으로 푸는데 왜 이런 차이가 나는지 모르겠습니다.

grapecw 4년 전 0

herdson 4년 전 1

grapecw 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말