카테고리 없음

[프로그래머스 입문문제] Day 9 수학, 문자열, 해시, 완전탐색, 조건문

Devleoper_yh 2025. 1. 21. 23:05
반응형

1. 개미 군단

개미 군단이 사냥을 나가려고 합니다. 개미군단은 사냥감의 체력에 딱 맞는 병력을 데리고 나가려고 합니다. 장군개미는 5의 공격력을, 병정개미는 3의 공격력을 일개미는 1의 공격력을 가지고 있습니다. 예를 들어 체력 23의 여치를 사냥하려고 할 때, 일개미 23마리를 데리고 가도 되지만, 장군개미 네 마리와 병정개미 한 마리를 데리고 간다면 더 적은 병력으로 사냥할 수 있습니다. 사냥감의 체력 hp가 매개변수로 주어질 때, 사냥감의 체력에 딱 맞게 최소한의 병력을 구성하려면 몇 마리의 개미가 필요한지를 return하도록 solution 함수를 완성해주세요.


제한사항

  • hp는 자연수입니다.
  • 0 ≤ hp ≤ 1000

입출력 예

hp result
23 5
24 6
999 201

입출력 예 설명

입출력 예 #1

  • hp가 23이므로, 장군개미 네마리와 병정개미 한마리로 사냥할 수 있습니다. 따라서 5를 return합니다.

입출력 예 #2

  • hp가 24이므로, 장군개미 네마리 병정개미 한마리 일개미 한마리로 사냥할 수 있습니다. 따라서 6을 return합니다.

입출력 예 #3

  • hp가 999이므로, 장군개미 199 마리 병정개미 한마리 일개미 한마리로 사냥할 수 있습니다. 따라서 201을 return합니다.

제출 답안

import Foundation

// hp를 24로 가정
func solution(_ hp:Int) -> Int {
    let generalAnt = 5
    let soldierAnt = 3
    let workAnt = 1
    
    // 24 / 5 = 4 장군 개미는 4마리 필요
    let generalAntCount = hp / generalAnt
    
    // 24 / 5 의 나머지 = 4
    // 장군 개미로 사냥 이후 남은 hp : 4
    let generalAntRemain = hp % generalAnt
    
    // 4 / 3 = 1
    // 병정 개미는 한 마리 필요
    let soldierAntCount = generalAntRemain / soldierAnt
    let soldierAntRemain = generalAntRemain % soldierAnt
    
    // 1 / 1 = 1 일개미는 한마리
    let workAntCount = soldierAntRemain / workAnt
    
    return generalAntCount + soldierAntCount + workAntCount
}

2. 모스부호(1)

머쓱이는 친구에게 모스부호를 이용한 편지를 받았습니다. 그냥은 읽을 수 없어 이를 해독하는 프로그램을 만들려고 합니다. 문자열 letter가 매개변수로 주어질 때, letter를 영어 소문자로 바꾼 문자열을 return 하도록 solution 함수를 완성해보세요.
모스부호는 다음과 같습니다.

morse = { 
    '.-':'a','-...':'b','-.-.':'c','-..':'d','.':'e','..-.':'f',
    '--.':'g','....':'h','..':'i','.---':'j','-.-':'k','.-..':'l',
    '--':'m','-.':'n','---':'o','.--.':'p','--.-':'q','.-.':'r',
    '...':'s','-':'t','..-':'u','...-':'v','.--':'w','-..-':'x',
    '-.--':'y','--..':'z'
}

제한사항

  • 1 ≤ letter의 길이 ≤ 1,000
  • return값은 소문자입니다.
  • letter의 모스부호는 공백으로 나누어져 있습니다.
  • letter에 공백은 연속으로 두 개 이상 존재하지 않습니다.
  • 해독할 수 없는 편지는 주어지지 않습니다.
  • 편지의 시작과 끝에는 공백이 없습니다.

입출력 예

letter result
".... . .-.. .-.. ---" "hello"
".--. -.-- - .... --- -." "python"

입출력 예 설명

입출력 예 #1

  • .... = h
  • . = e
  • .-.. = l
  • .-.. = l
  • --- = o
  • 따라서 "hello"를 return 합니다.

입출력 예 #2

  • .--. = p
  • -.-- = y
  • - = t
  • .... = h
  • --- = o
  • -. = n
  • 따라서 "python"을 return 합니다.

  • a ~ z에 해당하는 모스부호가 순서대로 담긴 배열입니다.
  • {".-","-...","-.-.","-..",".","..-.","--.","....","..",".---","-.-",".-..","--","-.","---",".--.","--.-",".-.","...","-","..-","...-",".--","-..-","-.--","--.."}

