[250106] 프로그래머스 코딩테스트 입문 문제

개미 군단 🐜

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

• 제한사항
  • `hp`는 자연수입니다.
  • 0 ≤ `hp` ≤ 1000
• 입출력 예

hp	result
23	5
24	6
999	201

 

A.

def solution(hp):
    cap_ant_pw = 5
    soldier_ant_pw = 3
    ant_pw = 1
    
    low_hp = hp % cap_ant_pw # 예: 23 % 5 = 3, 24 % 5 = 4
    total_cap_ant = hp // cap_ant_pw # 23 // 5 = 4, 24 // 5 = 4
    total_soldier_ant = low_hp // soldier_ant_pw # 3 // 3 = 1, 4 // 3 = 1
    total_ant = (low_hp % soldier_ant_pw) * ant_pw # (3 % 3) * 1 = 0, (4 % 3) * 1 = 1

    return total_cap_ant + total_soldier_ant + total_ant # 4 + 1 + 0, 4 + 1 + 1

# {장군개미 공격력 : 5, 병정개미 : 3, 일개미 : 1}
# if 체력 23 여치사냥:
#     일개미23마리 == (장군개미4마리 + 병정개미1마리 --> 가성비는 이쪽이 더 좋음)
# 사냥감 체력 변수 hp
# 사냥감 hp에 맞는 ***최소한의 병력 구성*** 을 위한 개미 숫자
# hp는 자연수, 0부터 1000 범위
# 예) hp = 23 : ant = 총 5마리, hp = 24 : ant = 6
# 뭘 해야 하는가
# hp를 ant로 나누고 나머지 계산하기
# 근데 이제 가성비 따져야 하니까 ant의 기본값을 5 라고 생각해야함

---

오답노트 ✔️

처음 제 코드는 이랬어요,,

def solution(hp):
    cap_ant_pw = 5
    soldier_ant_pw = 3
    ant_pw = 1

    while hp:
        low_hp = hp % cap_ant_pw # 예: 23 % 5 = 3, 24 % 5 = 4
        total_cap_ant = hp // cap_ant_pw # 23 // 5 = 4, 24 // 5 = 4
        total_soldier_ant = low_hp // soldier_ant_pw # 3 // 3 = 1, 4 // 3 = 1
        total_ant = (low_hp % soldier_ant_pw) * ant_pw # (3 % 3) * 1 = 0, (4 % 3) * 1 = 1

        if low_hp != 0:
            return total_cap_ant + total_soldier_ant + total_ant # 4 + 1 + 0, 4 + 1 + 1
        else:
            return total_cap_ant + total_soldier_ant + total_ant

테스트 케이스를 돌려봤을 때는 모두 통과돼서 막상 제출을 해보면

1개의 케이스에서 불통이 되어버리길래 내 코드에서 개선해야할 점이 무엇인지 알아보았어요

---

q. 위 코드의 개선점은 어떤 것인지?

 

해당 코드는 개미 병력을 효율적으로 구성하여 사냥감의 체력을 소모하는 문제에 대한 해결책을
제시하고 있습니다만, 몇 가지 개선할 수 있는 점이 있습니다.

개선 사항

불필요한 반복문 (while hp)

• 주어진 코드에서 while hp: 반복문은 hp가 0이 될 때까지 반복되도록 작성되어 있지만, 실제로는 hp가 0이 되는 경우는 없습니다.
• 함수 내에서 return 문장이 실행되면 함수가 즉시 종료되므로, 반복문은 한 번만 실행됩니다.
• 따라서 while 반복문은 필요하지 않으며, 제거해도 코드는 동일하게 동작합니다.

if 문 조건

• 현재 코드에서는 if low_hp != 0: 조건을 사용하고 있지만, low_hp가 0이 아닌 경우와 0인 경우 모두 동일한 계산 결과(total_cap_ant + total_soldier_ant + total_ant)를 반환합니다. 따라서 if 문은 불필요합니다.
• return 문을 반복문 밖으로 빼서 한 번만 실행되도록 수정할 수 있습니다.

---

# 수정 된 코드
def solution(hp):
    cap_ant_pw = 5
    soldier_ant_pw = 3
    ant_pw = 1
    
    low_hp = hp % cap_ant_pw
    total_cap_ant = hp // cap_ant_pw
    total_soldier_ant = low_hp // soldier_ant_pw
    total_ant = (low_hp % soldier_ant_power) * ant_pw
    
    return total_cap_ant + total_soldier_ant + total_ant

개선된 코드의 장점

• 코드 간결성: 불필요한 반복문과 조건문을 제거하여 코드의 길이가 짧아지고 가독성이 향상되었습니다.
• 효율성: 불필요한 연산(반복문, 조건문, 형변환)이 제거되어 코드가 더욱 효율적으로 실행됩니다.

결론

개선된 코드는 원래 코드의 기능은 유지하면서 불필요한 부분을 제거하여 더욱 간결하고 효율적으로 만들어졌습니다. 코드를 작성할 때 불필요한 반복문, 조건문, 형변환을 제거하여 코드의 효율성과 가독성을 높이는 것이 중요합니다.

 

이번 문제풀이를 통해 배운 것:
아무때나 반복문을 쓴다고 되는게 아니다~!