코딩 공부 연습

프로그래머스 시저 암호

miffy짱 2022. 8. 10. 23:52
반응형
#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] == ' ')
            continue;
        if (s[i] >= 'A' && s[i] <= 'Z')
        {
            if (s[i] + n > 'Z')
                s[i] = s[i] + n - 26;
            else
                s[i] = s[i] + n;
        }
        else if (s[i] >= 'a' && s[i] <= 'z')
        {
            if (s[i] + n > 'z')
                s[i] = s[i] + n - 26;
            else
                s[i] = s[i] + n;
        }
    }
    return s;
}