[Programmers 프로그래머스] 12926 시저 암호

programmers.co.kr/learn/courses/30/lessons/12926

코딩테스트 연습 - 시저 암호

시저 암호는 각 알파벳을 일정한 거리만큼 밀어서 다른 알파벳으로 바꾸는 암호화 방식을 시저 암호라고 한다.

그래서 "AB"를 1만큼 밀면 "BC"가 되고, 3만큼 밀면 "DE"가 된다. "z"는 1만큼 밀면 "a"가 된다.

 

대문자일 때와 소문자일 때를 나눠서 구현해 준다. 입력값에 숫자를 더해주고 그 더해준 값이 Z를 넘어간다면 "A"값으로 나누어 주어 해당하는 값을 출력해 주는 방식으로 문제를 풀었다.

 

class Solution {
    public String solution(String s, int n) {

		String answer = "";

		char[] arr = s.toCharArray();

		for (int i = 0; i < arr.length; i++) {
			if (arr[i] >= 'A' && arr[i] <= 'Z') {
				if(arr[i]+n>'Z') {
					arr[i] = (char) ((arr[i] + n) % 'Z'+'A'-1);
					
				}else {
					arr[i] = (char) (arr[i] + n);
					
				}
			} else if (arr[i] == ' ') {
				arr[i] = ' ';
			} else if (arr[i] >= 'a' && arr[i] <= 'z') {
				if(arr[i]+n>'z') {
					arr[i] = (char) ((arr[i] + n) % 'z'+'a'-1);
					
				}else {
					arr[i] = (char) (arr[i] + n);
					
				}

			}
		}
		answer = String.copyValueOf(arr);
        return answer;
    }
}