[BAEKJOON] 1431 시리얼 번호 (C++)

문제링크

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

1431번: 시리얼 번호

 

문제

시리얼번호가 주어진다.

시리얼번호는 알파벳과 숫자로 이루어진 문자열이다.

시리얼번호를 주어진 조건에 따라 정렬하는 문제이다.

 

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

풀이

풀이 자체는 비교함수를 조건에 맞게 정의하고 sort함수를 이용하여 해결하였다.

 

알게된 것

2번 조건을 푸는 과정에서 atoi함수를 이용해 자리수의 합을 구했는데 예상했던 결과와 다르게 나왔다.

지금까지 stoi는 string 자료형을 int로 바꿔주고, atoi는 char 자료형을 int로 바꿔준다고 알고있었다. 잘못 알고 있었던 것은 atoi가 char 문자 1개만을 변환해준다고 착각했다.

 

atoi의 실제 원리는 매개변수로 들어온 위치부터 숫자가 아닌 문자가 나올 때까지 읽은 후 해당 문자열을 int로 변환해주는 것이다.

 

따라서 문자 한 개를 변환할 때는 s[0]-'0' 과 같이 직접 연산해주어야 한다.

 

코드

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
#define endl "\n"

using namespace std;

int N;
vector<string> V;

bool cmp(string& a, string& b) {
    if (a.length() != b.length()) return a.length() < b.length();
    else {
        int suma = 0, sumb = 0;
        for (int i = 0; i < a.size(); i++) if (!isalpha(a[i])) suma += a[i] - '0';
        for (int i = 0; i < b.size(); i++) if (isdigit(b[i])) sumb += b[i]-'0';

        if (suma != sumb) return suma < sumb;
        else return a < b;
    }
}
void Solve() {
    cin >> N;
    string s;
    for (int i = 0; i < N; i++) {
        cin >> s; 
        V.push_back(s);
    }

    sort(V.begin(), V.end(), cmp);

    for (string ss : V) {
        cout << ss << '\n';
    }
}


int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    Solve();

    return 0;
}