[파이썬] Day 7 반복문

푼 문제들

 

 

 

수열과 구간 쿼리 4

 

문제

 

 

정답

def solution(arr, queries):
    for query in queries:
        s, e, k = query
        for i in range(s, e + 1):
            if i % k == 0:
                arr[i] += 1
    return arr

s, e, k = query < 배열의 요소를 순서대로 각각의 변수에 할당한다.

range(a, b) < a ~ b -1 범위의 숫자를 반환한다.

 

 

 

배열 만들기 2

 

문제

 

 

정답

def solution(l, r):
    res = []
    for i in range(l, r + 1):
        if not set(str(i)) - set(['0', '5']):
            res.append(i)
    return res if res else [-1]

l ~ r의 범위의 수를 순회하면서 0또는 5로 이루어진 수를 res에 추가한다.

 

 

 

i가 특정숫자로만 이루어졌는지 검사하기 

if not set(str(i)) - set(['0', '5'])

# i = 305 -> str = [3, 0, 5]
# [3, 0, 5] - [0, 5] = [3]
# not [3] = false

집합(set)은 중복된 값을 포함하지 않아 i문자열에

0이나 5가 몇 개 있든 집합 내에서 같은 값은 하나만 존재하게 된다.

 

i의 요소를 하나하나 검사하기 위해 문자열로 변환해줘야 한다.

 

not은 해당 집합이 빈 집합일 때 true를 아니면 false를 반환한다.

i의 각 자리숫자가 0이나 5로 이루어져 있다면 조건문의 연산식은 빈 집합의 값을 가진다.

 

 

 

카운트 업

 

문제

 

 

정답

def solution(a, b):
    return [i for i in range(a, b + 1)]

리스트 컴프리헨션을 사용하여 간단하게 풀 수 있다.

 

 

 

콜라츠 수열 만들기

 

문제

 

 

정답

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

while을 이용해 n의 값이 1이 될때까지 반복문을 돌리면 된다.

 

 

 

 

배열 만들기 4

 

문제

 

 

정답

def solution(arr):
    stk = []
    i = 0

    while i < len(arr):
        if not stk:
            stk.append(arr[i])
            i += 1
        elif stk[-1] < arr[i]:
            stk.append(arr[i])
            i += 1
        elif stk[-1] >= arr[i]:
            stk.remove(stk[-1])
    return stk

문제 그대로 코드를 짜서 풀어주면 된다. 

 

append()로 배열에 값을 추가하고 remove()로 배열에서 값을 삭제할 수 있다.

stk[-1]은 stk의 마지막 요소에 해당하며 stk.remove(stk[-1])를 stk.pop()으로 대체할 수 있다.

 

 

다른 풀이

def solution(arr):
    stk = []
    for i in arr:
        while stk and stk[-1] >= i:
            stk.pop()
        stk.append(i)
    return stk

배열의 요소를 순회하는 방식으로도 풀 수 있다.

 

스택이 비어있지 않고 스택 맨 위 요소가 i보다 크거나 같은 값을 가질 때까지

스택 맨 위 요소를 제거함으로써 스택의 값들이 오름차순으로 유지된다.