암호 해독

2024. 9. 7. 18:00코딩테스트 입문

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


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

  • 암호화된 문자열 cipher를 주고받습니다.
  • 그 문자열에서 code의 배수 번째 글자만 진짜 암호입니다.

문자열 cipher와 정수 code가 매개변수로 주어질 때 해독된 암호 문자열을 return하도록 solution 함수를 완성해주세요.


제한사항

  • 1 ≤ cipher의 길이 ≤ 1,000
  • 1 ≤ codecipher의 길이
  • cipher는 소문자와 공백으로만 구성되어 있습니다.
  • 공백도 하나의 문자로 취급합니다.

입출력 예

cipher code result
"dfjardstddetckdaccccdegk" 4 "attack"
"pfqallllabwaoclk" 2 "fallback"

입출력 예 설명
입출력 예 #1

  • "dfjardstddetckdaccccdegk" 의 4번째, 8번째, 12번째, 16번째, 20번째, 24번째 글자를 합친 "attack"을 return합니다.

입출력 예 #2

  • "pfqallllabwaoclk" 의 2번째, 4번째, 6번째, 8번째, 10번째, 12번째, 14번째, 16번째 글자를 합친 "fallback"을 return합니다.

주어진 문자열 cipher에서 code번째 배수의 글자들을 출력하는 문제입니다.

 

문제 내에서 딱히 어려운 건 없고, 문제에서 말하는 1번째, 2번째가 인덱스 상으로는 0번째, 1번째 등으로 표현되는 것만 확인하면 됩니다.

파이썬으로 문제를 풀어보겠습니다.

def solution(cipher, code):
    answer = ''.join([c for i, c in enumerate(cipher) if (i + 1) % code == 0])
    return answer

enumerate를 사용해서 인덱스 i를 추출하고, 해당 인덱스 + 1이 code의 배수인지 확인한 뒤 join 함수로 합치는 방법입니다.

인덱스를 사용한다면 이 풀이가 일반적이라 생각합니다.

 

인덱스를 사용하지 않고 슬라이싱을 사용하는 풀이도 있습니다.

def solution(cipher, code):
    # code-1번째부터 끝까지 code만큼 뛰어넘는다.
    return cipher[code-1::code]

문자열 cipher을 슬라이싱할 때, 초기 인덱스만 code - 1로 설정한 뒤 code만큼 뛰어넘으면 됩니다.

 

저는 아래의 슬라이싱을 사용하는 풀이가 좀 더 파이썬스럽다고 생각되네요.

'코딩테스트 입문' 카테고리의 다른 글

영어가 싫어요  (0) 2024.09.09
대문자와 소문자  (1) 2024.09.08
369게임  (2) 2024.09.06
가까운 수  (0) 2024.09.05
삼각형의 완성조건 (1)  (0) 2024.09.04