[백준 / python] #11880 개미

 

11880번: 개미

 

문제 풀이

육면체를 펼쳐서 피타고라스 정리를 이용하여 대각선의 제곱을 구하는 문제이다. 경우의 수는 3가지가 가능하다.

처음에는 그냥 냅다 모든 경우를 ans1, ans2, ans3 변수를 할당하여, 이 중에서 가장 작은 값을 구하는 문제라고 생각했는데, 이 방식보다 더 효율적으로 풀어낼 수 없을까 고민하다가 순열조합을 적용해보자 생각했다. 로직상 숫자가 변수에 할당되는 순서에 따라 값이 달라지므로 순열이 적합할 것이다.

 

제출 코드

from itertools import permutations as pm
import sys
input = sys.stdin.readline

T = int(input())
for _ in range(T):
    arr = list(map(int, input().split()))
    ans = list()
    for a, b, c in pm(arr, 3):
        res = a ** 2 + (b + c) ** 2
        ans.append(res)
    print(min(ans))

 

느낀 점

항상 냅다 접근하지 말고, 문제를 보면서 종이에 필기해보며 파악하는 시간이 필요할 것 같다. 이를 통해 해당 문제에 적용할 수 있는 더 적합한 알고리즘이 무엇일지 판단할 수 있을 것으로 보인다.