[JAVA] 백준 15649번- N과 M(1)

https://www.acmicpc.net/problem/15649

15649번: N과 M (1)

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main{
    public static int n, m;
    public static int[] arr; // 값을 담을 배열
    public static boolean[] visit; // 방문 여부
    public static StringBuilder sb = new StringBuilder();

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        StringTokenizer st = new StringTokenizer(br.readLine());

        // 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열 모두 찾기

        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());

        arr = new int[m];
        visit = new boolean[n+1];

        dfs(0);
        System.out.println(sb);
    }

    public static void dfs(int cnt){

        if(cnt == m){
            for(int val:arr){
                sb.append(val).append(" ");
            }
            sb.append("\n");
            return;
        }

        for(int i=1; i<=n; i++){
            if(!visit[i]){ // 해당 값을 방문하지 않았따면
                visit[i] = true; // 방문 상태로 변경
                arr[cnt] = i; // cnt의 인덱스에 i저장
                dfs(cnt+1);
                visit[i] = false; // 자식 노드 방문이 끝나고 돌아오면 방문 상태를 false로 변경
            }
        }
    }
}

 

아이디어

백트래킹 재귀함수 안에서, N만큼 for 돌면서 숫자 선택(이 때 방문 여부 확인)

재귀함수에서 M개를 선택할 경우 출력 

 

 

시간 복잡도

중복이 가능 할 때: N^N, N=8까지 가능

중복이 불가능할 때: N!, N= 10까지 가능

위의 경우 중복이 불가능 하므로 시간 복잡도는 N! 이다. 문제에서 N이 최대 8까지이므로 백트래킹으로 푸는게 가능하다.

 

 

 

tip

백트래킹 문제는 N이 작음(10 이하)

재귀함수 사용할 때 종료 시점 잊지 말기!