카카오 2021 문제 - 신규 아이디 추천(C++)

프로그래머스

출처임.

#include <iostream>
#include <string>
#include <vector>
#include <cctype>
 
using namespace std;
 
string solution(string new_id) {
    string answer = "";
 
    // - 1단계 new_id의 모든 대문자를 대응되는 소문자로 치환합니다.
    for(char& ch : new_id)
        ch = std::tolower(ch); // 대문자를 받으면 소문자로 변환 후 리턴!
    std::cout<<"1단계]"<<new_id<<std::endl;    
    
    // - 2단계 new_id에서 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.)를 제외한 모든 문자를 제거합니다.
    for(char& ch : new_id)
        if((ch >= 'a' && ch <= 'z') || (ch >= '0' && ch <= '9') || ch == '-' || ch == '_' || ch == '.' )
            answer.push_back(ch); // 위 조건에 해당되는 문자들만 넣어서 재구성함.
    std::cout<<"2단계]"<<answer<<std::endl;    
    
    // - 3단계 new_id에서 마침표(.)가 2번 이상 연속된 부분을 하나의 마침표(.)로 치환합니다.
    bool prev_dot_flag = false; // 이전에 .이 나왔었는지 확인하는 flag -> .이 나오면 일반 문자가 나올 때까지 true 상태.
    new_id = answer; // 갱신
    answer.clear();
    
    for(char& ch : new_id)
    {
        if(ch == '.') // . 이 나오면 연속적으로 .인지, 첫 .인지 확인 후, 첫 점이면 대입하고 이후 나오는 점이면 무시함.
        {
            if(prev_dot_flag == false)
            {
                answer.push_back(ch);
                prev_dot_flag = true;
            }
            continue;
        }
        answer.push_back(ch);
        prev_dot_flag = false;
    }
    std::cout<<"3단계]"<<answer<<std::endl;  
    
    // - 4단계 new_id에서 마침표(.)가 처음이나 끝에 위치한다면 제거합니다.
    
    if(answer.front() == '.') // 처음이 점이면
        answer.erase(0,1);
    
    if(answer.back() == '.') // 끝이 점이면
        answer.erase(answer.size()-1,1);
    std::cout<<"4단계]"<<answer<<std::endl;
    
    // - 5단계 new_id가 빈 문자열이라면, new_id에 "a"를 대입합니다.
    if(answer.empty())
        answer.push_back('a');
    std::cout<<"5단계]"<<answer<<std::endl;
    
    // - 6단계 new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
    //        만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제거합니다.
    if(answer.size() >= 16)
    {
        if(answer.at(14) == '.')
            answer = answer.substr(0, 14);
        else
            answer = answer.substr(0, 15);
    } // 그냥 15개를 자르고 마지막 answer.back() == '.' 이면 pop_back() 이렇게 처리해도 된다.
    std::cout<<"6단계]"<<answer<<std::endl;
    
    // - 7단계 new_id의 길이가 2자 이하라면, new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙입니다.
    if(answer.size() <= 2)
        while(answer.size() <= 2)
            answer.push_back(answer.back()); // answer += answer.back(); 이렇게도 가능하다.
    std::cout<<"7단계]"<<answer<<std::endl;
    
    return answer;
}