[프로그래머스 기초문제] Day 14 리스트(배열)

1. 홀수 vs 짝수

정수 리스트 num_list가 주어집니다. 가장 첫 번째 원소를 1번 원소라고 할 때, 홀수 번째 원소들의 합과 짝수 번째 원소들의 합 중 큰 값을 return 하도록 solution 함수를 완성해주세요. 두 값이 같을 경우 그 값을 return합니다.

---

제한사항

• 5 ≤ num_list의 길이 ≤ 50
• -9 ≤ num_list의 원소 ≤ 9

---

입출력 예

num_list	result
[4, 2, 6, 1, 7, 6]	17
[-1, 2, 5, 6, 3]	8

---

입출력 예 설명

입출력 예 #1

• 홀수 번째 원소들의 합은 4 + 6 + 7 = 17, 짝수 번째 원소들의 합은 2 + 1 + 6 = 9 이므로 17을 return합니다.

입출력 예 #2

• 홀수 번째 원소들의 합은 -1 + 5 + 3 = 7, 짝수 번째 원소들의 합은 2 + 6 = 8 이므로 8을 return합니다.

제출 답안

import Foundation

func solution(_ num_list: [Int]) -> Int {
    var oddSum = 0   // 홀수 번째 원소들의 합
    var evenSum = 0  // 짝수 번째 원소들의 합

    for (index, value) in num_list.enumerated() {
        if (index + 1) % 2 == 1 { // 1-based index 기준 홀수 번째
            oddSum += value
        } else { // 1-based index 기준 짝수 번째
            evenSum += value
        }
    }
    
    return max(oddSum, evenSum)
}

---

2. 5명씩

최대 5명씩 탑승가능한 놀이기구를 타기 위해 줄을 서있는 사람들의 이름이 담긴 문자열 리스트 names가 주어질 때, 앞에서 부터 5명씩 묶은 그룹의 가장 앞에 서있는 사람들의 이름을 담은 리스트를 return하도록 solution 함수를 완성해주세요. 마지막 그룹이 5명이 되지 않더라도 가장 앞에 있는 사람의 이름을 포함합니다.

---

제한사항

• 5 ≤ names의 길이 ≤ 30
• 1 ≤ names의 원소의 길이 ≤ 10
• names의 원소는 영어 알파벳 소문자로만 이루어져 있습니다.

---

입출력 예

names	result
["nami", "ahri", "jayce", "garen", "ivern", "vex", "jinx"]	["nami", "vex"]

---

입출력 예 설명

입출력 예 #1

• 앞에서부터 5명씩 두 그룹으로 묶을 수 있습니다. ["nami", "ahri", "jayce", "garen", "ivern"], ["vex", "jinx"] 이 두 그룹에서 가장 앞에 서있는 사람들의 이름을 담은 리스트인 ["nami", "vex"]를 return합니다.

제출 답안

import Foundation

func solution(_ names: [String]) -> [String] {
    var result: [String] = []
    
    for i in stride(from: 0, to: names.count, by: 5) {
        result.append(names[i]) // 각 그룹의 첫 번째 사람 추가
    }
    
    return result
}

---

3. 할 일 목록

오늘 해야 할 일이 담긴 문자열 배열 todo_list와 각각의 일을 지금 마쳤는지를 나타내는 boolean 배열 finished가 매개변수로 주어질 때, todo_list에서 아직 마치지 못한 일들을 순서대로 담은 문자열 배열을 return 하는 solution 함수를 작성해 주세요.

---

제한사항

• 1 ≤ todo_list의 길이 1 ≤ 100
• 2 ≤ todo_list의 원소의 길이 ≤ 20
  • todo_list의 원소는 영소문자로만 이루어져 있습니다.
  • todo_list의 원소는 모두 서로 다릅니다.
• finished[i]는 true 또는 false이고 true는 todo_list[i]를 마쳤음을, false는 아직 마치지 못했음을 나타냅니다.
• 아직 마치지 못한 일이 적어도 하나 있습니다.

---

입출력 예

