16-janghw0126

janghw0126/README.md

@@ -17,4 +17,5 @@
 | 13차시 | 2024.2.7 |  DFS |  <a href= "https://www.acmicpc.net/problem/2606">바이러스</a> |[#57](https://github.com/AlgoLeadMe/AlgoLeadMe-4/pull/57) |
 | 14차시 | 2024.2.12 |  Backtracking |  <a href= "https://www.acmicpc.net/problem/15649">N과 M (1)</a> |[#61](https://github.com/AlgoLeadMe/AlgoLeadMe-4/pull/61) |
 | 15차시 | 2024.2.15 |  Backtracking |  <a href= "https://www.acmicpc.net/problem/15650">N과 M (2)</a> |[#65](https://github.com/AlgoLeadMe/AlgoLeadMe-4/pull/65) |
+| 16차시 | 2024.2.18 |  Backtracking |  <a href= "https://www.acmicpc.net/problem/15654">N과 M (5)</a> |[#66](https://github.com/AlgoLeadMe/AlgoLeadMe-4/pull/66) |
 ---

janghw0126/백트래킹/Backtracking_2.py

@@ -1,19 +1,24 @@
 import sys
+input=sys.stdin.readline
+#자연수 N과 M을 입력받는다.
+N,M=map(int,input().split())
 
-def dfs(c):
-  if len(s) == m:
-    for j in s:
-      print(j, end=" ")
-    print()
-    return
-  for i in range(c, n + 1):
-    if i not in s:
-      s.append(i)
-      dfs(i)
-      s.pop()
+#답을 출력할 result 리스트를 선언한다.
+result=[]
 
-input = sys.stdin.readline
-n, m = map(int, input().split())
-s = []
-a = []
-dfs(1)
+#문제 해결을 위한 함수 solution을 정의한다.
+def solution(start):
+    #만약 리스트의 크기가 M이라면 모든 고른 것이 되므로 해답이 된다.
+    if len(result)==M:
+        #리스트의 각 요소를 문자열로 변환한 후 공백을 기준으로 결합하여 결과를 출력한다.
+        return print(' '.join(map(str,result)))
+    #자릿수가 아직 안찼으면 수행할 반복문이다.
+    for i in range(start,N):
+        #result리스트에 해당 숫자를 추가한다.
+        result.append(i+1)
+        #다음 자릿수의 숫자를 넣기 위해 재귀적으로 solution함수를 호출한다.
+        solution(i+1)
+        #함수 호출 후에는 해당 자릿수에서의 선택이 끝났으므로, result 리스트에서 마지막에 추가된 숫자를 제거한다.
+        result.pop()
+#위의 과정을 반복하여 모든 경우의 수를 찾는다.
+solution(0)

janghw0126/백트래킹/Backtracking_3.py

@@ -0,0 +1,28 @@
+#자연수 N과 M을 입력받는다.
+N, M = map(int, input().split())
+#숫자들을 입력받아 리스트에 저장한다.
+numbers = [int(x) for x in input().split()]
+#숫자들을 오름차순으로 정렬한다.
+numbers.sort()
+
+def backtracking(depth):
+    #현재까지 선택한 숫자의 개수가 M과 같다면 수행할 문이다.
+    if depth == M:
+        #박스에 저장된 숫자들을 문자열로 변환하여 공백을 기준으로 결합하여 출력한다.
+        print(' '.join(map(str,box)))
+        return
+    #0부터 N-1까지의 인덱스에 대해 반복한다.
+    for i in range(N):
+        #이미 선택된 숫자라면 건너뛴다.
+        if numbers[i] in box:
+            continue
+        #숫자를 박스에 추가한다.
+        box.append(numbers[i])
+        #다음 자릿수의 숫자를 선택하기 위해 재귀적으로 backtracking 함수를 호출한다.
+        backtracking(depth + 1)
+        #함수 호출 후에는 해당 자릿수에서의 선택이 끝났으므로, 박스에서 마지막에 추가된 숫자를 제거한다.
+        box.pop()
+#선택한 숫자들을 저장할 리스트를 선언한다.
+box = []
+#백트래킹 함수를 호출하여 문제를 해결한다.
+backtracking(0)