[Programmers] 시저 암호

프로그래머스 level1

Posted by kyoungIn on March 29, 2019

시저 암호

문제 링크 : (https://programmers.co.kr/learn/courses/30/lessons/12926)

풀이

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#include <string>
#include <vector>
#include <iostream>
using namespace std;

string solution(string s, int n) {
    string answer = "";
    int flag,data; 
    for(int i=0;i<s.length();i++){
        flag=0,data=0;
        if(s[i]==' ')continue;
        if(s[i]>='a' && s[i]<='z') flag=1;//소문자
        data=s[i]+n;
        if(flag == 1 && data>'z')
            data-=26;
        else if(flag == 0 && data >'Z')
            data-=26;
        s[i]=data;
    }
    return s;
}