[프로그래머스 기초문제] 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

---

입출력 예

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

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 합니다.

제출 답안

import Foundation

func solution(_ arr: [Int], _ queries: [[Int]]) -> [Int] {
    var result = arr // 가변 배열로 복사하여 사용
    
    for query in queries {
        let s = query[0], e = query[1], k = query[2] // 쿼리에서 시작, 끝, 배수 값 추출
        
        // s부터 e까지의 인덱스 중 k의 배수인 경우 값을 증가
        for i in s...e {
            if i % k == 0 {
                result[i] += 1
            }
        }
    }
    return result
}

---

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

제출 답안

import Foundation

func solution(_ l: Int, _ r: Int) -> [Int] {
    var result: [Int] = [] // 조건을 만족하는 숫자를 저장할 배열
    
    for num in l...r { // l부터 r까지 반복
        let strNum = String(num) // 숫자를 문자열로 변환
        
        // 모든 문자(숫자)가 0 또는 5 로만 이루어져 있는지 확인
        if strNum.allSatisfy({ $0 == "0" || $0 == "5" }) {
            result.append(num) // 조건을 만족하면 배열에 추가
        }
    }
    return result.isEmpty ? [-1] : result // 결과 배열이 비어 있으면 -1 반환
}

https://developer.apple.com/documentation/swift/array/allsatisfy(_:)

allSatisfy(_:) | Apple Developer Documentation

모든 요소가 조건을 충족하면 true를 반환한다.
하나라도 조건을 충족하지 않으면 false를 반환한다.

---

3. 카운트 업

정수 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합니다.

제출 답안

import Foundation

func solution(_ start_num:Int, _ end_num:Int) -> [Int] {
    return Array(start_num...end_num)
}

---

4. 콜라츠 수열 만들기

모든 자연수 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홀짝 여부

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

제출 답안

func solution(_ n: Int) -> [Int] {
    var result: [Int] = [n] // 결과 배열을 n으로 초기화
    var num = n
    
    while num != 1 {
        if num % 2 == 0 {
            num /= 2 // 짝수이면 2로 나누기
        } else {
            num = 3 * num + 1 // 홀수이면 3을 곱하고 1 더하기
        }
        result.append(num) // 계산된 값을 배열에 추가
    }
    
    return result // 콜라츠 수열 반환
}

---

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

---

입출력 예

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

---

제출 답안

func solution(_ arr: [Int]) -> [Int] {
    var stk: [Int] = [] // 결과 배열
    var i = 0 // 인덱스 변수 초기화
    
    while i < arr.count {
        if stk.isEmpty { 
            // stk가 비어있으면 arr[i] 추가 후 i 증가
            stk.append(arr[i])
            i += 1
        } else if stk.last! < arr[i] { 
            // stk 마지막 원소가 arr[i]보다 작으면 추가 후 i 증가
            stk.append(arr[i])
            i += 1
        } else {
            // stk 마지막 원소가 arr[i]보다 크거나 같으면 마지막 원소 제거
            stk.removeLast()
        }
    }
    return stk // 결과 반환
}

---

Day 7