题目地址:
https://www.acwing.com/problem/content/description/915/
有 n n n个人排队打水,第 i i i个人打水耗时 t i t_i ti,问如何安排打水次序可以使得所有人的等待时间总和最小。
输入格式:
第一行包含整数
n
n
n。第二行包含
n
n
n个整数,其中第
i
i
i个整数表示第
i
i
i个人装满水桶所花费的时间
t
i
t_i
ti。
输出格式:
输出一个整数,表示最小的等待时间之和。
数据范围:
1
≤
n
≤
1
0
5
1\le n\le 10^5
1≤n≤105
1
≤
t
i
≤
1
0
4
1\le t_i\le 10^4
1≤ti≤104
先对 t i t_i ti从小到大排序,按照这个顺序打水总等待时间最少。
算法正确性证明:
反证法。如果安排的顺序不是按
t
i
t_i
ti单调增来排的,那么必然存在某个
k
k
k使得
t
k
>
t
k
+
1
t_k>t_{k+1}
tk>tk+1,那么交换这两个人打水的次序,总打水时间会减少
t
k
−
t
k
+
1
>
0
t_k-t_{k+1}>0
tk−tk+1>0,矛盾。
代码如下:
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int a[N];
int main() {
int n;
cin >> n;
for (int i = 0; i < n; i++) cin >> a[i];
sort(a, a + n);
long res = 0;
for (int i = 0; i < n; i++) res += (long) (n - i - 1) * a[i];
cout << res << endl;
return 0;
}
时间复杂度 O ( n log n ) O(n\log n) O(nlogn),空间 O ( 1 ) O(1) O(1)。