[level 2] Title: 소수 찾기, Time: 5.91 ms, Memory: 9.48 MB -BaekjoonHub

Author: wlsgur11

프로그래머스/2/42839. 소수 찾기/README.md

@@ -4,7 +4,7 @@
 
 ### 성능 요약
 
-메모리: 10.3 MB, 시간: 10.09 ms
+메모리: 9.48 MB, 시간: 5.91 ms
 
 ### 구분
 
@@ -16,7 +16,7 @@
 
 ### 제출 일자
 
-2024년 09월 05일 21:21:10
+2025년 04월 12일 13:33:59
 
 ### 문제 설명
 

프로그래머스/2/42839. 소수 찾기/소수 찾기.py

@@ -1,17 +1,19 @@
 from itertools import permutations
 
+def isPrime(n):
+    for i in range(2, int(n**0.5)+1):
+        if n % i == 0:
+            return False       
+    return True
+
 def solution(numbers):
-    ans = []
-    nums = [n for n in numbers]
-    per = []
-    for i in range(1, len(numbers) + 1):
-        per += list(permutations(nums, i))
-    new_nums = set([int(("").join(p)) for p in per]) - {0, 1}
+    A = set()
+    for i in range(1, len(numbers)+1):
+        for j in permutations(list(numbers), i):
+            A.add(int("".join(j)))
 
-    cnt = len(new_nums)
-    for n in new_nums:
-        for i in range(2, int(n**0.5) + 1):
-            if n % i == 0:
-                cnt -= 1
-                break
-    return cnt
+    cnt = 0
+    for i in A:
+        if 2 <= i and isPrime(i):
+            cnt += 1
+    return cnt