拼多多4月10日笔试第四题

有n辆车需要从东侧渡河到西侧，轮船每次最多运送m辆车，单程渡河的时间为k
第一行输入n，m，k
接下来n行输入每辆车达到的时间
输出最后一辆车到达河对岸的最小时间

input
3 2 10
10
30
40

output
50

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1010, inf = 0x3f3f3f3f;
//f[i]表示第i辆车达到对岸的最短时间
int f[N], a[N];
int n, m, k;

int main()
{
    cin >> n >> m >> k;
    for (int i = 1; i <= n; i++) cin >> a[i];
    sort(a + 1, a + n + 1);
    
    //memset(f, 0x3f, sizeof f);
    f[1] = a[1] + k;
    
    for (int i = 2; i <= n; i++)
    {
        f[i] = inf;
        for (int j = 1; j <= min(i, m); j++)
            f[i] = min(f[i], max(a[i], f[i - j] + k) + k);
    }
    
    cout << f[n] << endl;
    return 0;
}