지구정복

[정렬] 백준 - ATM 본문

데이터 엔지니어링 정복/Algorithm

[정렬] 백준 - ATM

eeaarrtthh 2021. 8. 13. 16:28
728x90
반응형

https://www.acmicpc.net/problem/11399

 

11399번: ATM

첫째 줄에 사람의 수 N(1 ≤ N ≤ 1,000)이 주어진다. 둘째 줄에는 각 사람이 돈을 인출하는데 걸리는 시간 Pi가 주어진다. (1 ≤ Pi ≤ 1,000)

www.acmicpc.net

 

-문제해설

돈을 인출하는데 오래 걸리는 사람이 앞쪽에 오게되면 뒤에 있는 사람은 그 만큼 더 많이 기다려야한다.

만일 적게 걸리는 사람이 앞에 오면 뒤에 있는 사람은 덜 기다리게 되므로 p배열을 오름차순으로 정렬한 뒤 

각 사람이 기다린 시간인 tmp를 구하고 전체 기다리게 되는 시간이 sum에 tmp값을 더해주면 된다.

 

 

-자바

package math;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class BJ11399 {

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt( br.readLine() );
		
		int[] p = new int[n];
		StringTokenizer st = new StringTokenizer( br.readLine() );
		for( int i=0; i<n; i++ ) p[i] = Integer.parseInt( st.nextToken() );
		
		Arrays.sort( p );
		int sum = p[0];
		int tmp = p[0];
		for( int i=1; i<n; i++ ) {
			tmp += p[i];
			sum += tmp;
		}
		
		System.out.println( sum );
	}
}

 

 

-파이썬

n = int( input() )
p = list( map(int, input().split()) )
p.sort()
s = p[0]
tmp = p[0]
for i in range( 1, n ):
    tmp += p[i]
    s += tmp
print( s )

 

728x90
반응형
Comments