[백준 1431] 시리얼 번호

백준 1431번 - 시리얼 번호

백준 1431번

❓ 문제

다솜이는 기타를 많이 가지고 있다. 그리고 각각의 기타는 모두 다른 시리얼 번호를 가지고 있다. 다솜이는 기타를 빨리 찾아서 빨리 사람들에게 연주해주기 위해서 기타를 시리얼 번호 순서대로 정렬하고자 한다.

모든 시리얼 번호는 알파벳 대문자 (A-Z)와 숫자 (0-9)로 이루어져 있다.

시리얼번호 A가 시리얼번호 B의 앞에 오는 경우는 다음과 같다.

1. A와 B의 길이가 다르면, 짧은 것이 먼저 온다.
2. 만약 서로 길이가 같다면, A의 모든 자리수의 합과 B의 모든 자리수의 합을 비교해서 작은 합을 가지는 것이 먼저온다. (숫자인 것만 더한다)
3. 만약 1,2번 둘 조건으로도 비교할 수 없으면, 사전순으로 비교한다. 숫자가 알파벳보다 사전순으로 작다.
   시리얼이 주어졌을 때, 정렬해서 출력하는 프로그램을 작성하시오.

---

✏️ 풀이

algorithm 라이브러리에서 지원하는 sort 함수를 이용한 풀이.

기본적으로 vector 형태의 벡터를 선언했을 때, 이를 sorting하려면, sort(vector.begin(), vector.end())를 써주면 된다.

💡 sort함수는 기본적으로 오름차순으로 정렬해준다.

하지만 내림차순, 특수한 정렬에 대해서는 직접 compare함수를 정의해줄 수도 있다.

sort(vector.begin(), vector.end(), compare) 이런식으로 작성을 하고, compare 함수에 대해 로직을 짜주면 된다.

이 문제의 조건으로 예시를 보자!

1. A와 B의 길이가 다르면, 짧은 것이 먼저 온다.
2. 만약 서로 길이가 같다면, A의 모든 자리수의 합과 B의 모든 자리수의 합을 비교해서 작은 합을 가지는 것이 먼저온다. (숫자인 것만 더한다)
3. 만약 1,2번 둘 조건으로도 비교할 수 없으면, 사전순으로 비교한다. 숫자가 알파벳보다 사전순으로 작다.

위와 같은 조건이 있기 때문에, 위의 조건을 모두 써주었다.

아래는 정답 코드 중, compare 함수만을 뽑아온 것이다.

...
sort(serial_number.begin(), serial_number.end(), compare);
...

bool compare(string a, string b) {
  if (a.length() == b.length()) {
    if (sum(a) == sum(b)) {
      return a < b;
    }
    return sum(a) < sum(b);
  }
  return a.length() < b.length();
}

int sum(string temp) {
  int sum=0;
  for (int i=0; i<temp.length(); i++) {
    if (temp[i] >= 48 && temp[i] <=57)
      sum+= temp[i] - 48;
  }
  return sum;
}

이렇게, 사용자가 직접 sort condition을 정의해줄 수 있다는 사실을 기억하자!

---

✔️ 정답 코드

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

vector<string> serial_number;

bool compare(string a, string b);
int sum(string temp);

int main(void) {
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);

  int N;
  cin >> N;
  
  for (int i=0; i<N; i++) {
    string temp;
    cin >> temp;
    serial_number.push_back(temp);
  }

  sort(serial_number.begin(), serial_number.end(), compare);

  vector<string>::iterator it;
  for (it=serial_number.begin(); it!=serial_number.end(); it++) {
    cout << *it;
    if (it != serial_number.end())
      cout << "\n";
  }
  return 0;
}

bool compare(string a, string b) {
  if (a.length() == b.length()) {
    if (sum(a) == sum(b)) {
      return a < b;
    }
    return sum(a) < sum(b);
  }
  return a.length() < b.length();
}

int sum(string temp) {
  int sum=0;
  for (int i=0; i<temp.length(); i++) {
    if (temp[i] >= 48 && temp[i] <=57)
      sum+= temp[i] - 48;
  }
  return sum;
}