[프로그래머스 기초문제] Day 20 함수(메서드)

1. 배열의 길이를 2의 거듭제곱으로 만들기

정수 배열 arr이 매개변수로 주어집니다. arr의 길이가 2의 정수 거듭제곱이 되도록 arr 뒤에 정수 0을 추가하려고 합니다. arr에 최소한의 개수로 0을 추가한 배열을 return 하는 solution 함수를 작성해 주세요.

---

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
• 1 ≤ arr의 원소 ≤ 1,000

---

입출력 예

arr	result
[1, 2, 3, 4, 5, 6]	[1, 2, 3, 4, 5, 6, 0, 0]
[58, 172, 746, 89]	[58, 172, 746, 89]

---

입출력 예 설명

입출력 예 #1

• 예제 1번의 arr의 길이는 6입니다. arr의 길이를 2의 정수 거듭제곱으로 만드는 방법은 0을 2개, 10개, 26개,..., 추가하는 방법이 있고 그중 최소한으로 0을 추가하는 방법은 2개를 추가하는 것입니다. 따라서 [1, 2, 3, 4, 5, 6, 0, 0]을 return 합니다.

입출력 예 #2

• 예제 2번의 arr의 길이는 4이고 이미 2의 정수 거듭제곱입니다. 따라서 뒤에 0을 추가하지 않아도 되므로 [58, 172, 746, 89]를 return 합니다.

제출 답안

import Foundation

func solution(_ arr: [Int]) -> [Int] {
    let length = arr.count // 현재 배열의 길이
    var powerOfTwo = 1 // 2의 거듭제곱을 찾기 위한 변수
    
    // length 이상이 되는 가장 작은 2의 거듭제곱 찾기
    while powerOfTwo < length {
        powerOfTwo *= 2
    }
    
    // 부족한 개수만큼 0을 추가한 배열 반환
    return arr + Array(repeating: 0, count: powerOfTwo - length)
}

---

2. 배열 비교하기

이 문제에서 두 정수 배열의 대소관계를 다음과 같이 정의합니다.

• 두 배열의 길이가 다르다면, 배열의 길이가 긴 쪽이 더 큽니다.
• 배열의 길이가 같다면 각 배열에 있는 모든 원소의 합을 비교하여 다르다면 더 큰 쪽이 크고, 같다면 같습니다.

두 정수 배열 arr1과 arr2가 주어질 때, 위에서 정의한 배열의 대소관계에 대하여 arr2가 크다면 -1, arr1이 크다면 1, 두 배열이 같다면 0을 return 하는 solution 함수를 작성해 주세요.

---

제한사항

• 1 ≤ arr1의 길이 ≤ 100
• 1 ≤ arr2의 길이 ≤ 100
• 1 ≤ arr1의 원소 ≤ 100
• 1 ≤ arr2의 원소 ≤ 100
• 문제에서 정의한 배열의 대소관계가 일반적인 프로그래밍 언어에서 정의된 배열의 대소관계와 다를 수 있는 점에 유의해주세요.

---

입출력 예

arr1	arr2	result
[49, 13]	[70, 11, 2]	-1
[100, 17, 84, 1]	[55, 12, 65, 36]	1
[1, 2, 3, 4, 5]	[3, 3, 3, 3, 3]	0

---

입출력 예 설명

입출력 예 #1

• 예제 1번에서는 arr1의 길이는 2이고 arr2의 길이는 3으로 arr2의 길이가 더 깁니다. 따라서 arr2가 arr1보다 크므로 -1을 return 합니다.

입출력 예 #2

• 예제 2번에서는 arr1의 길이과 arr2의 길이가 4로 같습니다. arr1의 모든 원소의 합은 100 + 17 + 84 + 1 = 202이고 arr2의 모든 원소의 합은 55 + 12 + 65 + 36 = 168으로 arr1의 모든 원소의 합이 더 큽니다. 따라서 arr1이 arr2보다 크므로 1을 return 합니다.

입출력 예 #3

• 예제 3번에서는 arr1의 길이와 arr2의 길이가 5로 같고 각 배열의 모든 원소의 합 또한 15로 같습니다. 따라서 arr1과 arr2가 같으므로 0을 return 합니다.

제출 답안

import Foundation

func solution(_ arr1: [Int], _ arr2: [Int]) -> Int {
    // 배열의 길이를 비교
    if arr1.count > arr2.count {
        return 1
    } else if arr1.count < arr2.count {
        return -1
    }
    
    // 배열의 길이가 같다면 원소의 합을 비교
    let sum1 = arr1.reduce(0, +)
    let sum2 = arr2.reduce(0, +)
    
    return sum1 == sum2 ? 0 : (sum1 > sum2 ? 1 : -1)
}

