문제풀이(Problem Solving)

같은 숫자는 싫어, Python3 [프로그래머스]

게임이 더 좋아 2021. 3. 9. 23:26
반응형
728x170

 

문제 설명

배열 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 함수를 완성해 주세요.

제한사항

  • 배열 arr의 크기 : 1,000,000 이하의 자연수

  • 배열 arr의 원소의 크기 : 0보다 크거나 같고 9보다 작거나 같은 정수

 

 


 

def solution(arr):
    answer = []
    for i in range(len(arr)):
        if arr[i-1] != arr[i] or len(answer) == 0:
            answer.append(arr[i])
    return answer

 

아무 생각 없이 짰는데... 바로 통과 ㅋㅋㅋ

여기서 핵심은,, 기존의 것과 다른 것이 들어왔을 때랑 가장 처음의 원소는 어떻게 넣느냐? 에 중점을 뒀다.

 

즉, 마지막으로 들어온 list의 마지막 원소가 어떻게 되느냐이다. 그래서 pop 쓸라다가. 

return이 되어버리니 그냥 다르게 썼다.

 

문제를 잘 읽고 결국 원하는게 뭐지? 라고 생각해보면 

저렇게 추상화가 가능하다.

 

하지만 다른 사람의 풀이를 보고 넘어가자

여기서 중요한 사항은 !!!

빈 배열에다 a[-1:]해도 그냥 먹는 것이다... 나처럼 굳이 맨 처음을 고려할 필요가 없었다.또 한 수 배워간다.

def no_continuous(s):
    a = []
    for i in s:
        if a[-1:] == [i]: continue
        a.append(i)
    return a

 

 

728x90
반응형
그리드형