프로그래머스 시저 암호 Swift

문제 설명

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

제한 조건

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

입출력 예

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

풀이:처음에 생각했을 땐 아주 간단한 문제라고 생각했다.

1.공백인지 아닌지 구별한다.

2.대소문자를 구별한다.

3.해당 알파벳이 몇 번째인지 구한다.

4.해당 알파벳 순서에 n만큼 더해준 순서에 있는 알파벳을 구한다.

5.알파벳이 26개니깐 26이 넘어가면 -26을 해준 알파벳을 구한다.

이런 식으로 그냥 모두 for문을 쓰니 테스트케이스 마지막에서 시간초과라고 떴다. 내가 생각해도 s의 길이가 1000자만 넘어가도 아주 비효율적이라고 생각했다.

그래서 생각한 게 내가 좋아하는 딕셔너리였다. 알파벳을 key값으로 알파벳의 순서를 value값으로 넣어줬다. 또한 key값만 모아서 정렬해놓은 배열으로 keys를 만들어주었다.

우선 number라는 변수를 만들어서 해당 알파벳의 순서에 해당하는 값을 알아낸 뒤 n만큼 더해주고 

key들만 모아놓은 keys에 해당 숫자를 넣어서 n만큼 더한 알파벳을 알아낸다 -> value

다시 알파벳딕셔너리에 value를 넣어줘서 해당 알파벳이 몇 번째인지 알아낸다 -> value1

그리고 알파벳딕셔너리의 키 중 value의 순서가 있는 알파벳을 차례대로 대문자라면 그냥 더해주고

아니면 소문자로 변형해서 넣어준다.

 

 

func solution(_ s:String, _ n:Int) -> String {
    var answer = String()
    let alphabet = ["A":0,"B":1,"C":2,"D":3,"E":4,"F":5,"G":6,"H":7,"I":8,"J":9,"K":10,"L":11,"M":12,"N":13,"O":14,"P":15,"Q":16,"R":17,"S":18,"T":19,"U":20,"V":21,"W":22,"X":23,"Y":24,"Z":25]
       let keys = alphabet.keys.sorted()
 
    for i in 0..<Array(s).count{
        if Array(s)[i] == " "{
            answer += " "
            continue
        }
            var number = alphabet[String(Array(s)[i].uppercased())]! + n
                                     if number > 25 {
                                     number = number - 26
                                              }
                                     let value = keys[number]
                                     let value1 = alphabet.index(forKey: value)
       if Array(s)[i].isUppercase{
            answer += alphabet.keys[value1!]
        }else {
            answer += alphabet.keys[value1!].lowercased()
        }
    }
    return answer
}

다른 사람의 풀이 중 가장 좋다고 생각하는 것

풀이: 우선 s를 unicodeScalars와 map을 이용해서 s의 문자들의 unicedeScalars의 값을 Int로 형변환한 값 중 65~90은 대문자 97~122는 소문자 32는 공백인데 대문자,소문자인 경우 알파벳의 크기로 나눈다음 다시 최솟값을 더해줘 크기를 넘지 않게 해주고 공백인 경우는 그냥 출력하게끔 해서 opted라는 배열에 담는다. 그 배열에 있는 아스키코드 숫자들을 다시 Unicode.Scalar.init($0)로 해당 문자로 바꿔준 다음

flatMap으로 nil이 아닌 것들을 모아서 (flatMap -> compactMap으로 바뀌었다.) 다시 문자로 변형해준다음 합쳐준다.

사실 이해가 백프로 되진 않음.... 하지만

flatMap과 unicodeScalars를 사용한 것에 감탄..

func solution(_ s:String, _ n:Int) -> String {
 
  let opted = s.unicodeScalars
    .map { unicodeScala -> Int in
      let number = Int(unicodeScala.value)
      switch number  {
      case 65...90: return (number - 65 + n) % 26 + 65
      case 97...122: return (number - 97 + n) % 26 + 97
      case 32: return number
      default:
        print("NG")
        return number
      }
    }
    .map { Unicode.Scalar.init($0) }
    .flatMap { $0 }
    .map { Character.init($0) }
 
  return String(opted)
}