BOJ 12886 돌 그룹(Python 3)

https://www.acmicpc.net/problem/12886

물통이랑 비슷한 문제입니다.

A,B,C 중복 체크를 확인해주는 것만 생각하면 되는 문제입니다.

A,B의 값을 알고 있다면 C를 구할 수 있으니 2차원 배열로 나타낼 수 있습니다.

(2차원 배열 중복 확인은 A,B를 하든, B,C를 하든 입맛에 맞게 하면 됩니다.)

저는 최솟값, 최댓값으로 중복 체크를 했습니다.

그리고 제 코드는 BFS를 돌리기 전에 A+B+C가 3의 배수가 아니면 0을 출력하고 종료하는 것으로 만들었는데, 이 부분은 BFS 돌리면서 A,B,C의 값이 같을 때 처리해도 무방합니다.

코드확인

from collections import deque
a,b,c=map(int,input().split())
z=a+b+c
if (a+b+c)%3!=0:
    print(0)
    exit()
D=[[0]*(z+1) for _ in range(z+1)]
D[min(a,b,c)][max(a,b,c)]=1
q=deque()
q.append((a,b,c))
while q:
    a,b,c=q.popleft()
    if a==b==c:
        print(1)
        exit()
    for nx,ny in [(a,b),(a,c),(b,c)]:
        if nx<ny:
            nx,ny=ny,nx
        nx=nx-ny
        ny+=ny
        nz=z-nx-ny
        Min=min(nx,ny,nz)
        Max=max(nx,ny,nz)
        if D[Min][Max]==0:
            q.append((nx,ny,nz))
            D[Min][Max]=1
print(0)