코딩 기초 트레이닝 5

수열과 구간 쿼리 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 합니다.

---

내 풀이

def solution(arr, queries):
    for s, e, k in queries:
        for i in range(len(arr[s:e+1])):
            if s+i % k == 0:
                arr[i] = 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 합니다.

---

내 풀이

def solution(l, r):
    answer = [number for number in range(l, r+1) if all(char in '50' for char in str(number))]
    if not answer:
        return [-1]
    return answer

이 문제는 메서드의 무지로 인해 어려움을 겪었던 것 같다.

---

카운트 업

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

제한사항
0 ≤ start ≤ end ≤ 50
입출력 예
start end result
3 10 [3, 4, 5, 6, 7, 8, 9, 10]
입출력 예 설명
입출력 예 #1

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

---

내 풀이

def solution(start, end):
    answer = [number for number in range(start, end+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 합니다.

---

내 풀이

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

지금 보니까 왜 append를 위에 만들어서 밑에 append(1)을 하게 만든 건지 모르겠다... 그냥 while문 제일 밑에 하나만 두면 될 것을...

---

배열 만들기 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 합니다.

---

내 풀이

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

 

다른 풀이

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

다른 분들에 비해 코드가 난잡하고 볼품없지만 그래도 문제를 해결해 나가는 것에 성취감을 느끼는 것 같다.

더 파이팅 해야지

---