[Algorithm/Java] 알고리즘 자바 순열 구하기 DFS (코딩테스트, DFS, BFS, 중복을 허락하지 않은 순열, 중복하지 않는 순열)

---

1. 문제 설명

• 설명
  10 이하의 N개의 자연수가 주어지면 이 중 M개를 뽑아 일렬로 나열하는 방법을 모두 출력합니다.
• 입력
  첫 번째 줄에 자연수 N(3<=N<=10)과 M(2<=M<=N)이 주어집니다.
  두 번째 줄에 N개의 자연수가 오름차순으로 주어집니다.
• 출력
  첫 번째 줄에 결과를 출력합니다.
  출력순서는 사전 순으로 오름차순으로 출력합니다.

 

2. 문제 코드 및 풀이 설명

import java.util.Scanner;

public class Permutation {
  static int[] pm, ch, arr;
  static int n, m;

  public static void main(String[] args) {
    Permutation T = new Permutation();
    Scanner kb = new Scanner(System.in);

    n = kb.nextInt();
    m = kb.nextInt();
    arr = new int[n]; // 나열할 수들
    for (int i = 0; i < n; i++) arr[i] = kb.nextInt();
    ch = new int[n]; // 수를 선택했는지 체크
    pm = new int[m]; // 현재 입력된 순열
    T.DFS(0);
  }

  // 해결방법
  // 입력된 수를 중복을 하지 않고 뽑는 경우의 수
  // 입력된 수 중에 이전에 뽑지 않았던 수만 재귀함수로 돌림
  // 입력되었음을 체크하고 재귀함수가 끝나면 그 수를 다시 체크해제
  // 이때 입력된 수를 순열에 추가
  // 뽑아야하는 개수만큼 뽑았다면 출력
  public void DFS(int L) {
    if (L == m) { // m개 만큼 수를 뽑았다면 입력된 순열을 출력
      for (int x : pm) System.out.print(x + " ");
      System.out.println();
    } else { // 아직 덜 뽑았다면 다음 수를 뽑아줌
      for (int i = 0; i < n; i++) {
        // 체크하지 않은, 곧 이전에 뽑지 않은 수만 다음 재귀함수로
        if (ch[i] == 0) {
          ch[i] = 1; // 뽑았으니 체크
          pm[L] = arr[i]; // 뽑은 수를 순열에 추가
          DFS(L + 1); // 다음 수를 뽑으러
          ch[i] = 0; // 재귀함수가 끝났다면 현재 뽑은 수를 체크해제
        }
      }
    }
  }
}

 

 

3. 출력 예시

입력
3 2
3 6 9

출력
3 6
3 9
6 3
6 9
9 3
9 6

 

---

위의 내용은 인프런에서 수강할 수 있는 김태원님의 자바 알고리즘 문제풀이 강의를 바탕으로 공부한 내용을 정리한 내용입니다!

https://inf.run/iAi6

자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 - 인프런 | 강의