[C++] 백준(BOJ) - 1421번 시리얼 번호

풀이 코드

c++의 sort 메소드를 사용하여 간단히 풀 수 있다.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int sum(string s){
  int l = s.size();
  int result=0;
  for(int i=0;i<l;i++){
    if((s[i] - '0')>=1 && (s[i]-'0')<=9)
      result += (s[i]-'0');
  }
  return result;
}

bool cmp(string a, string b){
  //1번조건
  if(a.size() != b.size())
    return a.size() < b.size();
  else{
    //2번조건
    if(sum(a) != sum(b))
      return sum(a)<sum(b);
      //3번조건(알파벳순정렬)
    else
      return a<b;
  }
}
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  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;
}

문제

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

1431번: 시리얼 번호