2019 카카오 개발자 겨울 인턴십/튜플

https://programmers.co.kr/learn/courses/30/lessons/64065?language=python3

코딩테스트 연습 - 튜플

- 문제 설명

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(tuple)이라고 합니다. n개의 요소를 가진 튜플을 n-튜플(n-tuple)이라고 하며, 다음과 같이 표현할 수 있습니다.

• (a1, a2, a3, ..., an)

튜플은 다음과 같은 성질을 가지고 있습니다.

1. 중복된 원소가 있을 수 있습니다. ex : (2, 3, 1, 2)

2. 원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플입니다. ex : (1, 2, 3) ≠ (1, 3, 2)

3. 튜플의 원소 개수는 유한합니다.

원소의 개수가 n개이고, 중복되는 원소가 없는 튜플 (a1, a2, a3, ..., an)이 주어질 때(단, a1, a2, ..., an은 자연수), 이는 다음과 같이 집합 기호 '{', '}'를 이용해 표현할 수 있습니다.

• {{a1}, {a1, a2}, {a1, a2, a3}, {a1, a2, a3, a4}, ... {a1, a2, a3, a4, ..., an}}

예를 들어 튜플이 (2, 1, 3, 4)인 경우 이는

• {{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}

와 같이 표현할 수 있습니다. 이때, 집합은 원소의 순서가 바뀌어도 상관없으므로

• {{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}

• {{2, 1, 3, 4}, {2}, {2, 1, 3}, {2, 1}}

• {{1, 2, 3}, {2, 1}, {1, 2, 4, 3}, {2}}

는 모두 같은 튜플 (2, 1, 3, 4)를 나타냅니다.

특정 튜플을 표현하는 집합이 담긴 문자열 s가 매개변수로 주어질 때, s가 표현하는 튜플을 배열에 담아 return 하도록 solution 함수를 완성해주세요.

 

[제한사항]

• s의 길이는 5 이상 1,000,000 이하입니다.

• s는 숫자와 '{', '}', ',' 로만 이루어져 있습니다.

• 숫자가 0으로 시작하는 경우는 없습니다.

• s는 항상 중복되는 원소가 없는 튜플을 올바르게 표현하고 있습니다.

• s가 표현하는 튜플의 원소는 1 이상 100,000 이하인 자연수입니다.

• return 하는 배열의 길이가 1 이상 500 이하인 경우만 입력으로 주어집니다.

 

효율성을 고려하기보단, 일단 구현을 하기 위해 코드를 짜보았다...

def solution(s):
    answer = []
    numbers = []
    num = ''
    i=check=0
    for i in range(len(s)):
        if s[i]=='{':
            check+=1
            continue
        elif s[i]==',' and check==1:
            continue
        elif s[i]==',' and check==2:
            numbers.append(int(num))
            num=''
            continue
        elif s[i]=='}' and check==2:
            check-=1
            numbers.append(int(num))
            num=''
            continue
        elif s[i]=='}':
            check-=1
            continue      
        num+=str(s[i])
    
    answer = list(set(numbers))
    return answer

위와 같이 작성해 실행해보았더니..

 

테스트 케이스를 통과하지 못한다. (기댓값과 결괏값의 요소들의 순서가 다름)

ex) 입력값이 "{{1,2,3},{2,1},{1,2,4,3},{2}}" 이면

{2} > {2,1} > {1,2,3} > {1,2,4,3} 순으로 돌아야한다.

[2, 1, 3, 4]

 

위 문제 때문에 애를 좀 먹었다.

 

조금 많이 더럽긴한데 암튼 다음과 같다... 

나중에 다시 풀어봐야할듯

def solution(s):
    answer = []
    numbers = [] #2차원리스트(톱니형 리스트)
    numbers.append([])
    num = ''
    i=check=0
    x=y=0 #y는 numbers에 append되는 원소의 개수를 세기위해 선언
    
    for i in range(len(s)):	#문자열s로부터 숫자를 추출해 2차원리스트 numbers에 담아준다.
        if s[i]=='{':
            check+=1
            continue
        elif s[i]==',' and check==1:
            continue
        elif s[i]==',' and check==2:
            numbers[x].append(int(num))
            y+=1
            num=''
            continue
        elif s[i]=='}' and check==2:
            check-=1
            numbers[x].append(int(num))
            y+=1
            numbers.append([])
            x+=1
            num=''
            continue
        elif s[i]=='}':
            check-=1
            continue      
        num+=str(s[i])
    numbers.pop() #마지막 반복에서 빈 리스트가 append된 것을 pop
    
    temp=[] #numbers를 순서에 맞게 넣어줄 리스트
    length=1
    while(True): 
        for i in range(len(numbers)):
            if len(numbers[i])==length:
                for j in range(len(numbers[i])):
                    temp.append(numbers[i][j])
        length+=1
        if len(temp)==y:
            break
    
    for i in range(len(temp)): #중복된 숫자 없이 answer에 append
        check=1
        for j in range(len(answer)):
            if(answer[j]==temp[i]):
                check=0
                break
        if(check):
            answer.append(temp[i])
    return answer