[프로그래머스 C++] 시저 암호

링크

https://school.programmers.co.kr/learn/courses/30/lessons/12926

프로그래머스

문제 설명 및 제한 조건

나의 코드

#include <string>
#include <vector>

using namespace std;

string solution(string s, int n) {
    string answer = "";
    for (int i = 0; i < s.size(); i++) {
        if (s[i] == ' ')            //공백
            answer += s[i];
        else {
            if (s[i] < 'a') {	    //대문자
                if (s[i] + n > 'Z') 
                    answer += s[i] + n - 26;
                else 
                    answer += s[i] + n;
            }
            else {	                //소문자
                if (s[i] + n > 'z') 
                    answer += s[i] + n - 26;
                else 
                    answer += s[i] + n;
            }
        }
    }
    return answer;
}

채점 결과

다수 코드

#include <string>
#include <vector>

using namespace std;

string solution(string s, int n) {
    string answer = "";

    for(auto &a: s) {
        if(islower(a)) {
            if(!islower(a+=n%26))
                a-=26;
        } else if(isupper(a)) {
            if(!isupper(a+=n%26))
                a-=26;
        }
    }
    return s;
}