[백준/자바] 15665번: N과 M (11)

문제

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

풀이

• N개의 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.
• 중복되는 수열을 여러 번 출력하면 안 된다.
• 오름차순

중복되는 값이 존재하는 N개의 자연수 중에서 중복 순열을 뽑아내는 문제입니다.

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

• 오름차순으로 출력해야 되기 때문에, 입력 받을 때 정렬을 해줬습니다.

	private static void dfs(int idx) {
		
		if (idx == m) {
			for (int i : li) sb.append(i).append(" ");
			sb.append("\n");
			return;
		}
		
		int prev = 0; // 동일 레벨에서 중복값을 판단하는 변수
		for (int i = 0; i < n; i++) {
			if (prev == nums[i]) continue; // 중복되는 값이라면 pass
			li[idx] = nums[i];
			prev = nums[i]; // 값 할당
			dfs(idx + 1); // 재귀
		}
	}

• 중복되는 순열을 제거하기 위해 변수를 선언하여 동일 레벨에서 중복되는 값이 들어가지 못하게 하였습니다.
• 중복 순열이므로 visited 배열은 필요하지 않습니다.

코드

import java.io.*;
import java.util.*;

public class Main_15665 {
	static StringBuilder sb = new StringBuilder();
	static int n, m;
	static int[] li, nums;
	
	private static void dfs(int idx) {
		
		if (idx == m) {
			for (int i : li) sb.append(i).append(" ");
			sb.append("\n");
			return;
		}
		
		int prev = 0;
		for (int i = 0; i < n; i++) {
			if (prev == nums[i]) continue;
			li[idx] = nums[i];
			prev = nums[i];
			dfs(idx + 1);
		}
	}
	
	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());
		
		st = new StringTokenizer(br.readLine());
		nums = new int[n];
		for (int i = 0; i < n; i++) nums[i] = Integer.parseInt(st.nextToken());
		Arrays.sort(nums);
		
		li = new int[m];
		
		dfs(0);
		System.out.println(sb.toString());
	}

}