[JAVA] 백준 15663번- N과 M(9)

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

15663번: N과 M (9)

 

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

public class Main {
    public static int n, m;
    public static int[] arr, nums;
    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());

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

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

        nums = new int[n + 1];

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            nums[i] = Integer.parseInt(st.nextToken());
        }
        Arrays.sort(nums);

        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;
        }

        int before = 0;
      for(int i=1; i<=n; i++){
          if(visit[i]) continue;

          if(before!=nums[i]){
              visit[i] = true;
              arr[cnt] = nums[i];
              before = nums[i];
              dfs(cnt+1);
              visit[i] = false;
          }

      }
    }
}

 

 

중복되는 수열을 배제시키기 위해 before변수를 만든다.

길이가 같은 바로 직전 수열 마지막값과 새로 추가되는 값이 같을 때, 중복되는 수열이 만들어지므로

before와 nums[i]를 비교해 같을 경우 중복되는 수열을 제거해준다. 

nums 배열이 오름차순으로 되어있기 때문에 이와 같이 해줄 수 있다.