algorithm 백준 11399 ATM (c)

1 minute read

사람들이 줄을 서는 순서에 따라서, 돈을 인출하는데 필요한 시간의 합이 달라지게 된다. 예를 들어, 총 5명이 있고, P1 = 3, P2 = 1, P3 = 4, P4 = 3, P5 = 2 인 경우를 생각해보자. [1, 2, 3, 4, 5] 순서로 줄을 선다면, 1번 사람은 3분만에 돈을 뽑을 수 있다. 2번 사람은 1번 사람이 돈을 뽑을 때 까지 기다려야 하기 때문에, 3+1 = 4분이 걸리게 된다. 3번 사람은 1번, 2번 사람이 돈을 뽑을 때까지 기다려야 하기 때문에, 총 3+1+4 = 8분이 필요하게 된다. 4번 사람은 3+1+4+3 = 11분, 5번 사람은 3+1+4+3+2 = 13분이 걸리게 된다. 이 경우에 각 사람이 돈을 인출하는데 필요한 시간의 합은 3+4+8+11+13 = 39분이 된다.

줄을 [2, 5, 1, 4, 3] 순서로 줄을 서면, 2번 사람은 1분만에, 5번 사람은 1+2 = 3분, 1번 사람은 1+2+3 = 6분, 4번 사람은 1+2+3+3 = 9분, 3번 사람은 1+2+3+3+4 = 13분이 걸리게 된다. 각 사람이 돈을 인출하는데 필요한 시간의 합은 1+3+6+9+13 = 32분이다. 이 방법보다 더 필요한 시간의 합을 최소로 만들 수는 없다.

줄을 서 있는 사람의 수 N과 각 사람이 돈을 인출하는데 걸리는 시간 Pi가 주어졌을 때, 각 사람이 돈을 인출하는데 필요한 시간의 합의 최솟값을 구하는 프로그램을 작성하시오.

[출처] : 백준 온라인 저지

풀이

#include <stdio.h>
#include <stdlib.h>
#define SWAP(x,y,t) (t=x, x=y, y=t)

int main() {
	int num, i, j, key;
	scanf("%d", &num);
	int* list = (int*)malloc(sizeof(int)*(num + 1));
	for (i = 0; i < num; i++) {
		scanf("%d", &list[i]);
	}
	for (i = 0; i < num; i++) {
		key = list[i];
		j = i - 1;
		while (j >= 0 && list[j] > key) {
			list[j + 1] = list[j];
			j--;
		}
		list[j + 1] = key;
	}
	int sum = list[0];
	for (i = 1; i < num; i++) {
		list[i] += list[i - 1];
		sum += list[i];
	}
	
	printf("%d", sum);
	free(list);
}

  • 오름차순으로 서있을 때, 걸리는 시간의 합이 최소가 되므로 삽입정렬 Insertion Sort로 오름차순 정렬 구현
  • 누적합을 구해야 하기 때문에 list[i] += list[i-1];로 각 인덱스가 이전 값을 더한 값을 가지고 있도록 했고, 이를 sum += list[i]; 로 다시 더했다

더보기

Categories:

Updated:

Comments