[Level 1] 시저 암호 (C++/Python)

C++)

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

string solution(string s, int n) {
    string answer = "";
    
    for(int i = 0; i < s.size(); i++){
        int pos = s.at(i) + n;
        
        if('a'<= s.at(i) && s.at(i)<='z'){
            if(pos > 'z')
                pos -= 26;
            answer += pos;
        }
        else if('A'<= s.at(i) && s.at(i)<='Z'){
            if(pos > 'Z')
                pos -= 26;
            answer += pos;
        } 
        else
            answer += " ";
    }
    
    return answer;
}

 

 

 

 Python)

def solution(s, n):
    result = ''
    for ch in s:
        pos = ord(ch)+n
        
        if 'a' <= ch <= 'z':
            if pos > ord('z'):
                pos -= 26
            result += chr(pos)
        elif 'A' <= ch <= 'Z':
            if pos > ord('Z'):
                pos -= 26
            result += chr(pos)
        else:
            result += ' '
    return result