[프로그래머스 Python] 코딩 기초 트레이닝 Day 19 - 문자열, 리스트(배열)

---

1. 세 개의 구분자

임의의 문자열이 주어졌을 때 문자 “a”, “b”, “c”를 구분자로 사용해 문자열을 나누고자 합니다.
예를 들어 주어진 문자열이 “baconlettucetomato”라면 나눠진 문자열 목록은 [“onlettu”, “etom”, “to”] 가 됩니다.
문자열 myStr이 주어졌을 때 위 예시와 같이 “a”, “b”, “c”를 사용해 나눠진 문자열을 순서대로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.
단, 두 구분자 사이에 다른 문자가 없을 경우에는 아무것도 저장하지 않으며, return할 배열이 빈 배열이라면 [“EMPTY”]를 return 합니다.

제한사항

• 1 ≤ myStr의 길이 ≤ 1,000,000
  • myStr은 알파벳 소문자로 이루어진 문자열 입니다.

코드 작성

import re
def solution(myStr):
    myStr = re.sub('a|b|c', ' ', myStr).strip()
    return myStr.split() if len(myStr) > 0 else ['EMPTY']

정규표현식을 사용하여(모듈 re 로드) 특정 문자를 공백으로 변환하고 좌우 공백을 제거해준다. 그 후 공백을 기준으로 문자열을 나눈다. 공백을 기준으로 나눠 리스트에 담은 결과 컨테이너(리스트) 속에 요소가 하나라도 있으면 그 리스트를 반환하고 리스트 내 요소가 존재하지 않는다면 [‘EMPTY’] 를 반환하도록 했다.

2. 배열의 원소만큼 추가하기

아무 원소도 들어있지 않은 빈 배열 X가 있습니다. 양의 정수 배열 arr가 매개변수로 주어질 때, arr의 앞에서부터 차례대로 원소를 보면서 원소가 a라면 X의 맨 뒤에 a를 a번 추가하는 일을 반복한 뒤의 배열 X를 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ arr의 길이 ≤ 100
• 1 ≤ arr의 원소 ≤ 100

코드 작성

def solution(arr):
    l = []
    for i in arr:
        for j in range(i):
            l.append(i)
    return l

코드 작성(간단한 ver)

def solution(arr):
    return [i for i in arr for j in range(i)]

리스트 컴프리헨션으로 지저분해 보이던 코드를 간결하게 작성했다. 중첩된 반복문을 통해 원하는 배열을 만들 수 있었다. 첫 번째 코드와 두 번째 코드(간단한 ver)는 동일한 값을 반환하며 1번 코드를 함축한 결과가 2번 코드이다.

3. 빈 배열에 추가, 삭제하기

아무 원소도 들어있지 않은 빈 배열 X가 있습니다. 길이가 같은 정수 배열 arr과 boolean 배열 flag가 매개변수로 주어질 때, flag를 차례대로 순회하며 flag[i]가 true라면 X의 뒤에 arr[i]를 arr[i] × 2 번 추가하고, flag[i]가 false라면 X에서 마지막 arr[i]개의 원소를 제거한 뒤 X를 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ arr의 길이 = flag의 길이 ≤ 100
• arr의 모든 원소는 1 이상 9 이하의 정수입니다.
• 현재 X의 길이보다 더 많은 원소를 빼는 입력은 주어지지 않습니다.

코드 작성

def solution(arr, flag):
    l = []
    for i, f in zip(arr, flag):
        if f:
            for j in range(i):
                l.append(i)
                l.append(i)
        else:
            l = l[:-i]
    return l

4. 배열 만들기 6

0과 1로만 이루어진 정수 배열 arr가 주어집니다. arr를 이용해 새로운 배열 stk을 만드려고 합니다.
i의 초기값을 0으로 설정하고 i가 arr의 길이보다 작으면 다음을 반복합니다.

• 만약 stk이 빈 배열이라면 arr[i]를 stk에 추가하고 i에 1을 더합니다.
• stk에 원소가 있고, stk의 마지막 원소가 arr[i]와 같으면 stk의 마지막 원소를 stk에서 제거하고 i에 1을 더합니다.
• stk에 원소가 있는데 stk의 마지막 원소가 arr[i]와 다르면 stk의 맨 마지막에 arr[i]를 추가하고 i에 1을 더합니다.
  위 작업을 마친 후 만들어진 stk을 return 하는 solution 함수를 완성해 주세요.
  단, 만약 빈 배열을 return 해야한다면 [-1]을 return 합니다.

제한사항

• 1 ≤ arr의 길이 ≤ 1,000,000
  • arr의 원소는 0 또는 1 입니다.

코드 작성

def solution(arr):
    stk = []
    for i in range(len(arr)):
        if len(stk) == 0:
            stk.append(arr[i])
        elif stk[-1] == arr[i]:
            stk = stk[:-1]
        else:
            stk.append(arr[i])
    return stk if len(stk) > 0 else [-1]

위 코드에서 stk의 마지막 요소가 arr[i]와 같다면 마지막 요소를 제거할 때 인덱싱을 사용했는데, pop 매서드를 사용해도 같은 결과가 나온다. pop을 사용한 코드는 아래와 같다.

코드 작성(pop)

def solution(arr):
    stk = []
    for i in range(len(arr)):
        if len(stk) == 0:
            stk.append(arr[i])
        elif stk[-1] == arr[i]:
            stk.pop()
        else:
            stk.append(arr[i])
    return stk if len(stk) > 0 else [-1]

5. 무작위로 K개의 수 뽑기

랜덤으로 서로 다른 k개의 수를 저장한 배열을 만드려고 합니다. 적절한 방법이 떠오르지 않기 때문에 일정한 범위 내에서 무작위로 수를 뽑은 후, 지금까지 나온적이 없는 수이면 배열 맨 뒤에 추가하는 방식으로 만들기로 합니다.
이미 어떤 수가 무작위로 주어질지 알고 있다고 가정하고, 실제 만들어질 길이 k의 배열을 예상해봅시다.
정수 배열 arr가 주어집니다. 문제에서의 무작위의 수는 arr에 저장된 순서대로 주어질 예정이라고 했을 때, 완성될 배열을 return 하는 solution 함수를 완성해 주세요.
단, 완성될 배열의 길이가 k보다 작으면 나머지 값을 전부 -1로 채워서 return 합니다.

제한사항

• 1 ≤ arr의 길이 ≤ 100,000
  • 0 ≤ arr의 원소 ≤ 100,000
• 1 ≤ k ≤ 1,000

코드 작성

def solution(arr, k):
    l = []
    for a in arr:
        if len(l) < k and a not in l:
            l.append(a)
    # 배열의 길이가 k보다 작은 경우 -1로 채우는 과정
    if len(arr) < k:
        for _ in range(k-len(arr)):
            l.append(-1)
    return l[:k]

이번 문제에서 set으로 중복된 숫자를 제거하지 않은 이유는 배열의 순서가 무너질 수 있기 때문이다. arr을 set으로 중복제거하여 다시 list로 변환하는 과정에서 이전의 배열이 무너질 수 있기 때문이다.

📍 문제 출처

• 세 개의 구분자
• 배열의 원소만큼 추가하기
• 빈 배열에 추가, 삭제하기
• 배열 만들기 6
• 무작위로 K개의 수 뽑기

👩🏻‍💻개인 공부 기록용 블로그입니다
오류나 틀린 부분이 있을 경우 댓글 혹은 메일로 따끔하게 지적해주시면 감사하겠습니다.