[백준 / C++] 1431번: 시리얼 번호 풀이

문제 설명

 

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

1431번: 시리얼 번호

 

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

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

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

 

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

 

시리얼이 주어졌을 때, 정렬해서 출력하는 프로그램을 작성하시오.

 

 

 

 

문제 풀이

 

비교 기준 1, 2, 3을 충족하는 compare 함수를 작성하면 됩니다. 

 

한자리 수 정수를 표현하는 char 자료형 데이터를 int 자료형으로 간단하게 변환하는 방법은 다음과 같습니다. 

char c = '9';
int num = c - '0';
cout << num; // 9

소스 코드 주석에 문제 풀이 추가 설명을 적어놓았습니다. 

 

 

 

소스 코드

 

// 1431번: 시리얼 번호 풀이
#include <iostream>
#include <algorithm>
#include <vector>
 
using namespace std;
 
int n;
vector<string> guitars;
 
// 문자열 포함된 숫자의 합을 구하는 함수
int getSum(string a) {
    int len = a.length(), sum = 0;
    for (int i = 0; i < len; i++) {
        // 숫자만 sum에 더해줌
        if (a[i] - '0' <= 9 && a[i] - '0' >= 0)
            sum += a[i] - '0';
    }
 
    return sum;
}
 
// 비교 함수
bool compare(string a, string b) {
    // 길이가 다르면 짧은 것이 먼저
    if (a.length() != b.length()) {
        return a.length() < b.length();
    }
    else {
        int aSum = getSum(a);
        int bSum = getSum(b);
 
        // 숫자의 합이 작은 것 먼저
        if (aSum != bSum) {
            return aSum < bSum;
        }
        // 사전 순서대로 정렬
        else {
            return a < b;
        }
    }
}
 
int main() {
    cin >> n;
    string input;
 
    for (int i = 0; i < n; i++) {
        cin >> input;
        guitars.push_back(input);
    }
 
    sort(guitars.begin(), guitars.end(), compare);
 
    for (int i = 0; i < n; i++)
        cout << guitars[i] << endl;
    
    return 0;
}

 

 

 

 

 

 

공부한 것을 정리한 내용입니다. 수정할 부분이 있다면 알려주시면 감사하겠습니다 :)