[백준/자바] 15664번: N과 M (10)

문제

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

풀이

• 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, int start) {
		
		if (idx == m) {
			for (int i : li) sb.append(i).append(" ");
			sb.append("\n");
			return;
		}
		
		int prev = 0; // 동일 레벨에서 중복값을 판단하는 변수
		for (int i = start; i < n; i++) {
			if (prev != nums[i]) { // 같은 값을 넣으려고 하면 안 됨
				li[idx] = nums[i];
				prev = nums[i]; // 값 할당
				dfs(idx + 1, i + 1); // 재귀
			}
		}
	}

• 조합을 찾는 로직이므로 visited 배열은 필요하지 않습니다.

코드

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

public class Main_15664 {
	static StringBuilder sb = new StringBuilder();
	static int n, m;
	static int[] li, nums;
	
	private static void dfs(int idx, int start) {
		
		if (idx == m) {
			for (int i : li) sb.append(i).append(" ");
			sb.append("\n");
			return;
		}
		
		int prev = 0;
		for (int i = start; i < n; i++) {
			if (prev != nums[i]) {
				li[idx] = nums[i];
				prev = nums[i];
				dfs(idx + 1, i + 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, 0);
		
		System.out.println(sb.toString());
	}

}