[BaekJoon] 1431번 : 시리얼 번호

• 이 글은 c++로 풀이를 작성하였습니다.

문제

 

예시

 

풀이

• sort 함수를 구현하는 되는 문제로 원하는 조건에 맞게 구현합니다.
• sort 함수에서 strict weak ordering에 맞게 하지 않으면 런타임 에러가 발생할 수도 있습니다.

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

bool cmp(string& s1, string &s2){
    if(s1.size()<s2.size()){
        return true;
    }else if(s1.size() == s2.size()){
        int sum1 = 0, sum2 = 0;
        for(int i=0;i<s1.size();i++){
            if(isdigit(s1[i])){
                sum1+=(s1[i]-'0');
            }
        }
        
        for(int i=0;i<s2.size();i++){
            if(isdigit(s2[i])){
                sum2+=(s2[i]-'0');
            }
        }
        if(sum1<sum2){
            return true;
        }else if(sum1 == sum2){
            if(s1<s2){
                return true;
            }else{
                return false;
            }
        }else{
            return false;
        }
    }else{
        return false;
    }
}

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

• 추가로 궁금한 점이나 수정할 부분 있으면 댓글로 남겨주세요.