프로그래머스 연습 문제
배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다. 단, 제거된 후 남은 수들을 반환할 때는 배열 arr의 원소들의 순서를 유지해야 합니다. 예를 들면,
arr = [1, 1, 3, 3, 0, 1, 1] 이면 [1, 3, 0, 1] 을 return 합니다.
arr = [4, 4, 4, 3, 3] 이면 [4, 3] 을 return 합니다.
배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.
answer에 arr[0]을 담아두고 이후의 것들을 비교하는 방법
def solution(arr): answer = [arr[0]] for i in range(1, len(arr)): if arr[i] != arr[i-1] : answer.append(arr[i]) return answer
if arr[i] != arr[i-1] :
: arr[i] 와 arr[i+1]을 비교한다면, 마지막 반복 때 마지막 인덱스인 arr[i+1]은 arr에 없는 인덱스이기 때문에 오류가 난다. 따라서 i와 i-1을 비교하는 if문을 작성한다.
for i in range(1, len(arr)):
: if 문에서 i-1을 사용하기 위해 range의 범위를 1부터 len(arr)까지로 설정한다.
answer = [arr[0]]
: range가 1부터 시작하므로, 첫 번째 요소인 arr[0]을 미리 answer 리스트에 담아둔다
for 문에서 answer에 arr[i]를 담고, 이후에 if 문으로 arr[i]가 연속되는 숫자인지 비교해서, 연속된다면 pop()으로 빼내는 방법
def solution(arr): answer = [] for i in range(len(arr)): answer.append(arr[i]) if [arr[i]] == arr[i + 1: i + 2]: answer.pop() return answer
if [arr[i]] == arr[i + 1: i + 2]:
pop()을 사용하지 않고 간소화한 방법
def solution(arr): answer = [] for i in range(len(arr)): if [arr[i]] != arr[i + 1: i + 2]: answer.append(arr[i]) return answer
len(answer)의 값이 0일 때 담고, answer에 담긴 이전값과 i가 같지 않다면(연속되지 않는 숫자라면) answer 변수에 담는 방법
def solution(arr): answer = [] for i in arr: if len(answer) == 0 or answer[-1] != i: answer.append(i) return answer
answer 변수에 i를 처음부터 담고, 이후에 answer[-1]와 i를 비교한다.
답안 분석 2번을 간소화한 방법과 비슷하지만 if 문에서 len(answer) == 0 인 것을 활용한 것이 새롭게 느껴져서 분석해보았다.
arr[-1]부터 비교해서 answer에 담는 방법.
def solution(arr): answer = [] for i in arr: if answer[-1:] == [i]: continue answer.append(i) return answer
arr[-1]부터도!! 비교가 가능하다.
if answer[-1:] == [i]:
: if answer[-1] == [i] 을 사용하면 answer 리스트가 비어있는 경우 인덱스 오류가 발생한다.
따라서 [-1:]를 통해 answer 리스트의 마지막 요소만 선택한 후, 리스트 [i]와 비교하는 구문이다.