[프로그래머스] 개미군단 (Swift)

https://school.programmers.co.kr/learn/courses/30/lessons/120837

프로그래머스

변수 설정

• general = 5의 공격을 가진 장군개미
• solider = 3의 공격을 가진 병정개미
• ant = 1의 공격을 가진 일반개미
• generalcount = 장군개미수
• solider = 병정개미수
• antcount = 일반 개미수  

내 풀이) 

hp가 general,solider,ant 보다 크거나 같다면

개미의 수를 hp / 공격력으로 구해줌

hp는 hp % 공격력의 나머지

 

count를 모두합해서 반환

import Foundation

func solution(_ hp:Int) -> Int {
    let general = 5
    var generalcount = 0
    let soldier = 3
    var soldiercount = 0
    let ant = 1
    var antcount = 0
    var hp = hp
    
    if hp >= general{
        generalcount = hp / general
        hp = hp % general
    }
    if hp >= soldier{
        soldiercount = hp / soldier
        hp = hp % soldier
    }
    if hp >= ant{
        antcount = hp / ant
        hp = hp % three
    }
 
    return generalcount + soldiercount + antcount
}