문제: 1431번: 시리얼 번호
basic-algo-lecture/workbook/0x0F.md at master · encrypted-def/basic-algo-lecture
basic-algo-lecture/workbook/0x0F.md at master · encrypted-def/basic-algo-lecture
바킹독의 실전 알고리즘 강의 자료. Contribute to encrypted-def/basic-algo-lecture development by creating an account on GitHub.
github.com
#include <queue>
#include <iostream>
#include <algorithm>
using namespace std;
int sums(string str) {
int sum = 0;
for (int i = 0; i < str.size(); i++) {
if ('0' <= str[i] && str[i] <= '9')
sum += str[i] - '0'; // 진짜 수 더하기
}
return sum;
}
bool compare(string a, string b) {
if (a.size() == b.size()) {
// 자리수 합 비교해서 작은 값 먼저 오도록 정렬
if (sums(a) != sums(b))
return sums(a) < sums(b);
else {
return a < b; // 사전순 비교를 직접 수행
}
}
else {
// 사이즈 작은것부터 정렬..
return a.size() < b.size();
}
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n;
cin >> n;
vector<string> gitars(n);
for (int i = 0; i < n; i++)
cin >> gitars[i];
sort(gitars.begin(), gitars.end(), compare);
for (int i = 0; i < n; i++)
cout << gitars[i] << "\n";
return 0;
}
;; 그냥 아스키 수로 더해도 될 줄 알았는데 그게 아니었다.
원래 처음에 생각했던 건 0~9 에서 9로 갈 때 어차피 아스키 값도 커지니까 그냥 더해도 된다는 거였다. 근데 간과한 점이 있는데 0을 더하는 건 총 합에 아무런 영향도 주지 않는데 아스키 값을 더해버리면 영향을 주게된다;;;
흠.. 객기 부리지 말고 그냥 처음부터 '0' 빼서 수로 더해줄걸 이라는 생각이 들었다.
참고자료:
[바킹독의 실전 알고리즘] 0x0F강 - 정렬 II - YouTube
-> 위 글 덕분에 틀린 이유 찾을 수 있었다;;