데이터 엔지니어링 정복/Algorithm
[정렬] 백준 - ATM
noohhee
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
반응형