[백준] 1431번 시리얼 번호 (C++) - sort, compare

문제

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

1431번: 시리얼 번호

코드

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

using namespace std;

bool compare(string s1, string s2){
  if(s1.length()==s2.length()){
    int s1Result = 0;
    int s2Result = 0;
    for(int i=0; i<s1.length(); i++){
      int s1Num = s1[i]-'0';
      int s2Num = s2[i]-'0';
      if(s1Num>=0&&s1Num<10) s1Result+=s1Num;
      if(s2Num>=0&&s2Num<10) s2Result+=s2Num;
    }
    if(s1Result==s2Result) return s1<s2;
    else return s1Result<s2Result;
  }else return s1.length()<s2.length();
}

int main(){
  int n;
  cin >> n;
  vector<string> v;
  for(int i=0; i<n; i++){
    string s;
    cin >> s;
    v.push_back(s);
  }
  sort(v.begin(), v.end(), compare);
  for(int i=0; i<n; i++){
    cout << v[i] << '\n';
  }
  return 0;
}

정리

sort함수의 매개변수로 들어갈 수있는 compare함수를 잘 구현하면 풀 수 있는 문제이다.

참조

https://kdongree.tistory.com/24