본문 바로가기
코딩테스트

프로그래머스 입문 Day 9 파이썬

by 피갓자 2025. 5. 5.

프로그래머스
프로그래머스

개미 군단

문제 설명

개미 군단이 사냥을 나가려고 합니다. 개미군단은 사냥감의 체력에 딱 맞는 병력을 데리고 나가려고 합니다.

장군개미는 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합니다.

solution.py

def solution(hp):
    answer = 0
    answer += hp // 5
    hp %= 5
    answer += hp // 3
    hp %= 3
    answer += hp
    return answer

모스부호 (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에 해당하는 모스부호가 순서대로 담긴 배열입니다.
  • {".-","-...","-.-.","-..",".","..-.","--.","....","..",".---","-.-",".-..","--","-.","---",".--.","--.-",".-.","...","-","..-","...-",".--","-..-","-.--","--.."}

solution.py

def solution(letter):
    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'
    }

    return ''.join(morse[i] for i in letter.split())

가위 바위 보

문제 설명

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

solution.py

def solution(rsp):
    answer = {'2':'0', '0':'5', '5':'2'}
    return ''.join(answer[i] for i in rsp)

구슬을 나누는 경우의 수

문제 설명

머쓱이는 구슬을 친구들에게 나누어주려고 합니다. 구슬은 모두 다르게 생겼습니다.

머쓱이가 갖고 있는 구슬의 개수 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입니다. 
    서로 다른 구슬 3개 중 2개를 고르는 경우의 수

입출력 예 #2

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


Hint

  • 서로 다른 n개 중 m개를 뽑는 경우의 수 공식은 다음과 같습니다. 
    서로 다른 n개 중 m개를 뽑는 경우의 수

solution.py

def factorial(n):
    result = 1
    for i in range(1, n + 1):
        result *= i
    return result

def solution(balls, share):
    return factorial(balls) // (factorial(share) * factorial(balls - share))