[프로그래머스 기초문제] Day 24 조건문 활용, 반복문 활용, 이차원 리스트(배열)

1. 커피 심부름

팀의 막내인 철수는 아메리카노와 카페 라테만 판매하는 카페에서 팀원들의 커피를 사려고 합니다. 아메리카노와 카페 라테의 가격은 차가운 것과 뜨거운 것 상관없이 각각 4500, 5000원입니다. 각 팀원에게 마실 메뉴를 적어달라고 하였고, 그 중에서 메뉴만 적은 팀원의 것은 차가운 것으로 통일하고 "아무거나"를 적은 팀원의 것은 차가운 아메리카노로 통일하기로 하였습니다.

각 직원이 적은 메뉴가 문자열 배열 order로 주어질 때, 카페에서 결제하게 될 금액을 return 하는 solution 함수를 작성해주세요. order의 원소는 아래의 것들만 들어오고, 각각의 의미는 다음과 같습니다.

order의 원소	의미
"iceamericano", "americanoice"	차가운 아메리카노
"hotamericano", "americanohot"	따뜻한 아메리카노
"icecafelatte", "cafelatteice"	차가운 카페 라떼
"hotcafelatte", "cafelattehot"	따뜻한 카페 라떼
"americano"	아메리카노
"cafelatte"	카페 라떼
"anything"	아무거나

---

제한사항

• 1 ≤ order의 길이 ≤ 1,000

---

입출력 예

order	result
["cafelatte", "americanoice", "hotcafelatte", "anything"]	19000
["americanoice", "americano", "iceamericano"]	13500

---

입출력 예 설명

입출력 예 #1

• 예제 1번의 입력을 표로 나타내면 다음과 같습니다.

order[i]	팀원의 적은 음료	실제 주문 음료
"cafelatte"	카페 라테	차가운 카페 라떼
"americanoice"	차가운 아메리카노	차가운 아메리카노
"hotcafelatte"	따뜻한 카페 라떼	따뜻한 카페 라뗴
"anything"	아무거나	차가운 아메리카노

철수는 차가운 아메리카노 2잔, 차가운 카페 라테 1잔, 따뜻한 카페 라테 1잔을 주문하게 되므로 결제 금액은 4500 × 2 + 5000 × 2 = 19000원입니다. 따라서 19000을 return 합니다.

입출력 예 #2

• 예제 2번의 입력에서는 모든 음료를 차가운 아메리카노로 시키게 됩니다. 따라서 결제 금액은 4500 × 3 = 13500원이고 13500을 return 합니다.

제출 답안

import Foundation

func solution(_ order: [String]) -> Int {
    var totalPrice = 0 // 총 결제 금액을 저장할 변수
    
    for coffee in order {
        switch coffee {
        // 아메리카노 종류 (차가운 아메리카노, 뜨거운 아메리카노 포함)
        // "anything"은 차가운 아메리카노로 간주
        case "americano", "iceamericano", "americanoice", "hotamericano", "americanohot", "anything":
            totalPrice += 4500
        
        // 카페 라테 종류 (차가운 카페 라테, 뜨거운 카페 라테 포함)
        case "cafelatte", "icecafelatte", "cafelatteice", "hotcafelatte", "cafelattehot":
            totalPrice += 5000
        
        default:
            break // 예외적인 경우는 없지만, 안전하게 처리
        }
    }
    
    return totalPrice // 총 결제 금액 반환
}

---

2. 그림 확대

직사각형 형태의 그림 파일이 있고, 이 그림 파일은 1 × 1 크기의 정사각형 크기의 픽셀로 이루어져 있습니다. 이 그림 파일을 나타낸 문자열 배열 picture과 정수 k가 매개변수로 주어질 때, 이 그림 파일을 가로 세로로 k배 늘린 그림 파일을 나타내도록 문자열 배열을 return 하는 solution 함수를 작성해 주세요.

---

제한사항

• 1 ≤ picture의 길이 ≤ 20
• 1 ≤ picture의 원소의 길이 ≤ 20
• 모든 picture의 원소의 길이는 같습니다.
• picture의 원소는 '.'과 'x'로 이루어져 있습니다.
• 1 ≤ k ≤ 10

제출 답안

import Foundation

