[BOJ] 1744번 수 묶기

문제 바로가기

접근

sort()의 비교 함수를 작성해서 양의 정수, 음의 정수, 0 순서대로 정렬했다. 처음에는 양의 정수끼리와 음의 정수끼리는 무조건 곱하도록 했는데, 생각해보니 1,1의 경우에는 곱하기보다 더하기의 값이 더 컸다. 따라서 arr[i] * arr[i + 1] > arr[i] + arr[i + 1]일 경우에 두 수를 묶도록 고치는 걸로 바꿨다.

#include <iostream>
#include <algorithm>
using namespace std;

#define MAX 10000

bool comparator(const int& a, const int& b) {
  if(a <= 0 && b <= 0)
    return a < b;
  return a > b;
}

int main() {
  std::ios::sync_with_stdio(false);
  int N;
  int sum = 0;
  int arr[MAX];

  cin >> N;
  for(int i = 0; i < N; i++)
    cin >> arr[i];
  sort(arr, arr + N, comparator);

  for(int i = 0; i < N; i++) {
    if(i + 1 < N) {
      if(arr[i] * arr[i + 1] > arr[i] + arr[i + 1])
        sum += arr[i] * arr[++i];
      else
        sum += arr[i];
    }
    else {
      sum += arr[i];
    }
  }

  cout << sum;
}