【ACWing】913. 排队打水

题目地址:

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 1n105
1 ≤ t i ≤ 1 0 4 1\le t_i\le 10^4 1ti104

先对 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 tktk+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)

猜你喜欢

转载自blog.csdn.net/qq_46105170/article/details/114133788
今日推荐