[백준] N과 M (6) (java) 조합, Combination, 중복 X, 응용ver

2020-03-31

N과 M (6) (백준> Math / 조합, Combination, 중복 X, 응용ver)

문제 링크

문제설명

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다.

  • N개의 자연수 중에서 M개를 고른 수열
  • 고른 수열은 오름차순이어야 한다.

예제

//입력
3 1
4 5 2

//출력
2
4
5

예제

//입력
4 2
9 8 7 1

//출력
1 7
1 8
1 9
7 8
7 9
8 9


풀이

package Math;

import java.util.Arrays;
import java.util.LinkedList;
import java.util.Scanner;

public class beak_N과M_6 {

	static int n;
	static int m;
	static LinkedList<Integer> result;
	static StringBuilder sb;
	static LinkedList<Integer> numbers;
	
	public static void comb(int count) {
		if(count == m) {
			for(int i:result)
				sb.append(i + " ");
			
			sb.append("\n");
			return ;
		}
		
		for(int i=0; i<n; i++) {
			if(result.contains(numbers.get(i)))
				continue;
			
			//result가 가진 끝값의 number인덱스가 i이상이어야 한다.            
			if(result.size()!=0 && i < numbers.indexOf(result.getLast()))
				continue;
			
			result.add(numbers.get(i));
			comb(count+1);
			result.removeLast();
			
		}
	}
	
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		
		n = sc.nextInt();
		m = sc.nextInt();
		
		sb = new StringBuilder();
		numbers = new LinkedList<>();
		result = new LinkedList<>();
		
		for(int i=0; i<n; i++)
			numbers.add(sc.nextInt());
		
		numbers.sort(null);
		
		comb(0);
		
		System.out.println(sb.toString());
	}
}

후기 (20min)

푸는 방식은 (5)와 비슷하다.

다만, 1,2,3 순차적으로 들어오는 것이 아니기때문에 리스트로 numbers에 값을 받고

중복을 허용하지 않기위해서

		if(result.contains(numbers.get(i)))
			continue;

를 추가해주고,

numbers가 가지는 마지막 값보다는 i가 커야하기때문에 (어차피 같다는 위에서 걸러줌) 조건 추가

	if(result.size()!=0 && i < numbers.indexOf(result.getLast()))
				continue;