알고리즘/프로그래머스

[프로그래머스] LEVEL2 - 튜플 (Python)

Coblin 2020. 4. 27. 21:12
반응형

https://programmers.co.kr/learn/courses/30/lessons/64065

 

코딩테스트 연습 - 튜플

"{{2},{2,1},{2,1,3},{2,1,3,4}}" [2, 1, 3, 4] "{{1,2,3},{2,1},{1,2,4,3},{2}}" [2, 1, 3, 4] "{{4,2,3},{3},{2,3,4,1},{2,3}}" [3, 2, 4, 1]

programmers.co.kr

문제설명

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(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 이하인 경우만 입력으로 주어집니다.

입출력 예

s result
"{{2},{2,1},{2,1,3},{2,1,3,4}}" [2, 1, 3, 4]
"{{1,2,3},{2,1},{1,2,4,3},{2}}" [2, 1, 3, 4]
"{{20,111},{111}}" [111, 20]
"{{123}}" [123]
"{{4,2,3},{3},{2,3,4,1},{2,3}}" [3, 2, 4, 1]

입출력 예 설명

입출력 예 #1

문제 예시와 같습니다.

입출력 예 #2

문제 예시와 같습니다.

입출력 예 #3

(111, 20)을 집합 기호를 이용해 표현하면 {{111}, {111,20}}이 되며, 이는 {{20,111},{111}}과 같습니다.

입출력 예 #4

(123)을 집합 기호를 이용해 표현하면 {{123}} 입니다.

입출력 예 #5

(3, 2, 4, 1)을 집합 기호를 이용해 표현하면 {{3},{3,2},{3,2,4},{3,2,4,1}}이 되며, 이는 {{4,2,3},{3},{2,3,4,1},{2,3}}과 같습니다.

answer

def solution(s):
    answer = []
    
    #s를 출력해보면 {{2},{2,1},{2,1,3},{2,1,3,4}}가 나오는데 앞의 {{와 마지막의 }}를 제거해준다
    list_s = s[2:-2] 
    
    #위의 코드를 출력해보면 2},{2,1},{2,1,3},{2,1,3,4가 나오는데 },{가 규칙적이므로 스플릿해준다.
    list_s = list_s.split("},{")
    
    #스플릿하여 나온 리스트를 각요소의 길이별로 정렬한다.
    list_s.sort(key=len)
    
    #요소의 길이가 2개이상인경우 다시 스플릿하여 담을 변수
    temp = ""
    
    #리스트를 i에 담아 for문을 돌리고 ","가 없는(숫자가 한개인)경우 answer에 붙여준다.
    for i in list_s:
        if "," not in i:
            answer.append(i)
        #","가 있는(숫자가 두개이상)경우 ","로 스플릿하여 temp에 넣는다.
        else:
            temp = i.split(",")
            #다시 for문으로 j가 answer에 있는지 확인하고 없을경우에 answer에 붙여준다.
            for j in temp:
                if j not in answer:
                    answer.append(j)
	#answer의 요소들이 str이므로 map을 사용해 int로 형변환 해주고 list로 결과를 리스트로 만들어준다.
    answer = list(map(int, answer))
    return answer

막상 풀고나니 별거아닌 문제였는데 [111, 20]이 계속 1,2,0 이런씩으로 한글자씩 나와서 코드를 조금씩 수정하다 전부 지우고 새로운 방법으로 접근해보니 그제서야 풀리더라구요 ㅠㅠ 두시간이 좀 넘게 걸린것 같아요

반응형