[알고리즘] 백준 10819 - 차이를 최대로 (파이썬)

 

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

10819번: 차이를 최대로

 

문제

N개의 정수로 이루어진 배열 A가 주어진다. 이때, 배열에 들어있는 정수의 순서를 적절히 바꿔서 다음 식의 최댓값을 구하는 프로그램을 작성하시오.

|A[0] - A[1]| + |A[1] - A[2]| + ... + |A[N-2] - A[N-1]|

입력

첫째 줄에 N (3 ≤ N ≤ 8)이 주어진다. 둘째 줄에는 배열 A에 들어있는 정수가 주어진다. 배열에 들어있는 정수는 -100보다 크거나 같고, 100보다 작거나 같다.

출력

첫째 줄에 배열에 들어있는 수의 순서를 적절히 바꿔서 얻을 수 있는 식의 최댓값을 출력한다.

풀이

서로 다른 N개의 숫자를 조합하는 모든 경우의 수(N의 최댓값이 8이기 때문에 최대 40320개의 경우의 수가 존재할 수 있다.)를 구한 후에 각 경우마다의 원소들의 차의 절댓값을 ans에 넣고 최댓값을 추출해주었다.

from sys import stdin as s
from itertools import permutations

s = open("input.txt", "rt")

n = int(s.readline())
data = [int(a) for a in s.readlines()[0].split()]
arr = list(permutations(data, n))
ans = []

for x in arr:
    sum = 0

    for j in range(n - 1):
        sum += abs(x[j] - x[j+1])

    ans.append(sum)


print(max(ans))