[프로그래머스] “신규 아이디 추천” C++ 풀이

문제 링크

• https://programmers.co.kr/learn/courses/30/lessons/72410

문제 풀이

• 예제 코드를 참고해, 각 단계별로 문자열 처리

예제 코드

#include <string>
#include <vector>
#include <iostream>
#include <regex>

using namespace std;

string solution(string new_id) {
    string answer = "";

    // 1단계 new_id의 모든 대문자를 대응되는 소문자로 치환합니다.
    for (auto i=0; i<new_id.size(); ++i){
        new_id[i] = tolower(new_id[i]);
    }

    // 2단계 new_id에서 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.)를 제외한 모든 문자를 제거합니다.
    new_id = regex_replace(new_id, regex("[^a-z0-9-_.]+"), "");

    // 3단계 new_id에서 마침표(.)가 2번 이상 연속된 부분을 하나의 마침표(.)로 치환합니다.
    new_id = regex_replace(new_id, regex("[.]+"), ".");

    // 4단계 new_id에서 마침표(.)가 처음이나 끝에 위치한다면 제거합니다.
    if (new_id[0] == '.'){
        new_id.erase(new_id.begin());
    }
    if (new_id[new_id.size()-1] == '.'){
        new_id.erase(new_id.end()-1);
    }

    // 5단계 new_id가 빈 문자열이라면, new_id에 "a"를 대입합니다.
    if (new_id == ""){
        new_id = "a";
    }

    // 6단계 new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
    //      만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제거합니다.
    if (new_id.size()>=16){
        new_id.erase(15);
        if (new_id[new_id.size()-1]=='.'){
            new_id.erase(new_id.end()-1);
        }
    }

    // 7단계 new_id의 길이가 2자 이하라면, new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙입니다.
    if (new_id.size() == 1 || new_id.size() == 2){
        char last_char = new_id[new_id.size()-1];
        for (auto i=new_id.size(); i<3; ++i){
            new_id += last_char;
        }
    }

    answer = new_id;

    return answer;
}