一道做法多种多样的题,DP做法的状态也很多。
我用\(dp[i]\)表示在第i秒发车的时间和,然后dp方程就很好写了
\(dp[i] = dp[j] + i车的等待时间\)j属于i-2m ~ i-m。
然后i车的等待时间可以用前缀和来优化求出,虽然很慢,但是足以通过这道题了。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define N 6011101
using namespace std;
int n, m, t[N]; //等待时间不是最终时间
int minn = 2147483647, dp[N], tim[N], per[N];//dp[i]表示第i时间所等待的时间和的最小值。
inline void init()
{
scanf("%d%d", &n, &m);
minn = n * m;
for (int i = 1; i <= n; i++) scanf("%d", &t[i]);
sort(t + 1, t + 1 + n);
for (int i = 0, k = 1; i <= t[n] + m; i++)
{
tim[i] = tim[i - 1];
per[i] = per[i - 1];
while (t[k] == i)
{
tim[i] += t[k];
per[i] ++;
k++;
}
}
}
int main()
{
init();
memset(dp, 123, sizeof(dp));
for (int i = 0; i <= m; i++)//预处理m以内的dp
dp[i] = per[i] * i - tim[i];
for (int i = m; i <= t[n] + m - 1; i++)
{
for (int j = max(i - m - m, 0); j <= i - m; j++)
dp[i] = min(dp[i], dp[j] + (per[i] - per[j]) * i - (tim[i] - tim[j]));
}
for (int i = t[n]; i <= t[n] + m - 1; i++)
minn = min(minn, dp[i]);
printf("%d", minn);
return 0;
}