IT-garden

문제 설명

숫자 야구 게임이란 2명이 서로가 생각한 숫자를 맞추는 게임입니다. 게임해보기

각자 서로 다른 1~9까지 3자리 임의의 숫자를 정한 뒤 서로에게 3자리의 숫자를 불러서 결과를 확인합니다. 그리고 그 결과를 토대로 상대가 정한 숫자를 예상한 뒤 맞힙니다.

예를 들어, 아래의 경우가 있으면

이때 가능한 답은 324와 328 두 가지입니다.

질문한 세 자리의 수, 스트라이크의 수, 볼의 수를 담은 2차원 배열 baseball이 매개변수로 주어질 때, 가능한 답의 개수를 return 하도록 solution 함수를 작성해주세요.

제한조건

• 질문의 수는 1 이상 100 이하의 자연수입니다.
• baseball의 각 행은 [세 자리의 수, 스트라이크의 수, 볼의 수] 를 담고 있습니다.

입출력 예

나의 풀이

from itertools import permutations

def solution(baseball):
    a=list(permutations(range(1,10),3))
    r=0
    for i in a:
        ans=i
        c=0
        for b in baseball:
            st=0
            ba=0
            ch=list(map(int,(str(b[0]))))
            for j in range(3):
                if ans[j] == ch[j]:
                    st += 1
            for k in ch:
                if k in ans:
                    ba += 1
            ba -= st
            if b[1] == st and b[2] == ba:
                c += 1
        if c == len(baseball):
            r += 1
    return r

1. 1부터 9까지 중 숫자야구의 숫자가 될 수있는 것을 순열을 이용해 뽑습니다.
2. 해당 숫자가 baseball을 통해 숫자야구의 답이 될 수 있는지 확인합니다.
2-1. 스트라이크의 개수를 구합니다.
2-2. 볼의 개수를 구합니다. 볼의 개수를 구하는 것에 스트라이트도 포함되어 있기 때문에 스트라이크
개수를 뺍니다.
2-3. 스트라이크와 볼의 개수가 baseball의 개수와 같은지 확인합니다.
2-4. 각 baseball의 경우를 계산하고 c가 baseball의 길이와 같으면 숫자야구의 답이 될 수 있기 때문에 r에 1을 더합니다.
3. 순열에 대해 모두 검사하고 r을 리턴합니다.

프로그래머스 '숫자 야구' : https://programmers.co.kr/learn/courses/30/lessons/42841