본문 바로가기
2021/코딩테스트

[프로그래머스] Lv1 - 3진법 뒤집기_파이썬

by 에이도 2020. 11. 24.

 

이 문제는 3진법 뒤집기이다.

 

처음 입력받은 n을 3진법으로 변환 후 

3진법을 앞뒤 반전시키고

앞뒤가 반전된 3진법을 다시 10진법으로 표현하는 것이다.

 

따라서 총 3단계로 나누어서 계획을했다.

 

  1. n을 3진법으로 표현 
  2. 3진법을 앞뒤 반전시키기
  3. 앞뒤 반전된 삼진법을 10진법으로 표현하기.

 

우선 3진법은  y  = (0,1,2) * 3^0 + (0,1,2)* 3^1 + (0,1,2) * 3^3 + (0,1,2)* 3^4 + ···· 으로 계산할 수 있다.

따라서 입력받은 값 n = 45라면,  3^3 = 27 과 2* 3^2 =18 의 합으로 만들 수 있다.

 

이 원리를 가지고 

  1. n을 3진법으로 표현 
def solution(n):
    answer = []
    for i in range(30, -1 , -1):        # 나눌 수 있는 가장 큰 값부터 찾아들어가기
        a = 3**i                        # 3^i 제곱
        quo = n/a                       
        
        if 0 <= quo <= 3 :              #몫이 3이하 =  
            if quo >= 2:
                answer.append(2)        # 3진법 표현으로 2추가
                n = n - a*2            # n -2*3^i 
            elif 1<= quo < 2:
                answer.append(1)
                n = n - a*1

            else:
                if len(answer) >=1:    # 3진법의 맨 앞자리가 0이 아닌경우를 조건으로 지정했다.
                    answer.append(0)   # 그렇지 않으면 00012001 이런식으로 맨 앞에 불필요한 자릿수가 생기기 때문이다.
        
                else:
                    pass
                
    return answer

 

n = 45이라면, 

a = 3^i 가 45를 나누었을 때

0이상, 3이하 값이 나올때까지 i가 감소한다.

 

몫이 0 이상 3이하라면, 

몫의 값에 따라 조건문으로 들어가게 된다.

 

3^3 = 27이므로 45/27 = 1.667 이다. 

따라서 n = 45 - 27 = 18 이된다.

 

그리고 삼진법 리스트에는 1이 추가된다.

 

또한 다음 loof에서는

3^2 = 9이므로, 18/9 = 2이다.

따라서 n = 18 - 9*2 가 되어 

n = 0이된다. 

 

그리고 len(answer)이 1이상이 되었기 때문에 3진법의 0제곱 자리까지 0이 생성이된다. 

 

2. 3진법을 앞뒤 반전시키기

여기서는 리스트의 특징을 이용했다.

 

리스트 index에 마이너스부호를 붙이게 되면 

맨 끝에서부터 -1 , -2 ... 이렇게 감소하게 된다.

 

이러한 특성을 이용해서

기존 3진법이 저장되어 있는 리스트(answer)을

맨 앞부터 맨 끝까지 역순으로 리스트를 새로 저장했다.

 

후에 reverse를 쓸 수 있다는걸 알게되었다.

 

 

 

 

 

 

 

 

 

 

 

 

 

3. 앞뒤 반전된 삼진법을 10진법으로 표현하기.

이제 마지막으로 앞뒤 반전된 3진법을 가지고 10진법으로 변환해야한다.

 

앞뒤 전환 된 삼진법은 위의 값처럼 앞에 0이 존재할 수 있다.

하지만 이는 제한된 만크의 0이기에 감안하고 계산을 하였다.

 

앞뒤 반전된 3진법(answer)의 len로 리스트길이 -1 의 정수 a를 얻는다.

이는 3진법 최댓값의 제곱수와 같아진다.  (여기서는 3^3)

 

그리고 그 최댓값의 계수(answer[a-i])를 곱해서 최종 값에 더하고,

그 다음 제곱수와 계수, 그리고 그다음 ... 

for문을 통해 모든 값을 계산해서 더한 후 

값을 출력한다. 

 

돌아 보면 쉬운 문제였지만.. 진행하면서 3진법 리스트 안에 0이 맨 앞에 추가되지 않도록, 중간에 0이 추가 되지 않는 문제 등 많은 에러가 발생했다.

 

코딩테스트 많이 남지 않았는데..! 더 열심히 해야겠다 !.! 화이팅

댓글