func solution(_ picture: [String], _ k: Int) -> [String] {
    var enlargedPicture: [String] = [] // 변환된 이미지를 저장할 배열
    
    for row in picture {
        // 각 픽셀을 k번 반복하여 가로 크기를 k배 확장
        let enlargedRow = row.map { String(repeating: $0, count: k) }.joined()
        
        // 해당 행을 k번 추가하여 세로 크기를 k배 확장
        for _ in 0..<k {
            enlargedPicture.append(enlargedRow)
        }
    }
    
    return enlargedPicture // 최종적으로 변환된 이미지 반환
}

---

3. 조건에 맞게 수열 변환하기 3

정수 배열 arr와 자연수 k가 주어집니다. 

만약 k가 홀수라면 arr의 모든 원소에 k를 곱하고, k가 짝수라면 arr의 모든 원소에 k를 더합니다.

이러한 변환을 마친 후의 arr를 return 하는 solution 함수를 완성해 주세요.

---

제한사항

• 1 ≤ arr의 길이 ≤ 1,000,000
  • 1 ≤ arr의 원소의 값 ≤ 100
• 1 ≤ k ≤ 100

---

입출력 예

arr	k	result
[1, 2, 3, 100, 99, 98]	3	[3, 6, 9, 300, 297, 294]
[1, 2, 3, 100, 99, 98]	2	[3, 4, 5, 102, 101, 100]

---

입출력 예 설명

입출력 예 #1

• 주어진 k인 3은 홀수이므로, 전체 배열에 3을 곱합니다. 따라서 [3, 6, 9, 300, 297, 294]을 return 합니다.

입출력 예 #2

• 주어진 k인 2는 짝수이므로, 전체 배열에 2를 더합니다. 따라서 [3, 4, 5, 102, 101, 100]을 return 합니다.

제출 답안

import Foundation

func solution(_ arr: [Int], _ k: Int) -> [Int] {
    // k가 홀수이면 각 원소에 k를 곱하고, 짝수이면 k를 더함
    return arr.map { k % 2 == 1 ? $0 * k : $0 + k }
}

---

4. I로 만들기

알파벳 소문자로 이루어진 문자열 myString이 주어집니다. 알파벳 순서에서 "l"보다 앞서는 모든 문자를 "l"로 바꾼 문자열을 return 하는 solution 함수를 완성해 주세요.

---

제한사항

• 1 ≤ myString ≤ 100,000
  • myString은 알파벳 소문자로 이루어진 문자열입니다.

---

입출력 예

myString	result
"abcdevwxyz"	"lllllvwxyz"
"jjnnllkkmm"	"llnnllllmm"

 

입출력 예 설명

입출력 예 #1

• 0 ~ 4번 인덱스의 문자 "a","b","c","d","e"는 각각 "l"보다 앞서는 문자입니다. 따라서 "l"로 고쳐줍니다.
• 그 외의 문자는 모두 "l"보다 앞서지 않는 문자입니다. 따라서 바꾸지 않습니다.
• 따라서 "lllllvwxyz"을 return 합니다.

입출력 예 #2

• 0번, 1번, 6번, 7번 인덱스의 문자 "j","j","k","k"는 각각 "l"보다 앞서는 문자입니다. 따라서 "l"로 고쳐줍니다.
• 그 외의 문자는 모두 "l"보다 앞서지 않는 문자입니다. 따라서 바꾸지 않습니다.
• 따라서 "llnnllllmm"을 return 합니다.

제출 답안

import Foundation

func solution(_ myString: String) -> String {
    // 각 문자가 l보다 앞서는 알파벳인 경우 l로 변환
    return myString.map { $0 < "l" ? "l" : String($0) }.joined()
}

---

5. 특별한 이차원 배열 1

정수 n이 매개변수로 주어질 때, 다음과 같은 n × n 크기의 이차원 배열 arr를 return 하는 solution 함수를 작성해 주세요.

• arr[i][j] (0 ≤ i, j < n)의 값은 i = j라면 1, 아니라면 0입니다.

---

제한사항

• 1 ≤ n ≤ 100

---

제출 답안

import Foundation

func solution(_ n: Int) -> [[Int]] {
    // n × n 크기의 이차원 배열을 만들어 대각선에만 1을 배치하고 나머지는 0
    return (0..<n).map { i in
        (0..<n).map { j in
            return i == j ? 1 : 0
        }
    }
}

---