백준 1431 시리얼 번호

치즈·2022년 11월 27일

BOJ

목록 보기
21/45
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>

using namespace std;
int N;
vector<string> v;

int digit(string a){
  int ret = 0;
  for(int i = 0; i < a.length(); i++){
    int tmp = (int) a[i];
    if(48 <= tmp && tmp <= 57){
      //숫자이면
      ret += (tmp - 48);
    }
  }
  return ret;
}

void input(){
  cin >> N;
  for(int i = 0; i < N; i++){
    string s;
    cin >> s;
    v.push_back(s);
  }
}

bool cmp(string a, string b){
  if(a.length() != b.length()) return a.length() < b.length();
  else{
    if(digit(a) != digit(b)) return digit(a) < digit(b);
    else return a < b;
  }
}
void solve(){
  sort(v.begin(), v.end(), cmp);
  for(int i = 0; i < N; i++){
    cout << v[i] << "\n";
  }
}

int main(void){
  ios_base::sync_with_stdio(false);   
  cin.tie(0);
  cout.tie(0);
  input();
  solve();
  return 0;
}

profile
차근차근 배워나가요

0개의 댓글