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

문제

시저 암호 : 문제 링크

문제 분석

• 어떤 문장의 각 알파벳을 일정한 거리만큼 밀어서 다른 알파벳으로 바꾸는 암호화 방식을 시저 암호라고 한다. 예를 들어 "AB"는 1만큼 밀면 "BC"가 되고, 3만큼 밀면 "DE"가 된다. "z"는 1만큼 밀면 "a"가 된다. 문자열 s와 거리 n을 입력받아 s를 n만큼 민 암호문을 만드는 함수, solution을 완성

• 제한 조건

• 공백은 아무리 밀어도 공백이다.
• s는 알파벳 소문자, 대문자, 공백으로만 이루어져 있다.
• s의 길이는 8000이하이다.
• n은 1 이상, 25이하인 자연수이다.

• for loop를 통해 문자열 s의 첫번째부터 마지막 원소까지 순환하고, if문을 사용하여 공백이라면 다음 인덱스로 넘어감. else 문을 사용하여 알파벳인경우 연산을 수행하고, if ~ else if문을 통해 대문자와 소문자의 경우를 구분. 현재 문자에 n - 26를 더하고 저장하고, 또 다른 if문을 사용하여 각각이 'A', 'a'보다 작다면 26을 더함.(아스키 코드를 벗어남을 방지). loop 탈출 후, 최종적으로 저장된 문자열 s를 return

풀이

#include <string>

using namespace std;

string solution(string s, int n) {
    for(int i = 0; i < s.size(); ++i) {
        if(s[i] == ' ') continue;
        else {
            if('A' <= s[i] && s[i] <= 'Z') {
                s[i] += n - 26;
                if(s[i] < 'A') s[i] += 26;   
            } 
            else if('a' <= s[i] && s[i] <= 'z') {
                s[i] += n - 26;
                if(s[i] < 'a') s[i] += 26;
            }
        }
    }
    return s;
}