[BOJ] 백준 #1431. 시리얼 번호 (C++)

🎨 문제

문제 링크: https://www.acmicpc.net/problem/1431

• 알고리즘 분류: 정렬
• 난이도: Silver 3

 

💬 풀이

1. string을 여러 개 담은 vector, 즉 vector<string>에 입력을 받고,

2. 문제 조건에 따라 cmp()를 정의해 sort해준다.

 

맞왜틀(맞는데 왜 자꾸 틀리다고 하지)을 한 945098번 하고 나서야

return s1<s2;에서 return을 빼먹었다는 걸 찾아냈다. ㅎㅎㅎㅎㅎㅎㅎㅎㅎㅎㅎㅎㅎㅎㅎㅎㅎ하. 바본가 진짜

혹시 const가? 아님 설마 &때문에......? 등등 도대체 어디가 틀린건지 알아보면서 덕분에 개념 복습을 할 수 있었다. 굿^^,,

 

 

👩‍💻 코드

C++

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;

int getTotal(const string &s) {
	int total = 0;
	for (int i = 0; i < s.size(); i++) {
		if ('0' <= s[i] && s[i] <= '9') total += (s[i] - '0');
	}
	return total;
}

bool cmp(const string &s1, const string &s2) {
	int len1 = s1.size(), len2 = s2.size();
	if (len1 == len2) {
		int total1 = getTotal(s1);
		int total2 = getTotal(s2);
		if (total1 == total2) return s1 < s2;
		else return total1 < total2;
	}
	else return len1 < len2;
}

int main(void) {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	//freopen("input.txt", "rt", stdin);

	int N;  cin >> N;
	vector<string> v(N);
	for(int i = 0; i < N; i++) {
		cin >> v[i];
	}

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

	for (int i = 0; i < N; i++) {
		cout << v[i] << '\n';
	}

	return 0;
}