[SWEA/d2] 1945. 간단한 소인수분해 (Python) _딕셔너리 활용 팁

📚 문제 링크

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV5Pl0Q6ANQDFAUq&categoryId=AV5Pl0Q6ANQDFAUq&categoryType=CODE&problemTitle=1945&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=&pageSize=10&pageIndex=1

SW Expert Academy

 

---

 

📒 코드

내 풀이. 딕셔너리 활용

T = int(input())
for tc in range(1, T+1):
    num = int(input()) # 6791400
    prime = [2, 3, 5, 7, 11]
    dic = {i:0 for i in prime}
    
    while num != 1:
        for k, v in dic.items():
            if num % k == 0: # 나누어 떨어지면
                dic[k] = v + 1
                num = num // k
                
    print("#%d" %(tc), end=' ')
    for i in dic.values():
        print(i, end=' ')
    print()

 

•  2, 3, 5, 7, 11을 key로 가지고 value는 다 0인 딕셔너리 생성
• num이 1이 될 때까지 key 중에 나뉘는 걸로 계속 나눠주기

 

다른 풀이. 리스트 2개 활용

T = int(input())

for tc in range(1, T+1):
    N = int(input())
    num_lst = [2,3,5,7,11]
    cnt_lst = [0,0,0,0,0]
    for i in range(5):
        while N % num_lst[i] == 0:
            cnt_lst[i] += 1
            N //= num_lst[i]
    print(f'#{tc} ', end='')
    print(*cnt_lst)

 

• 인덱스는 다 0~4이니까 리스트 2개로 풀기도 가능하구나 ..
• while N % num_lst[i] == 0 을 함으로써 나누어 떨어지는 게 없을 때까지 계속 나누게 처리 가능!! ⭐️ 
• 출처: https://velog.io/@yunhlim/SWEA-1945.-간단한-소인수분해-D2

 

이 외에도, 계속 elif 문을 연결해서 푸신 분도 계셨다.

 

 

---

💡 새롭게 알게 된 지식 (딕셔너리)

딕셔너리 생성하는 방법 (딕셔너리 컴프리헨션으로 값 추가)

num = ['a', 'b', 'c']
dic1 = {i : 0 for i in num}
dic2 = {i : x for x, i in enumerate(num)} # value 증가하도록 !

print(dic1) # {'a': 0, 'b': 0, 'c': 0}
print(dic2) # {'a': 0, 'b': 1, 'c': 2}

 

 

아니면 dict.fromkeys()를 사용할수도!

dic3 = dict.fromkeys(num, 0)
print(dic3) # {'a': 0, 'b': 0, 'c': 0}

 

 

리스트의 원소를 value로 추가하기

dic4 = {i: num[i] for i in range(len(num))}
print(dic4) # {0: 'a', 1: 'b', 2: 'c'}

 

 

---

🔎 결과