[프로그래머스 Lv.2] 튜플 파이썬 풀이
문제 설명
셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(tuple)이라고 합니다. n개의 요소를 가진 튜플을 n-튜플(n-tuple)이라고 하며, 다음과 같이 표현할 수 있습니다.
- (a1, a2, a3, ..., an)
튜플은 다음과 같은 성질을 가지고 있습니다.
- 중복된 원소가 있을 수 있습니다. ex : (2, 3, 1, 2)
- 원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플입니다. ex : (1, 2, 3) ≠ (1, 3, 2)
- 튜플의 원소 개수는 유한합니다.
원소의 개수가 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] |
문제 예시와 같습니다.
입출력 예 #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}}과 같습니다.
나의 풀이
문제를 보고 2가지 구현 방법을 생각했다.
첫 번째 방법
1. 문자열 s를 리스트로 만들어준다.
2. 만들어진 리스트를 원소 길이를 기준으로 정렬해준다.
3. 제일 안쪽 원소를 돌면서 answer 리스트에 추가해준다. (만약 answer에 있다면 추가하지 않는다.)
두 번째 방법
1. 문자열 s를 리스트로 만들어준다.
2. 만들어진 리스트를 2차원 리스트로 만들어주고, 원소 길이를 기준으로 정렬해준다.
3. 2차원 리스트를 1차원 리스트로 만들어준다.
4. 리스트 내에 중복을 제거한다.
우선 첫 번째 방법이다.
아이디어 1: 리스트로 만들기
아이디어 2: 원소 길이를 기준으로 정렬하기
아이디어 3: answer
for nums in s:
num = nums.split(",")
for n in num:
if int(n) not in answer:
answer.append(int(n))안쪽 원소도 리스트로 만들어준 뒤, 숫자 하나 하나 검사한다.
만약 answer에 없다면, append 해준다.
코드
def solution(s):
answer = []
s = s[2:-2].split("},{")
s.sort(key=len)
for nums in s:
num = nums.split(",")
for n in num:
if int(n) not in answer:
answer.append(int(n))
return answer
두 번째 방법 코드
def solution(s):
s = [ x.split(",") for x in s[2:-2].split("},{")]
s.sort(key=len)
ss = [int(y) for x in s for y in x]
answer = list(dict.fromkeys(ss))
return answer두 번째 방법은
ss = [int(y) for x in s for y in x]answer = list(dict.fromkeys(ss))위 두 부분만 알면 된다.
[y for x in s for y in s]
List comprehension을 이용하여 2차원 리스트를 1차원 리스트로 변경해주는 부분이다.
list(dict.fromkeys(ss))
리스트의 중복을 순서를 보장하며 없애주는 부분이다.