[프로그래머스] Level2 - 타겟 넘버

 

---

문제 설명

n개의 음이 아닌 정수가 있습니다. 이 수를 적절히 더하거나 빼서 타겟 넘버를 만들려고 합니다. 예를 들어 [1, 1, 1, 1, 1]로 숫자 3을 만들려면 다음 다섯 방법을 쓸 수 있습니다.

-1+1+1+1+1 = 3

+1-1+1+1+1 = 3

+1+1-1+1+1 = 3

+1+1+1-1+1 = 3

+1+1+1+1-1 = 3

사용할 수 있는 숫자가 담긴 배열 numbers, 타겟 넘버 target이 매개변수로 주어질 때 숫자를 적절히 더하고 빼서 타겟 넘버를 만드는 방법의 수를 return 하도록 solution 함수를 작성해주세요.

 

제한사항

• 주어지는 숫자의 개수는 2개 이상 20개 이하입니다.
• 각 숫자는 1 이상 50 이하인 자연수입니다.
• 타겟 넘버는 1 이상 1000 이하인 자연수입니다.

입출력 예

numbers	target	return
[1, 1, 1, 1, 1]	3	5

입출력 예 설명

문제에 나온 예와 같습니다.

---

알고리즘

1. number 리스트의 합을 구함

2. 

 

 

필요 함수 정리

최대값

• max()
  • 형식 : max(x)
  • x는 iterable한 자료형이어야함
  • 인수로 받은 자료형 내에서 최소값을 찾아서 반환하는 함수

 

 

범위

• range()
  • 형식 : range(시작숫자, 종료숫자, step)
  • 시작숫자부터 종료숫자 바로 앞 숫자까지 컬렉션을 만듬

 

 

 

 

내 소스 코드

def solution(answers):
    score = [0,0,0]
    answer = []
    s1 = [1,2,3,4,5]
    s2 = [2,1,2,3,2,4,2,5]
    s3 = [3,3,1,1,2,2,4,4,5,5]
    
    for i in range(len(answers)):
        if answers[i] == s1[(i%5)] : score[0]+=1
        if answers[i] == s2[(i%8)] : score[1]+=1
        if answers[i] == s3[(i%10)] : score[2]+=1
            
    for i in range(3):
        if max(score) == score[i] : answer.append(i+1)
            
    return answer;

 

결과 

 

 

 

다른 사람 소스 코드

1. enumerate

def solution(answers):
    p = [[1, 2, 3, 4, 5],
         [2, 1, 2, 3, 2, 4, 2, 5],
         [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]]
    s = [0] * len(p)

    for q, a in enumerate(answers):
        for i, v in enumerate(p):
            if a == v[q % len(v)]:
                s[i] += 1
    return [i + 1 for i, v in enumerate(s) if v == max(s)]

* enumerate