제출 답안

import Foundation

func solution(_ letter:String) -> String {
    let morse: [String: String] = [
        ".-": "a", "-...": "b", "-.-.": "c", "-..": "d", ".": "e",
        "..-.": "f", "--.": "g", "....": "h", "..": "i", ".---": "j",
        "-.-": "k", ".-..": "l", "--": "m", "-.": "n", "---": "o",
        ".--.": "p", "--.-": "q", ".-.": "r", "...": "s", "-": "t",
        "..-": "u", "...-": "v", ".--": "w", "-..-": "x", "-.--": "y",
        "--..": "z"
    ]
    return letter
    	// letter를 " "기준으로 나눈다
        .split(separator: " ")
        // 요소를 순회하면서 morse의 알파벳을 가져오고 String으로 변환
        .compactMap { morse[String ( $0 )] }
        // 변환한 문자열을 결합
        .joined()
    
}

3. 가위 바위 보

가위는 2 바위는 0 보는 5로 표현합니다. 가위 바위 보를 내는 순서대로 나타낸 문자열 rsp가 매개변수로 주어질 때, rsp에 저장된 가위 바위 보를 모두 이기는 경우를 순서대로 나타낸 문자열을 return하도록 solution 함수를 완성해보세요.


제한사항

  • 0 < rsp의 길이 ≤ 100
  • rsp와 길이가 같은 문자열을 return 합니다.
  • rsp는 숫자 0, 2, 5로 이루어져 있습니다.

입출력 예

rsp result
"2" "0"
"205" "052"

입출력 예 설명

입출력 예 #1

  • "2"는 가위이므로 바위를 나타내는 "0"을 return 합니다.

입출력 예 #2

  • "205"는 순서대로 가위, 바위, 보이고 이를 모두 이기려면 바위, 보, 가위를 순서대로 내야하므로 “052”를 return합니다.

제출 답안

import Foundation

func solution(_ rsp: String) -> String {
    // 이기는 경우를 매핑하는 딕셔너리
    let winMapping: [Character : Character] = [
        "2" : "0", // 가위를 이기려면 바위
        "0" : "5", // 바위를 이기려면 보
        "5" : "2"  // 보를 이기려면 가위
    ]
    
    // rsp를 한 글자씩 순회하며 변환
    // String.init으로 Character를 String으로 변환
	// 변환된 문자 배열을 결합
    return rsp.map { winMapping[$0].map(String.init) ?? String($0) }.joined()
}

4. 구슬을 나누는 경우의 수

머쓱이는 구슬을 친구들에게 나누어주려고 합니다. 구슬은 모두 다르게 생겼습니다. 머쓱이가 갖고 있는 구슬의 개수 balls와 친구들에게 나누어 줄 구슬 개수 share이 매개변수로 주어질 때, balls개의 구슬 중 share개의 구슬을 고르는 가능한 모든 경우의 수를 return 하는 solution 함수를 완성해주세요.


제한사항

  • 1 ≤ balls ≤ 30
  • 1 ≤ share ≤ 30
  • 구슬을 고르는 순서는 고려하지 않습니다.
  • share  balls

입출력 예

balls share result
3 2 3
5 3 10

입출력 예 설명

입출력 예 #1

  • 서로 다른 구슬 3개 중 2개를 고르는 경우의 수는 3입니다. 
    경우의 수 예시

입출력 예 #2

  • 서로 다른 구슬 5개 중 3개를 고르는 경우의 수는 10입니다.

Hint

  • 서로 다른 n개 중 m개를 뽑는 경우의 수 공식은 다음과 같습니다. 
    Hint
    제출 답안
import Foundation

func solution(_ balls: Int, _ share: Int) -> Int64 {
    // 조합 계산 최적화를 위해 더 작은 값 선택
    // C(n, r) = C(n, n-r)이므로 r과 n-r 중 작은 값을 선택하여 계산량 줄이기
    var min = balls - share < share ? balls - share : share

    // 결과값을 저장할 변수 초기화
    var answer: Int64 = 1

    // 예외 처리: 선택할 개수가 0일 경우 경우의 수는 항상 1
    if min == 0 { return 1 }

    // 반복문을 통해 조합 계산
    // C(n, r) = (n * (n-1) * ... * (n-r+1)) / (r * (r-1) * ... * 1)
    for i in 1...min {
        // 분자를 계산: (balls - min + i)는 n에서 감소하면서 필요한 값만 곱함
        answer *= Int64(balls - min + i)
        // 분모를 나눔: i는 1부터 r까지
        answer /= Int64(i)
    }

    // 최종 계산된 경우의 수 반환
    return answer
}

Day 9

반응형