[SWEA/Python] 1928. Base64 Decoder

SW Expert Academy

 

문제를 이해하는 것부터 시간이 좀 걸렸는데 요약하자면

 

1. input 문자열의 각 문자를 주어진 표의 10진수 숫자로 매핑한다.

2. 10진수 숫자들을 6비트짜리 2진수로 표현한다.

3. 8비트씩 끊어 10진수 숫자로 표현한다.

4. 10진수 숫자들을 아스키 코드로 변환해 decoding된 원문 ouput 문자열을 출력한다.

 

def base64_decode(encoded_string):
    base64_table = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
 
    decoded_string = ""
 
    decoding_table = {char: index for index, char in enumerate(base64_table)}
 
    buffer = 0
    count = 0
    for char in encoded_string:
        value = decoding_table[char]
         
        buffer = (buffer << 6) | value
        count += 6
 
        if count >= 8:
            count -= 8
            decoded_string += chr((buffer >> count) & 0xFF)
            buffer &= (1 << count) - 1
 
    return decoded_string
 
 
T = int(input())
 
for tc in range(1, T + 1):
    encoded_string = input()
    decoded_string = base64_decode(encoded_string)
    print(f"#{tc} {decoded_string}")

 

사실 나는 10진수를 2진수로 변환하는 함수를 직접 구현해서 풀이하려 했는데, 댓글에 좋은 풀이가 있어서 참고했다. (Ro_LCH님..🙏)

 

쉬프트 연산자와 비트 연산을 사용해 2진수로 직접 변환하는 과정을 생략할 수 있었고,

buffer에 6비트를 적재하며 8비트가 넘어갈 때 바로 원문으로 한 글자씩 decoding을 수행했다.

 

비트 연산을 오랫동안 사용하지 않아서 거의 잊고 살았는데 다시금 공부하는 계기가 됐다! 👍