🤖 옌지니어의 IT 기록 🤖

https://school.programmers.co.kr/learn/courses/30/lessons/181894

문제

문제 설명

정수 배열 arr가 주어집니다. 배열 안의 2가 모두 포함된 가장 작은 연속된 부분 배열을 return 하는 solution 함수를 완성해 주세요.

단, arr에 2가 없는 경우 [-1]을 return 합니다.

제한 사항

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

입출력 예

소스 코드

# 1 (시간초과로 실패)
def solution(arr):
    cnt = arr.count(2)
    start = end = 0
    answer = []

    if cnt == 0:
        answer = [-1]
    elif cnt == 1:
        answer = [2]
    else:
        for idx in range(len(arr)):
            if arr[idx]==2 and cnt==arr.count(2):
                start = idx
                cnt -= 1
            elif arr[idx]==2 and cnt==1:
                end = idx
            elif arr[idx]==2:
                cnt -= 1
        answer = arr[start:end+1]
    
    return answer

풀이 # 1

• arr에서 2의 갯수를 cnt로 설정하고 2가 없으면 answer = [-1], 1개면 answer = [2]
• arr에서 2가 2개 이상인 경우
  • 2의 갯수 cnt에서 2가 등장할 때마다 1씩 빼서 몇 번째 등장인지 확인
  • start = 처음 2가 등장한 인덱스 / end = 마지막 2가 등장한 인덱스
  • answer = arr의 start부터 end까지의 인덱스

# 2
def solution(arr):
    answer = []
    idx_list = []

    for idx in range(len(arr)):
        if arr[idx] == 2:
            idx_list.append(idx)

    cnt = len(idx_list)
    if cnt == 0:
        answer = [-1]
    else:
        answer = arr[idx_list[0]:idx_list[-1]+1]

    return answer

풀이 #2

• arr를 돌면서 2가 등장할 때마다 idx_list에 idx 추가
• idx_list의 갯수를 cnt로 두고 0이면 answer = [-1]
• idx_list의 갯수(=cnt)가 0아 아닌 경우
  • idx_list에는 2의 인덱스가 순서대로 정렬
  • idx_list[0] = 처음 2가 등장한 인덱스 / idx_list[-1]  = 마지막 2가 등장한 인덱스
  • answer = arr의 idx_list[0]부터 idx_list[-1]까지의 인덱스

# 3
def solution(arr):
    answer = []
    if 2 not in arr:
        answer = [-1]
    else:
        answer = arr[arr.index(2) : len(arr) - arr[::-1].index(2)]
    return answer

풀이 #3

• arr에 2가 포함되어 있지 않으면 answer = [-1]
• arr에 2가 포함되어 있는경우
  • arr.index(2) = 처음으로 2가 등장한 인덱스
  • arr[::-1].index(2) = arr를 거꾸로 봤을 때 처음 2가 등장한 인덱스 = 마지막으로 2가 등장한 인덱스
  • arr를 거꾸로 봤을 때의 인덱스이므로 전체 길이에서 빼기...!!

첫 번째 풀이는 이것 저것 조건을 덧붙이다 보니 시간이 오래 걸릴 수 밖에 없었다. 그래서 두 번째 풀이로는 2가 있는 경우의 인덱스만 추출해서 처음과 마지막 인덱스의 arr를 출력했다. 이것도 괜찮다고 생각했는데 다른 사람이 푼 세 번째 풀이를 참고해보니까 신박하고 깔끔했다...다른 사람의 풀이를 참고하면서 문제를 많이 풀어봐야겠다. 나도 센스쟁이가 되고 싶다...!!