[백한결] 126차 라이브 코테 제출

live12/test126/문제1/백한결.py

@@ -0,0 +1,27 @@
+from itertools import permutations
+
+
+def solution(numbers):
+    numberSet = set()
+
+    for i in range(1, len(numbers) + 1):
+        for p in permutations(numbers, i):
+            num = int(''.join(p))
+            numberSet.add(num)
+
+    count = 0
+
+    for num in numberSet:
+        if num < 2:
+            continue
+        isPrime = True
+
+        for i in range(2, int(num ** 0.5) + 1):
+            if num % i == 0:
+                isPrime = False
+                break
+
+        if isPrime:
+            count += 1
+
+    return count

live12/test126/문제2/백한결.py

@@ -0,0 +1,56 @@
+def solution(n, info):
+    maxDiff = -1
+    answer = [-1]
+
+    # DFS로 점수 구간을 하나씩 탐색하면서 라이언의 경우를 모두 탐색
+    def dfs(index, arrowsLeft, ryanShots):
+        nonlocal maxDiff, answer
+
+        # 모든 구간을 다 고려했을 때
+        if index == 11:
+            if arrowsLeft > 0:
+                ryanShots[10] += arrowsLeft
+
+            ryanScore = 0
+            apeachScore = 0
+
+            # 각 점수 계산
+            for i in range(11):
+                if info[i] == 0 and ryanShots[i] == 0:
+                    continue
+                if ryanShots[i] > info[i]:
+                    ryanScore += 10 - i
+                else:
+                    apeachScore += 10 - i
+
+            # 라이언이 이긴 경우
+            if ryanScore > apeachScore:
+                diff = ryanScore - apeachScore
+                if diff > maxDiff:
+                    maxDiff = diff
+                    answer = ryanShots[:]
+                elif diff == maxDiff:
+                    for i in range(10, -1, -1):
+                        if ryanShots[i] > answer[i]:
+                            answer = ryanShots[:]
+                            break
+                        elif ryanShots[i] < answer[i]:
+                            break
+
+            if arrowsLeft > 0:
+                ryanShots[10] -= arrowsLeft
+            return
+
+        # 현재 점수를 얻기 위해 필요한 화살 수
+        neededArrows = info[index] + 1
+        if neededArrows <= arrowsLeft:
+            # 이 점수를 가져가는 경우
+            ryanShots[index] = neededArrows
+            dfs(index + 1, arrowsLeft - neededArrows, ryanShots)
+            ryanShots[index] = 0
+
+        # 이 점수를 포기하는 경우
+        dfs(index + 1, arrowsLeft, ryanShots)
+
+    dfs(0, n, [0] * 11)
+    return answer