[Programmers] Lv2. [3차] 파일명 정렬 | C++

🔗문제 보러가기

프로그래머스

 

👨‍💻풀이 과정

 

• File 클래스를 만들어줍니다.
  • File 클래스엔 id(int), originalName(string), head(string), number(string), tail(string) 멤버 변수가 있습니다.
    • id 값은 문제 입력으로 주어지는 files 배열의 인덱스 순서입니다. head와 number가 같을 때, 정렬 기준으로 사용됩니다.
    • 생성자에서 원본 문자열을 head, number, tail로 잘 분리하여 저장하도록 합니다. 
  • < 연산자 오버로딩을 해줍니다.
    • BST 자료구조인 set에 File 목록들을 저장해 둘 예정입니다.
    • 커스텀 클래스를 set에 저장하려면 < 연산자 오버로딩이 필요합니다.
      • 문제에서 제시된 기준으로 작성해줍니다.
• files 배열을 순회하면서 set<File>에 객체를 등록해줍니다.
• set<File>을 순회하면서 answer 배열에 추가한 후, 리턴합니다.

 

✏️소스 코드 및 결과

#include <string>
#include <vector>
#include <set>
using namespace std;

class File {
public:
    File(int id, const string& fileName) {
        _id = id;
        _originalName = fileName;
        Init();
    }

    string ToLower(const string& str) {
        string result(str);
        for (int i = 0; i < result.length(); i++)
            result[i] = tolower(result[i]);
        return result;
    }

    bool operator<(const File& other) const {
        int dictionaryOrder = _head.compare(other._head);
        
        if (dictionaryOrder == 0) {
            if (_number.empty() or other._number.empty())
                return _id < other._id;

            int myNum = stoi(_number);
            int otherNum = stoi(other._number);
            return myNum != otherNum ? myNum < otherNum : _id < other._id;
        }
            
        return dictionaryOrder < 0;
    }

private:
    void Init() {
        int begin = 0, i = 0, end = _originalName.length();

        /* HEAD 처리 */
        for (; i < end; i++) {
            if (isdigit(_originalName[i])) {
                _head = ToLower(_originalName.substr(begin, i));
                begin = i;
                break;
            }
        }

        if (i == end) {
            _head = ToLower(_originalName.substr(begin, i));
            return;
        }

        /* NUMBER 처리 */
        for (; i < end; i++) {
            if (!isdigit(_originalName[i])) {
                _number = _originalName.substr(begin, i - begin);
                begin = i;
                break;
            }
        }

        if (i == end) {
            _number = _originalName.substr(begin, i - begin);
            return;
        }

        /* TAIL 처리 */
        _tail = _originalName.substr(begin);
    }

public:
    int _id;
    string _originalName;
    string _head;
    string _number;
    string _tail;
};

vector<string> solution(vector<string> files) {
    vector<string> answer;
    set<File> orderedFiles;

    for (int i = 0; i < files.size(); i++)
        orderedFiles.emplace(File(i, files[i]));

    for (const auto file : orderedFiles)
        answer.emplace_back(file._originalName);
    return answer;
}