글 읽기 - [파이파이3] 제 코드는 어디서 시간 더 줄일 수 있을까요??

blutics 5년 전 0

파이썬3로는 통과하기 힘들거 같고

파이파이3은 통과하신분이 있으시네요....

주석이 없어서 곤란하실 수도 있는데

수정할 부분이 아니라도 괜찮으니 조언 부탁드립니다.

import sys
from collections import deque
input=sys.stdin.readline
class Map():
    def __init__(self,x):
        self.glist=[]
        self.iter_input(x)

    def iter_input(self,x):
        for i in range(x):
            self.input_node()
        self.print_result()

    def input_node(self):
        new_node=Node([*map(int,input().split())])
        new_group=[]
        remain=[]

        for i in self.glist:
            if i.check_in_nlist(new_node):
                new_group+=i,
            else:
                remain+=i,

        if len(new_group)==0:
            self.glist.append(Group(new_node))
        else:
            if len(new_group)>1:
                new_group.sort(key=lambda x:len(x.nlist))
                for i in new_group[:-1]:
                    new_group[-1].combine(i)
            new_group[-1].add_element(new_node)
            self.glist=remain+[new_group[-1]]

    def print_result(self):
        print(len(self.glist))
        
class Group():
    gindex=0
    def __init__(self,x):
        self.nlist=[x]
        self.index=self.set_index()

    def add_element(self,x):
        self.nlist.append(x)
        x.set_group(self)
        
    def check_in_nlist(self,x):
        for i in self.nlist:
            if i.connection(x):
                return True
        else:
            return False
            
    def combine(self,g):
        self.nlist+=g.nlist
        for i in g.nlist:
            i.set_group(self)
            
    @classmethod
    def set_index(cls):
        cls.gindex+=1
        return cls.gindex-1

class Node():
    def __init__(self,x):
        self.x=x[0]
        self.y=x[1]
        self.range=x[2]

    def set_group(self,g):
        self.group=g
        
    def connection(self,b):
        if abs(self.x-b.x)+abs(self.y-b.y)<=self.range+b.range:
            return True
        else:
            return False
for i in range(int(input())):
    a=Map(int(input()))

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

문제

문제

출처

ICPC

[파이파이3] 제 코드는 어디서 시간 더 줄일 수 있을까요??

blutics 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말