[프로그래머스(Programmers)][자바(java)] (Lv3) 줄 서는 방법

 

programmers.co.kr/learn/courses/30/lessons/12936

코딩테스트 연습 - 줄 서는 방법

 

 

문제 설명

n명의 사람이 일렬로 줄을 서고 있습니다. n명의 사람들에게는 각각 1번부터 n번까지 번호가 매겨져 있습니다. n명이 사람을 줄을 서는 방법은 여러가지 방법이 있습니다. 예를 들어서 3명의 사람이 있다면 다음과 같이 6개의 방법이 있습니다.

• [1, 2, 3]
• [1, 3, 2]
• [2, 1, 3]
• [2, 3, 1]
• [3, 1, 2]
• [3, 2, 1]

사람의 수 n과, 자연수 k가 주어질 때, 사람을 나열 하는 방법을 사전 순으로 나열 했을 때, k번째 방법을 return하는 solution 함수를 완성해주세요.

 

제한사항

• n은 20이하의 자연수 입니다.
• k는 n! 이하의 자연수 입니다.

 

입출력 예

n	k	result
3	5	[3,1,2]

입출력 예시 설명

입출력 예 #1
문제의 예시와 같습니다.

 

 

문제 풀이

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Solution {
	
	public int[] solution( int m, long k ) {		// m : number of people  
													// k : k th number of cases
		List<Integer> p = new ArrayList<>();		// p : people list
		int ans[] = new int[m], n, idx = 0, i;		// i : picked people idx in the people list
		long f_m = 1;								// f_m : m factorial ( = m! )
		for( n = 1; n <= m; n++ ) {
			f_m *= n;
			p.add( n );		/* add peeple num in the list */
		}
		k--;
		while( m > 0 ) {
			f_m /= (m--);	/* f[m]/m; (= f[m-1];) m--; */
			i = (int)( k / f_m );
			ans[idx++] = p.remove( i );
			k %= f_m;
		}
		return ans;
    }

}

 

- 순열의 모든 경우의 수를 구하면 시간 초과 ( 결국 구글링 )

import java.util.Arrays;

public class Solution {
	
	long c;
	int[] ans;
	
	public void perm( int l, int n, long k, int a[], boolean v[] ) {
		if( l == n ) {
			if( ++c == k )
				ans = a.clone();
			return;
		}
		for( int i = 0; i < n; i++ ) {
			if( !v[i] ) {
				v[i] = true;
				a[l] = i+1;
				perm( l+1, n, k, a, v );
				v[i] = false;
			}
		}
	}
	
	public int[] solution( int n, long k ) {
        int a[] = new int[n];
        boolean v[] = new boolean[n];
        perm( 0, n, k, a, v );
        return ans;
    }

}

 

- list에 n명의 사람들의 번호를 넣음 ( 1~n, 총 n개 )

- list에서 먼저 줄을 설 사람의 인덱스 값을 구하기 위해 k에서 1을 뺌  ( 최초 한 번만 )

- list의 인덱스가 ( k / (f(n)/n) )인 사람이 먼저 줄을 섬

- 이미 줄을 선 사람은 list에서 제거하고,  해당 인덱스의 사람 번호를 정답 배열에 넣음 ( int ans[] )

- k은 f(n)의 나머지가 됨

- 모든 사람이 줄을 설 때까지 반복