10819 차이를 최대로

문제

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

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

입력

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

출력

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

예제 입력 1

Plain Text
복사

예제 출력 1

Plain Text
복사

My solution

20 1 15 8 4 10

|19|+|-14|+|7|+|4|+|-6|

= 50

const getSum = arr => {
  let sum = 0;
  for (let i = 0; i < N - 1; i++) {
    sum += Math.abs(arr[i] - arr[i + 1]);
  }
  return sum;
};
JavaScript
복사

순열을 구하는 기존의 함수를 가지고 N개만큼 뽑은 순열이 될 때마다 위의 함수로 합을 구한다.

합을 구했을 때는 최댓값인지 확인하여 max를 업데이트한다.

let max = Number.MIN_SAFE_INTEGER;
const permutation = (perm, L) => {
  if (L === N) { // N이 6이라면 N만큼 뽑았을 때 합을 구해 최댓값 업데이트
    max = Math.max(max, getSum(perm));
    return;
  }
  for (let i = 0; i < N; i++) {
    if (!visit[i]) {
      visit[i] = true;
      perm[L] = A[i]; // 0부터 차례대로 배열에 요소 추가
      permutation(perm, L + 1);
      visit[i] = false;
    }
  }
};
permutation([], 0); // 재귀호출을 시작할 때 배열에 사용할 초기값 [], 변수 선언 불필요
JavaScript
복사

Code

const solution = function (i) {
  const [N, ...A] = i.toString().trim().split(/\s/).map(Number);
  const visit = [...Array(N)].fill(false);

  const getSum = arr => {
    let sum = 0;
    for (let i = 0; i < N - 1; i++) {
      sum += Math.abs(arr[i] - arr[i + 1]);
    }
    return sum;
  };

  let max = Number.MIN_SAFE_INTEGER;
  const permutation = (perm, L) => {
    if (L === N) {
      max = Math.max(max, getSum(perm));
      return;
    }
    for (let i = 0; i < N; i++) {
      if (!visit[i]) {
        visit[i] = true;
        perm[L] = A[i];
        permutation(perm, L + 1);
        visit[i] = false;
      }
    }
  };

  permutation([], 0);
  console.log(max);
  return max;
};

describe('차이를최대로', () => {
  it('TC1', () => {
    expect(
      solution(`6
20 1 15 8 4 10`)
    ).toStrictEqual(62);
  });
});
JavaScript
복사