todo_list	finished	result
["problemsolving", "practiceguitar", "swim", "studygraph"]	[true, false, true, false]	["practiceguitar", "studygraph"]

---

입출력 예 설명

입출력 예 #1

• 예제 1번의 todo_list 중에서 "problemsolving"과 "swim"은 마쳤고, "practiceguitar"와 "studygraph"는 아직 마치지 못했으므로 todo_list에서 나온 순서대로 담은 문자열 배열 ["practiceguitar", "studygraph"]를 return 합니다.

제출 답안

import Foundation

func solution(_ todo_list: [String], _ finished: [Bool]) -> [String] {
    var result: [String] = [] // 완료되지 않은 할 일을 저장할 배열

    // todo_list를 인덱스와 함께 순회
    for (index, task) in todo_list.enumerated() {
        if !finished[index] { // finished 배열에서 해당 작업이 완료되지 않았다면
            result.append(task) // 결과 배열에 추가
        }
    }

    return result // 아직 끝내지 않은 할 일들만 반환
}

---

4. n보다 커질 때까지 더하기

정수 배열 numbers와 정수 n이 매개변수로 주어집니다. numbers의 원소를 앞에서부터 하나씩 더하다가 그 합이 n보다 커지는 순간 이때까지 더했던 원소들의 합을 return 하는 solution 함수를 작성해 주세요.

---

제한사항

• 1 ≤ numbers의 길이 ≤ 100
• 1 ≤ numbers의 원소 ≤ 100
• 0 ≤ n < numbers의 모든 원소의 합

---

입출력 예

numbers	n	result
[34, 5, 71, 29, 100, 34]	123	139
[58, 44, 27, 10, 100]	139	239

---

입출력 예 설명

입출력 예 #1

• 예제 1번의 numbers를 문제 설명대로 더해가는 과정을 나타내면 다음의 표와 같습니다.

i	numbers[i]	sum
		0
0	34	34
1	5	39
2	71	110
3	29	139

29를 더한 뒤에 sum 값은 139이고 `n` 값인 123보다 크므로 139를 return 합니다.

• 예제 2번의 numbers의 마지막 원소 전까지의 원소를 sum에 더하면 139입니다. 139는 n 값인 139보다 크지 않고 마지막 원소인 100을 더하면 139보다 커지므로 239를 return 합니다.

제출 답안

func solution(_ numbers: [Int], _ n: Int) -> Int {
    var sum = 0 // 합을 저장할 변수
    
    for num in numbers { // numbers 배열을 순회하며 하나씩 더하기
        sum += num
        if sum > n { // 합이 n보다 커지는 순간 반환
            return sum
        }
    }
    
    return sum // 모든 원소를 더해도 n보다 크지 않은 경우 (보통 문제 조건상 없겠지만 대비)
}

---

5. 수열과 구간 쿼리 1

정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. queries의 원소는 각각 하나의 query를 나타내며, [s, e] 꼴입니다.

각 query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 arr[i]에 1을 더합니다.

위 규칙에 따라 queries를 처리한 이후의 arr를 return 하는 solution 함수를 완성해 주세요.

---

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
  • 0 ≤ arr의 원소 ≤ 1,000,000
• 1 ≤ queries의 길이 ≤ 1,000
  • 0 ≤ s ≤ e < arr의 길이

---

입출력 예

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

---

입출력 예 설명

입출력 예 #1

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

i	queries[i]	arr
-	-	[0, 1, 2, 3, 4]
0	[0, 1]	[1, 2, 2, 3, 4]
1	[1, 2]	[1, 3, 3, 3, 4]
2	[2, 3]	[1, 3, 4, 4, 4]

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

제출 답안

import Foundation

func solution(_ arr: [Int], _ queries: [[Int]]) -> [Int] {
    var result = arr // 원본 배열을 수정할 수 있도록 복사
    
    for query in queries { // 각 쿼리를 처리
        let s = query[0]
        let e = query[1]
        
        for i in s...e { // s ≤ i ≤ e 범위 내의 모든 요소를 1 증가
            result[i] += 1
        }
    }
    
    return result // 변경된 배열 반환
}

---

Day 14