摆渡车
题目描述
有 n 名同学要乘坐摆渡车从人大附中前往人民大学,第 i 位同学在第 ti 分钟去 等车。只有一辆摆渡车在工作,但摆渡车容量可以视为无限大。摆渡车从人大附中出发、 把车上的同学送到人民大学、再回到人大附中(去接其他同学),这样往返一趟总共花费m分钟(同学上下车时间忽略不计)。摆渡车要将所有同学都送到人民大学。
凯凯很好奇,如果他能任意安排摆渡车出发的时间,那么这些同学的等车时间之和最小为多少呢?
注意:摆渡车回到人大附中后可以即刻出发。
输入格式
第一行包含两个正整数 n,m,以一个空格分开,分别代表等车人数和摆渡车往返 一趟的时间。
第二行包含 n 个正整数,相邻两数之间以一个空格分隔,第 i 个非负整数 ti 代 表第 i 个同学到达车站的时刻。
输出格式
输出一行,一个整数,表示所有同学等车时间之和的最小值(单位:分钟)。
样例数据
输入样例 #1
5 1
3 4 4 3 5
输出样例 #1
0
输入样例 #2
5 5
11 13 1 5 5
输出样例 #2
4
样例说明
样例 1 说明
同学 1 和同学 4 在第 3 分钟开始等车,等待 0 分钟,在第 3 分钟乘坐摆渡车出发。摆渡车在第 4 分钟回到人大附中。
同学 2 和同学 3 在第 4 分钟开始等车,等待 0 分钟,在第 4 分钟乘坐摆渡车 出发。摆渡车在第 5 分钟回到人大附中。
同学 5 在第 5 分钟开始等车,等待 0 分钟,在第 5 分钟乘坐摆渡车出发。自此所有同学都被送到人民大学。总等待时间为 0。
样例 2 说明
同学 3 在第 1 分钟开始等车,等待 0 分钟,在第 1 分钟乘坐摆渡车出发。摆渡 车在第 6 分钟回到人大附中。
同学 4 和同学 5 在第 5 分钟开始等车,等待 1 分钟,在第 6 分钟乘坐摆渡车 出发。摆渡车在第 11 分钟回到人大附中。
同学 1 在第 11 分钟开始等车,等待 2 分钟;同学 2 在第 13 分钟开始等车,等待 0 分钟。他/她们在第 13 分钟乘坐摆渡车出发。自此所有同学都被送到人民大学。总等待时间为 4。
可以证明,没有总等待时间小于 4 的方案。
数据范围
对于 10% 的数据,n≤10,m=1,0≤ti≤100。
对于 30% 的数据,n≤20,m≤2,0≤ti≤100。
对于 50% 的数据,n≤500,m≤100,0≤ti≤104。
另有 20% 的数据,n≤500,m≤10,0≤ti≤4×106。
对于 100% 的数据,n≤500,m≤100,0≤ti≤4×106。
正解:
#include<cstdio>//#include<bits/stdc++.h>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
int n,m,t[505],cnt[505],dp[505][505];
const int INF=0x3f3f3f3f;//最大值
void read(int &x)//输入优化
{
int f=1;x=0;char s=getchar();
while(s>'9'||s<'0'){if(s=='-') f=-1;s=getchar();}
while(s>='0'&&s<='9'){x=(x<<3)+(x<<1)+s-'0';s=getchar();}
x*=f;
}
void write(int x)//输出优化
{
if(x<0)
{
x=(~x)+1;
putchar('-');
}
if(x>9)
write(x/10);
putchar(x%10+'0');
}
int Min(int a,int b)//求较小值的min函数
{
if(a<b)
return a;
return b;
}
int main()
{
read(n);read(m);
for(int i=1;i<=n;i++)
read(t[i]);
sort(t+1,t+n+1);//先把输入的t值排序,不排序会错
for(int i=1;i<=n;i++)
cnt[i]=cnt[i-1]+t[i];//循环,算出到i的总时间长
memset(dp,INF,sizeof(dp));//将dp赋初值
for(int i=1;i<=n;i++)
{
dp[i][0]=i*t[i]-cnt[i];//算出到第i个人时发0次车时等待时间的和
for(int j=0;j<m;j++)
for(int k=1;k<i;k++)
{
int s=Min(m-1,t[i]+j-t[k]-m);//求出从k到i中不发车,到i发车的时间差
if(s<0)
continue;
dp[i][j]=Min(dp[i][j],dp[k][s]+(i-k)*(t[i]+j)-(cnt[i]-cnt[k]));//求出dp[i][j]的最小值
}
for(int j=1;j<m;j++)
dp[i][j]=Min(dp[i][j],dp[i][j-1]);//循环保证这是一个递减的序列
}
write(dp[n][m-1]);//输出
return 0;
}