프로그래머스 기초 Day 7 파이썬

프로그래머스

수열과 구간 쿼리 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

입출력 예

arr	queries	result
[0, 1, 2, 4, 3]	[[0, 4, 1], [0, 3, 2], [0, 3, 3]]	[3, 2, 4, 6, 4]

입출력 예 설명

입출력 예 #1

• 각 쿼리에 따라 arr가 다음과 같이 변합니다.

arr

[0, 1, 2, 4, 3]

[1, 2, 3, 5, 4]

[2, 2, 4, 5, 4]

[3, 2, 4, 6, 4]

• 따라서 [3, 2, 4, 6, 4]를 return 합니다.

solution.py

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

---

배열 만들기 2

문제 설명

정수 l과 r이 주어졌을 때, l이상 r이하의 정수 중에서 숫자 "0"과 "5"로만 이루어진 모든 정수를 오름차순으로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.

만약 그러한 정수가 없다면, -1이 담긴 배열을 return 합니다.

제한사항

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

입출력 예

l	r	result
5	555	[5, 50, 55, 500, 505, 550, 555]
10	20	[-1]

입출력 예 설명

입출력 예 #1

• 5 이상 555 이하의 0과 5로만 이루어진 정수는 작은 수부터 5, 50, 55, 500, 505, 550, 555가 있습니다. 따라서 [5, 50, 55, 500, 505, 550, 555]를 return 합니다.

입출력 예 #2

• 10 이상 20 이하이면서 0과 5로만 이루어진 정수는 없습니다. 따라서 [-1]을 return 합니다.

solution.py

def solution(l, r):
    answer = []
    
    for num in range(l, r + 1):
        if all(temp in '05' for temp in str(num)):
            answer.append(num)
            
    if not answer:
        return [-1]
    
    return answer

---

카운트 업

문제 설명

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

제한사항

• 0 ≤ start_num ≤ end_num ≤ 50

입출력 예

start_num	end_num	result
3	10	[3, 4, 5, 6, 7, 8, 9, 10]

입출력 예 설명

입출력 예 #1

• 3부터 10까지의 숫자들을 담은 리스트 [3, 4, 5, 6, 7, 8, 9, 10]를 return합니다.

solution.py

def solution(start_num, end_num):
    answer = list(range(start_num, end_num + 1))
    return answer

---

콜라츠 수열 만들기

문제 설명

모든 자연수 x에 대해서 현재 값이 x이면 x가 짝수일 때는 2로 나누고, x가 홀수일 때는 3 * x + 1로 바꾸는 계산을 계속해서 반복하면 언젠가는 반드시 x가 1이 되는지 묻는 문제를 콜라츠 문제라고 부릅니다.

그리고 위 과정에서 거쳐간 모든 수를 기록한 수열을 콜라츠 수열이라고 부릅니다.

계산 결과 1,000 보다 작거나 같은 수에 대해서는 전부 언젠가 1에 도달한다는 것이 알려져 있습니다.

임의의 1,000 보다 작거나 같은 양의 정수 n이 주어질 때 초기값이 n인 콜라츠 수열을 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ n ≤ 1,000

입출력 예

n	result
10	[10, 5, 16, 8, 4, 2, 1]

입출력 예 설명

입출력 예 #1

• 순서대로 연산한 결과를 표로 만들면 다음과 같습니다.

연산 횟수	x	홀짝 여부
0	10	짝수
1	5	홀수
2	16	짝수
3	8	짝수
4	4	짝수
5	2	짝수
6	1	홀수

• 따라서 [10, 5, 16, 8, 4, 2, 1]을 return 합니다.

solution.py

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

---

배열 만들기 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

입출력 예

arr	result
[1, 4, 2, 5, 3]	[1, 2, 3]

입출력 예 설명

입출력 예 #1

• 각 작업을 마친 후에 배열의 변화를 나타내면 다음 표와 같습니다.

i	arr[i]	stk
0	1	[]
1	4	[1]
2	2	[1, 4]
2	2	[1]
3	5	[1, 2]
4	3	[1, 2, 5]
4	3	[1, 2]
-	-	[1, 2, 3]

• 따라서 [1, 2, 3]을 return 합니다.

solution.py

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
        else:
            stk.pop()
            
    return stk