[백준] 15651번 N과 M (3) - Java, 자바

난이도

실버 3

문제

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

풀이

15649번 N과 M(1)번 문제와 유사하지만, 여기서 같은 수를 여러번 골라도 된다는 조건이 추가됐다.

    private static void dfs(int depth) {
        if (depth == m) {
            for (int val : arr) {
                sb.append(val).append(" ");
            }
            sb.append("\n");
            return;
        }

        for (int i = 0; i < n; i++) {
            if (!visit[i]) { //* 
                visit[i] = true;
                arr[depth] = i + 1;
                dfs(depth + 1);
                visit[i] = false;
            }
        }
    }

같은 수를 걸러내기 위해 "if (!visit[i]) = 방문하지 않는 배열" 이라는 조건을 추가했는데,
이 문제는 같은 수 허용이기 때문에 이 조건을 삭제해주면 된다.

코드

package 백트래킹;

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

public class boj_15651 {

    static StringBuilder sb = new StringBuilder();
    static int n, m;
    static boolean[] visit;
    static int[] arr;


    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());

        visit = new boolean[n];
        arr = new int[m];
        dfs(0);
        System.out.println(sb);
    }

    private static void dfs(int depth) {
        if (depth == m) {
            for (int val : arr) {
                sb.append(val).append(" ");
            }
            sb.append("\n");
            return;
        }

        for (int i = 0; i < n; i++) {
            visit[i] = true;
            arr[depth] = i + 1;
            dfs(depth + 1);
            visit[i] = false;
        }
    }
}