Problem

https://school.programmers.co.kr/learn/courses/30/lessons/120892

군 전략가 머쓱이는 전쟁 중 적군이 다음과 같은 암호 체계를 사용한다는 것을 알아냈습니다.

암호화된 문자열 cipher를 주고받습니다.
그 문자열에서 code의 배수 번째 글자만 진짜 암호입니다.
문자열 cipher와 정수 code가 매개변수로 주어질 때 해독된 암호 문자열을 return하도록 solution 함수를 완성해주세요.

Solution

인덱스는 0부터 시작하지만 글자로서는 1번째부터 시작하니까 code-1로 시작해서 계속 code 만큼을 더해가면서 해당 문자를 더한다.

Code

class Solution {
    public String solution(String cipher, int code) {
        // 인덱스가 0일 때 1번째 글자라고 간주하더라. 
        StringBuilder sb = new StringBuilder();
        int idx = code - 1;
        for(int i = 0; i < cipher.length(); i++) {
            if(idx >= cipher.length()) return sb.toString();
            sb.append(cipher.charAt(idx));
            idx += code;
        }
        
        return sb.toString();
    }
}

위 코드는 정답이지만

• for문에서 i를 전혀 사용하고 있지 않다.
• 게다가 idx가 out of range 예외가 발생하는 것을 막기 위해서 중간에 return문을 작성해야 했다.

그 점을 개선한 게 아래 코드다.

class Solution {
    public String solution(String cipher, int code) {
        StringBuilder sb = new StringBuilder();
        
        for (int i = code - 1; i < cipher.length(); i += code) {
            sb.append(cipher.charAt(i));
        }
        
        return sb.toString();
    }
}