[백준] N과 M (3) (java) 순열, Permutation, 중복 O

2020-03-31

N과 M (3) (백준> Math / 순열, Permutation, 중복 O)

문제 링크

문제설명

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

  • 1부터 N까지 자연수 중에서 M개를 고른 수열
  • 같은 수를 여러 번 골라도 된다.

예제

//입력
3 1

//출력
1
2
3

예제

//입력
4 2

//출력
1 1
1 2
1 3
1 4
2 1
2 2
2 3
2 4
3 1
3 2
3 3
3 4
4 1
4 2
4 3
4 4

풀이

package Math;

import java.util.*;

//순열 (중복 O)
public class beak_N과M_3 {
	static int n;
	static int m;
	static StringBuilder sb;
	static LinkedList<Integer> numbers;
	
	public static void perm(int count) {
		if(count == m) {
			for(int i:numbers)
				sb.append(i+" ");
			
			sb.append("\n");
			return;
		}
		
		for(int i=0; i<n; i++) {
			//중복 허용이니까
			numbers.add(i+1);
			perm(count+1);
			numbers.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<>();
		
		perm(0); // 시작 idx를 넣는다
		
		System.out.println(sb.toString());
	}
}

후기 (20min)

중복을 허용하는 순열은

if(numbers.contains(i+1)) continue 라는 조건만 빼주면 된다!