#3886 Cats Transport
题面
小 S 是农场主,他养了 M 只猫,雇了 P 位饲养员。农场中有一条笔直的路,路边有 N 座山,从 1 到 N 编号。第 i 座山与第 i−1 座山之间的距离是 Di。饲养员都住在 1 号山上。
有一天,猫出去玩。第 i只猫去 Hi 号山玩,玩到时刻 Ti 停止,然后在原地等饲养员来接。饲养员们必须回收所有的猫。每个饲养员沿着路从 1 号山走到 N 号山,把各座山上已经在等待的猫全部接走。饲养员在路上行走需要时间,速度为 1 米每单位时间。饲养员在每座山上接猫的时间可以忽略,可以携带的猫的数量为无穷大
例如有两座相距为 1 的山,一只猫在 2 号山玩,玩到时刻 3 开始等待。如果饲养员从 1 号山在时刻 2 或 3 出发,那么他可以接到猫,猫的等待时间为 0 或 1 。而如果他于时刻 1 出发,那么他将于时刻 2 经过 2 号山,不能接到当时仍在玩的猫。
你的任务是规划每个饲养员从 1 号山出发的时间,使得所有猫等待时间的总和尽量小。饲养员出发的时间可以为负。
输入
第一行三个整数 N,M,P
第二行 N−1 个正整数 Di ,表示第 i 座山与第 i−1座山之间的距离是 Di
接下去 M行每行两个整数 Hi,Ti
输出
输出一个整数表示答案。
样例输出
4 6 2
1 3 5
1 0
2 1
4 9
1 10
2 10
3 12
样例输出
3
提示
对于全部数据,
2
≤
N
≤
1
0
5
,
1
≤
M
≤
1
0
5
,
1
≤
p
≤
100
,
1
≤
D
i
<
1
0
4
,
1
≤
H
i
≤
N
,
0
≤
T
i
≤
1
0
9
2≤N≤10^5,1≤M≤10^5,1≤p≤100,1≤Di<10^4,1≤Hi≤N,0≤Ti≤10^9
2≤N≤105,1≤M≤105,1≤p≤100,1≤Di<104,1≤Hi≤N,0≤Ti≤109
SOL
二维的斜率优化DP,注意开long long。
对于第
i
i
i只猫,饲养员最早必须在
a
[
i
]
a[i]
a[i]时刻出发才能接到她。
则
a
[
i
]
=
∑
j
=
1
i
a
[
j
]
a[i]=\sum_{j=1}^{i}a[j]
a[i]=j=1∑ia[j]
显然可以对a数组进行一次排序,然后可以用s数组存前缀和。
设
f
[
i
]
[
j
]
f[i][j]
f[i][j]表示前
i
i
i个饲养员回收
j
j
j只猫的最小等待时间。
则:
f
[
i
]
[
j
]
=
m
i
n
(
f
[
i
−
1
]
[
k
]
+
a
[
j
]
∗
(
j
−
k
)
−
(
s
[
j
]
−
s
[
k
]
)
)
,
0
<
=
k
<
j
f[i][j]=min(f[i-1][k]+a[j]*(j-k)-(s[j]-s[k])),0<=k<j
f[i][j]=min(f[i−1][k]+a[j]∗(j−k)−(s[j]−s[k])),0<=k<j
然后进行斜率优化。
去掉min函数,调整一下左右位置,可以得到:
f
[
i
−
1
]
[
k
]
+
s
[
k
]
=
a
[
j
]
∗
k
+
f
[
i
]
[
j
]
+
s
[
j
]
−
a
[
j
]
∗
j
f[i-1][k]+s[k]=a[j]*k+f[i][j]+s[j]-a[j]*j
f[i−1][k]+s[k]=a[j]∗k+f[i][j]+s[j]−a[j]∗j
于是有了一个自变量为
k
k
k,因变量为
f
[
i
−
1
]
[
k
]
f[i-1][k]
f[i−1][k]的函数。
类似任务安排那道题,维护一个相邻点连线斜率递增的队列,同时保证斜率不小于
a
[
j
]
a[j]
a[j],得到的最小截距即为当前子问题的答案。
代码:
#include<bits/stdc++.h>
#define N 100005
#define int long long
using namespace std;
inline int rd(){
int data=0,w=1;static char ch=0;ch=getchar();
while(!isdigit(ch)&&ch!='-')ch=getchar();
if(ch=='-')w=-1,ch=getchar();
while(isdigit(ch))data=(data<<1)+(data<<3)+ch-'0',ch=getchar();
return data*w;
}
int n,m,p,l,r;
int d[N],a[N],s[N],g[N],q[N],f[110][N];
signed main(){
n=rd();m=rd();p=rd();
for(int register i=2;i<=n;i++){int di=rd();d[i]=d[i-1]+di;}
for(int register i=1;i<=m;i++){
int hi=rd(),ti=rd();
a[i]=ti-d[hi];
}
sort(a+1,a+m+1);
for(int register i=1;i<=m;i++)s[i]=s[i-1]+a[i];
memset(f,0x5f,sizeof(f));
for(int register i=1;i<=p;i++){
for(int register j=1;j<=m;j++)g[j]=f[i-1][j]+s[j];
l=r=1;q[l]=0;
for(int register j=1;j<=m;j++){
while(l<r&&g[q[l+1]]-g[q[l]]<=a[j]*(q[l+1]-q[l]))l++;
f[i][j]=min(f[i-1][j],g[q[l]]+a[j]*(j-q[l])-s[j]);
while(l<r&&(g[j]-g[q[r]])*(q[r]-q[r-1])<=(g[q[r]]-g[q[r-1]])*(j-q[r]))r--;
q[++r]=j;
}
}
printf("%lld",f[p][m]);
return 0;
}