#6 프로그래머스 개미군단/문자열안에문자열/제곱수판별

1. 개미군단

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

 

# 개미군단
def solution(hp):
    return hp//5 + (hp%5)//3 + (hp%5)%3

# 다른 사람 풀이
def solution(hp):
    answer = 0
    for ant in [5, 3, 1]:
        d, hp = divmod(hp, ant)
        answer += d
    return answer
#divmod(num1,num2) # num1을 num2 로 나눈 몫과 나머지를 출력하는 함수

 

 

2. 문자열 안에 문자열

문자열 str1, str2가 매개변수로 주어집니다. str1 안에 str2가 있다면 1을 없다면 2를 return하도록 solution 함수를 완성해주세요.

 

# 문자열안에 문자열
def solution(str1,str2):
    if str2 in str1:
        return 1
    else:
        return 2

# 다른 사람 풀이
def solution(str1, str2):
    return 1 + int(str2 not in str1)
    # 참이면 1, 거짓이면 0

def solution(str1, str2):
    if str1.count(str2):
        return 1
    else:
        return 2

 

 

3. 제곱수 판별!

어떤 자연수를 제곱했을 때 나오는 정수를 제곱수라고 합니다. 정수 n이 매개변수로 주어질 때, n이 제곱수라면 1을 아니라면 2를 return하도록 solution 함수를 완성해주세요.

 

# 제곱수 판별하기
def solution(n):
    if n**(1/2) == int(n**(1/2)):
        return 1
    else:
        return 2

# 다른 사람 풀이
def solution(n):
    return 1 if (n ** 0.5).is_integer() else 2