[Swift Algorithm] 시저 암호 Programmers(Lv.1)

문제 설명

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

 

제한 조건

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

 

입출력 예

s	n	result
"AB"	1	"BC"
"z"	1	"a"
"a B z"	4	"e F d"

 

내 문제 풀이

func solution(_ s:String, _ n:Int) -> String {
    let upperString = Array("abcdefghijklmnopqrstuvwxyz".uppercased())
    let lowerString = Array("abcdefghijklmnopqrstuvwxyz")
    let inputString = Array(s)
    var result = ""
   
    for i in inputString {
        // 주어진 문자열의 i번째 문자가 대문자 배열에 포함되었을 경우
        if upperString.contains(i) {
            // 대문자 배열에서 해당 문자의 위치를 찾고, 그 위치에 n을 더한 값을 저장
            var indexNum = upperString.firstIndex(of: i)! + n
            
            // 저장된 위치가 25보다 클 경우, 26을 빼줌으로써 다시 처음부터 위치를 찾을 수 있도록
            if indexNum > 25 {
                indexNum = indexNum - 26
            }

            // 대문자 배열에서 찾은 위치의 문자를 result 배열에 추가
            result.append(upperString[indexNum])
        }
        
        // 주어진 문자열의 i번째 문자가 소문자 배열에 포함되었을 경우
        else if lowerString.contains(i) {
            var indexNum = lowerString.firstIndex(of: i)! + n
            
            if indexNum > 25 {
                indexNum = indexNum - 26
            }
            
            result.append(lowerString[indexNum])
        }
        
        // 주어진 문자열의 i번째 문자가 공백일 경우
        else {
            result.append(i)
        }
    }
    return result
}

• 소문자, 대문자 배열을 각각 생성해주었다.
• 주어진 배열의 i 번째 문자가 대문자인지 소문자인지 판별하고 생성한 소문자 대문자 배열에서 위치를 찾았다.
• 그리고 찾은 위치에 n을 더해주어 새로운 위치의 문자를 result에 추가해주었는데,
  만약 더한 값이 알파벳 개수인 26개를 넘으면 다시 처음부터 위치를 찾아주기 위해 26을 빼주었다.
• 공백일 경우에는 그대로 공백을 추가해주었다.

 

인상적인 풀이

func solution(_ s:String, _ n:Int) -> String {
    return s.utf8.map {
        var code = Int($0)
        switch code {
            case 65...90:
                code = (code + n - 65) % 26 + 65
            case 97...122:
                code = (code + n - 97) % 26 + 97
            default:
                break
        }
        return String(UnicodeScalar(code)!)
    }.joined()
}

• 유니코드를 이용하여 계산해주었다.

 

💡 피드백

• 유니코드를 이용한 코딩 방법도 공부해봐야할 것 같다.
• 내가 푼 풀이는 내가 봐도 좀 복잡해서ㅜ 더 간결히 풀어보는 연습을 해야겠다.
  
  

---

 

🖍 [ 추가 ] 1주일 후 다시 풀어보기

func solution(_ s:String, _ n:Int) -> String {
    var result = ""
    let strLower = Array("abcdefghijklmnopqrstuvwxyz")
    let strUpper = Array("abcdefghijklmnopqrstuvwxyz".uppercased())
    
    for i in s {
        if i == " " { result.append(" ") }
        else {
            if strLower.contains(i) {
                result.append(String(strLower[(strLower.firstIndex(of: i)! + n) % 26])) }
            else {
                result.append(String(strUpper[(strUpper.firstIndex(of: i)! + n) % 26])) }
        }
    }
    return result
}

• 소문자, 대문자 배열을 생성해주었다.
• 공백이라면 공백을 추가해주고, 소문자 혹은 대문자일 경우 생성한 소문자 대문자 배열에서 위치를 찾아주었다.
• 나머지 연산을 사용하여 굳이 배열을 초과할 때를 따지지 않고, 바로 위치를 찾을 수 있게 했다.
• 나머지 연산을 이용하니 첫 번째 풀이에 비해 훨씬 효율적으로 코드가 완성됐고,
  전체 코드도 간결하게 잘 작성된 것 같다.

 

---

 

문제

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