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

2020-03-31

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

문제 링크

문제설명

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

  • N개의 자연수 중에서 M개를 고른 수열
  • 같은 수를 여러 번 골라도 된다.
  • 고른 수열은 비내림차순이어야 한다

예제

//입력
3 1
4 5 2

//출력
2
4
5

예제

//입력
4 2
9 8 7 1

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


풀이

package Math;

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

public class beak_N과M_8 {

	static int n;
	static int m;
	static LinkedList<Integer> result;
	static StringBuilder sb;
	static LinkedList<Integer> numbers;
	
	public static void perm(int count) {
		if(count == m) {
			for(int i:result)
				sb.append(i + " ");
			
			sb.append("\n");
			return ;
		}
		
		for(int i=0; i<n; i++) {
			
			//result가 가진 끝값의 number인덱스가 i이상이어야 한다.
			if(result.size() !=0 && i <numbers.indexOf(result.getLast()))
				continue;
			
			result.add(numbers.get(i));
			perm(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);
		
		perm(0);
		
		System.out.println(sb.toString());
	}
}

후기 (20min)

푸는 방식은 (6)과 비슷하다.

numbers가 가지는 마지막 값의 인덱스보다 i가 작은 경우에만 continue라는건,

크거나 같으면 add 해준다는거니까 중복을 만들 수 있다.

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