프로그래머스 기초문제 Day7 반복문

Day7

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

---

수열과 구간 쿼리 4

문제 설명

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

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

어제 문제랑 비슷해서 손쉽게 푸나 했더니 비슷한듯 다른 문제... 

인덱스 s~e까지 자르는게 아니라 배열은 그대로 가져가되

s~e까지만 반복하고, 그 안에서 i가 k의 배수일 때, 해당하는 배열에만 1을 더했다.

---

배열 만들기 2

문제 설명

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

 

def solution(l, r):
    answer = []
    Xlist= set(['1','2','3','4','6','7','8','9'])
    for i in range(l, r+1):
        num = set(str(i))
        if num & Xlist:
            pass
        else:
            answer.append(i)
    if not answer:
        answer.append(-1)
    return answer

Xlist와 l~r사이의 숫자 중에 교집합이 있으면(0, 5가 아닌 값이 있으면) pass

교집합이 없으면 (0, 5만 남아있을테니) 그때의 값을 answer에 append하기

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

다른 사람 풀이:

 if not set(str(num)) - set(['0', '5']): 값을 집합으로 표현하고

집합 - 집합 = 비어있으면 >> 1,2,3,4,6,7,8,9의 값이 없다는 것이니 그때의 값을 배열에 저장

이런 신박한 방법을 어찌 생각해내시는지.....

---

카운트 업

문제 설명

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

 

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

응? 갑자기 이렇게 쉬운게 나와버린다고? 감사합니다.

def solution(start, end):
    return list(range(start, end + 1))
    
def solution(start, end):
    return [i for i in range(start, end+1)]

파이썬은 사기야 .. 코드를 효율적으로 줄여쓰는 연습을 해보자 .....!

---

콜라츠 수열 만들기

문제 설명

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

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

1. n이 1일 때까지 무한 반복

2. 짝, 홀수일 때 연산다르게 하고, 배열에 값 저장

---

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

 

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

    return stk

문제 읽으면서 코드썼는데 정답이라 기뻐했는데 if, elif문 실행부분 똑같네 ......요.......ㅎ

정신 몽롱한 상태로 푸는 것은 좋지 않습니다 ...... 그래도 이제 잘 수 있어 기뻐요!

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

while문으로 먼저 조건에 만족하는 원소들을 모두 지운 후,

마지막 남은 원소들 모두 추가. 끝!

---