[백준.1138 - Java, Python] 한 줄로 서기

📌 1138. 한 줄로 서기

N명의 사람들은 매일 아침 한 줄로 선다. 이 사람들은 자리를 마음대로 서지 못하고 오민식의 지시대로 선다.

어느 날 사람들은 오민식이 사람들이 줄 서는 위치를 기록해 놓는다는 것을 알았다. 그리고 아침에 자기가 기록해 놓은 것과 사람들이 줄을 선 위치가 맞는지 확인한다.

 

사람들은 자기보다 큰 사람이 왼쪽에 몇 명 있었는지만을 기억한다. N명의 사람이 있고, 사람들의 키는 1부터 N까지 모두 다르다. 각 사람들이 기억하는 정보가 주어질 때, 줄을 어떻게 서야 하는지 출력하는 프로그램을 작성하시오.

 

 

✏️ 입력. 

첫째 줄에 사람의 수 N이 주어진다. N은 10보다 작거나 같은 자연수이다. 둘째 줄에는 키가 1인 사람부터 차례대로 자기보다 키가 큰 사람이 왼쪽에 몇 명이 있었는지 주어진다. i번째 수는 0보다 크거나 같고, N-i보다 작거나 같다. i는 0부터 시작한다.

 

📋 출력. 

첫째 줄에 줄을 선 순서대로 키를 출력한다.

 

✍️ 방법.

- 왼쪽에 나보다 키 큰 사람이 n명, 왼쪽 자리 개수가 m개 

- 나의 왼쪽에 키 작은 사람이 있을 수 있으니 m >= n 이 성립된다

- num : 입력받은 값을 저장하기위한 배열

- res : 최종적으로 자리를 저장하기위한 배열 

- left : 왼쪽에 있는 사람의 수를 저장할 변수

- idx : 내가 자리할 빈 자리의 인덱스를 저장할 변수

 

 

1. 반복문을 통해 num을 확인하고, res 채워 넣기  

    1-1. 아래의 조건을 모두 만족하면 break

        - left가 내 기억 속의 사람 수보다 많거나 같은 경우

        - 내 자리가 될 위치가 비어있는 경우

    1-2 아래의 조건 중 하나만 만족하면 left+1

        - 자리가 비워져 있는지 확인

        - 빈자리가 아니라면 내 키보다 큰 사람인지 확인 

2. res의 내 자리에 나의 키를 저장 (내 키 = 배열 num에서의 index+1)

3. res의 값을 출력 

 

 

 

✍️  JAVA

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

public class Main {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		
		int n = sc.nextInt();
		int[] num = new int[n];
		int[] result = new int[n];
		Arrays.fill(result, 0);
		
		// 입력받기
		for(int i = 0; i < n; i++) num[i] = sc.nextInt(); 
		
		// 자리 찾기
		for(int i = 0; i < n; i++) {
			int left = 0;
			int idx = 0;
			
			for(int j = 0; j < n; j++) {
				if(left >= num[i] && result[j] == 0) {
					idx = j;
					break;
				}
				if(result[j] == 0 || result[j] > i)  left++;
			}
			result[idx] = i+1;
		}
		
		for(int res : result) System.out.print(res + " ");
	}
}

 

✍️  Python 

n = int(input())
num = list(map(int, input().split()))
res = [0] * n

# 자리 찾기
for i in range(len(num)) :
  left = 0   # left : 왼쪽에 있는 자리 수
  idx = 0    # 빈 자리 index
  
  for j in range(len(res)) :
    # left 가 내 기억 속의 사람 수보다 많거나 같을 때
    # 그리고 현재 자리가 비워져 있을 때 break 
    if left >= num[i] and res[j] == 0 :
      idx = j
      break
    # 자리가 비워져 있거나, 내 키보다 큰 키인 경우 left + 1
    if res[j] == 0 or res[j] > i :
      left += 1

  # 내 키는 현재 index + 1 
  res[idx] = i+1

# 출력
for r in res :
  print(r, end=" ")