---

3. 문자열 묶기

문자열 배열 strArr이 주어집니다. strArr의 원소들을 길이가 같은 문자열들끼리 그룹으로 묶었을 때 가장 개수가 많은 그룹의 크기를 return 하는 solution 함수를 완성해 주세요.

---

제한사항

• 1 ≤ strArr의 길이 ≤ 100,000
  • 1 ≤ strArr의 원소의 길이 ≤ 30
  • strArr의 원소들은 알파벳 소문자로 이루어진 문자열입니다.

---

입출력 예

strArr	result
["a","bc","d","efg","hi"]	2

---

입출력 예 설명

입출력 예 #1

• 각 문자열들을 길이에 맞게 그룹으로 묶으면 다음과 같습니다.

문자열 길이	문자열 목록	개수
1	["a", "d"]	2
2	["bc", "hi"]	2
3	["efg"]	1

• 개수의 최댓값은 2이므로 2를 return 합니다.

제출 답안

import Foundation

func solution(_ strArr: [String]) -> Int {
    var lengthCount: [Int: Int] = [:]
    
    // 문자열 길이를 기준으로 그룹화하여 개수를 세기
    for str in strArr {
        let length = str.count
        lengthCount[length, default: 0] += 1
    }
    
    // 가장 많은 그룹의 크기 반환
    return lengthCount.values.max() ?? 0
}

---

4. 배열의 길이에 따라 다른 연산하기

정수 배열 arr과 정수 n이 매개변수로 주어집니다. arr의 길이가 홀수라면 arr의 모든 짝수 인덱스 위치에 n을 더한 배열을, arr의 길이가 짝수라면 arr의 모든 홀수 인덱스 위치에 n을 더한 배열을 return 하는 solution 함수를 작성해 주세요.

---

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
• 1 ≤ arr의 원소 ≤ 1,000
• 1 ≤ n ≤ 1,000

---

입출력 예

arr	n	result
[49, 12, 100, 276, 33]	27	[76, 12, 127, 276, 60]
[444, 555, 666, 777]	100	[444, 655, 666, 877]

---

입출력 예 설명

입출력 예 #1

• 예제 1번의 arr의 길이는 5로 홀수입니다. 따라서 arr의 짝수 인덱스 0, 2, 4에 주어진 n 값인 27을 더하면 [76, 12, 127, 276, 60]이 됩니다. 따라서 [76, 12, 127, 276, 60]를 return 합니다.

입출력 예 #1

• 예제 2번의 arr의 길이는 4로 짝수입니다. 따라서 arr의 홀수 인덱스 1, 3에 주어진 n 값인 100을 더하면 [444, 655, 666, 877]이 됩니다. 따라서 [444, 655, 666, 877]를 return 합니다.

제출 답안

import Foundation

func solution(_ arr: [Int], _ n: Int) -> [Int] {
    var result = arr // 결과 배열 초기화
    let isOdd = arr.count % 2 == 1 // 배열의 길이가 홀수인지 여부 확인
    
    for i in 0..<arr.count {
        // 배열 길이가 홀수이면 짝수 인덱스에 n을 더함
        // 배열 길이가 짝수이면 홀수 인덱스에 n을 더함
        if (isOdd && i % 2 == 0) || (!isOdd && i % 2 == 1) {
            result[i] += n
        }
    }
    
    return result // 수정된 배열 반환
}

---

5. 뒤에서 5등까지

정수로 이루어진 리스트 num_list가 주어집니다. num_list에서 가장 작은 5개의 수를 오름차순으로 담은 리스트를 return하도록 solution 함수를 완성해주세요.

---

제한사항

• 6 ≤ num_list의 길이 ≤ 30
• 1 ≤ num_list의 원소 ≤ 100

---

입출력 예

num_list	result
[12, 4, 15, 46, 38, 1, 14]	[1, 4, 12, 14, 15]

---

입출력 예 설명

입출력 예 #1

• [12, 4, 15, 46, 38, 1, 14]를 정렬하면 [1, 4, 12, 14, 15, 38, 46]이 되고, 앞에서 부터 5개를 고르면 [1, 4, 12, 14, 15]가 됩니다

제출 답안

import Foundation

func solution(_ num_list: [Int]) -> [Int] {
    return Array(num_list.sorted().prefix(5))
}

---

Day 20