[프로그래머스 Python] 코딩 기초 트레이닝 Day 7 - 반복문

---

1. 수열과 구간 쿼리 4

정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. queries의 원소는 각각 하나의 query를 나타내며, [s, e, k] 꼴입니다. 각 query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 i가 k의 배수이면 arr[i]에 1을 더합니다. 위 규칙에 따라 queries를 처리한 이후의 arr를 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
  • 0 ≤ arr의 원소 ≤ 1,000,000
• 1 ≤ queries의 길이 ≤ 1,000
  • 0 ≤ s ≤ e < arr의 길이
  • 0 ≤ k ≤ 5

코드 작성

def solution(arr, queries):
    for q in queries:
        for i in range(q[0], q[1]+1):
            if i % q[-1] == 0:
                arr[i] = arr[i] + 1
    return arr

2. 배열 만들기 2

정수 l과 r이 주어졌을 때, l 이상 r이하의 정수 중에서 숫자 “0”과 “5”로만 이루어진 모든 정수를 오름차순으로 저장한 배열을 return 하는 solution 함수를 완성해 주세요. 만약 그러한 정수가 없다면, -1이 담긴 배열을 return 합니다.

제한사항

• 1 ≤ l ≤ r ≤ 1,000,000

코드 작성

def solution(l, r):
    # set 으로 중복제거 후, 길이가 2 이하
    num_list = []
    condition = set(['0', '5'])
    
    for n in range(l, r+1):
        if n % 5 == 0 and len(set(str(n))-set(condition)) == 0:
            num_list.append(n)

    return num_list if len(num_list) >= 1 else [-1]

3. 카운트 업

정수 start와 end가 주어질 때, start부터 end까지의 숫자를 차례로 담은 리스트를 return하도록 solution 함수를 완성해주세요.

제한사항

• 0 ≤ start ≤ end ≤ 50

코드 작성

def solution(start, end):
    answer = [n for n in range(start, end+1)]
    return answer

4. 콜라츠 수열 만들기

모든 자연수 x에 대해서 현재 값이 x이면 x가 짝수일 때는 2로 나누고, x가 홀수일 때는 3 * x + 1로 바꾸는 계산을 계속해서 반복하면 언젠가는 반드시 x가 1이 되는지 묻는 문제를 콜라츠 문제라고 부릅니다. 그리고 위 과정에서 거쳐간 모든 수를 기록한 수열을 콜라츠 수열이라고 부릅니다. 계산 결과 1,000 보다 작거나 같은 수에 대해서는 전부 언젠가 1에 도달한다는 것이 알려져 있습니다. 임의의 1,000 보다 작거나 같은 양의 정수 n이 주어질 때 초기값이 n인 콜라츠 수열을 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ n ≤ 1,000

코드 작성

def solution(n):
    num = n
    num_list = [num]
    while num != 1:
        if num % 2 == 0: 
            num = num // 2
            num_list.append(num)
        else:
            num = num * 3 + 1
            num_list.append(num)
    return num_list

5. 배열 만들기 4

정수 배열 arr가 주어집니다. arr를 이용해 새로운 배열 stk를 만드려고 합니다.

변수 i를 만들어 초기값을 0으로 설정한 후 i가 arr의 길이보다 작으면 다음 작업을 반복합니다.

• 만약 stk가 빈 배열이라면 arr[i]를 stk에 추가하고 i에 1을 더합니다.
• stk에 원소가 있고, stk의 마지막 원소가 arr[i]보다 작으면 arr[i]를 stk의 뒤에 추가하고 i에 1을 더합니다.
• stk에 원소가 있는데 stk의 마지막 원소가 arr[i]보다 크거나 같으면 stk의 마지막 원소를 stk에서 제거합니다.

위 작업을 마친 후 만들어진 stk를 return 하는 solution 함수를 완성해 주세요.

제한사항

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

코드 작성

def solution(arr):
    stk = []
    i = 0
    while i < len(arr):
        if len(stk) == 0: 
            stk.append(arr[i])
            i += 1
        else:
            if stk[-1] < arr[i]:
                stk.append(arr[i])
                i += 1
            else:
                stk = stk[:-1]
    return stk

6. 문제 출처

• 수열과 구간 쿼리 4
• 배열 만들기 2
• 카운트 업
• 콜라츠 수열 만들기
• 배열 만들기 4

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