题意:
有n个点,2n-3条边的图。点 1 到每个点有一条边,编号相邻的两个点有一条边。q次询问,每次询问一个增量d,问图中每条边都增加 d 后,1 到 n 的最短路是多少。增量独立,不累加。
1 <= n,q <= 1e5
分析:
用 dp[i] 表示经过 i 条边到达 n 的最短路长度,每条路是唯一的。如果每条边增加了长度 d 后最短路发生了改变,那新的最短路一定是边数更少了。所以问题转化成了对于所有 dp[i] ,面对一个增量 k,dp[i] + k * i 最小值是多少。
若dp[i] < dp[j] && i > j, 假如增加增量后dp[i] + k * i >= dp[j] + k* j,即最短路换成了 j 条边的,有 (dp[j] - dp[i]) / (j - i) >= -k。
维护一个边数由少到多,距离由多到少的单调栈。使得相邻两点间的斜率慢慢变大(斜率都是负数,斜率绝对值是慢慢变小的)。
考虑这样一种情况,栈中原本有A和B,此时C要入栈,我们应该把B踢出栈。为什么呢?一种简单的思考方法是,假如ABC共线,那么取到最小值的不是C就是AB,A和B是一样的。那现在B高出来了一点点,而A没变,B显然就竞争不过A,B永远取不到最小值。另一种考虑方法是,写出AB,AC,BC的斜率,分类讨论可以发现,无论增量 k 取何值,最小值都是A或C。
解决了这个问题,就可以维护一个这个样子的栈。
将增量由小到大排序,然后逐个查询。因为 k 由小到大,所以-k由大到小,越靠后的 k 越有可能在栈中比较的越远。
大佬写的代码:
#include <bits/stdc++.h>
using namespace std;
struct Node{
int x;
long long y;
Node(){}
Node(int x,long long y):x(x),y(y){}
};
int cnt;
int last;
const int N = 1e5+10;
int a[N],aa[N];
long long sum[N],dp[N];
Node T[N];
struct Query{
int id;
int val;
long long ans;
};
Query que[N];
bool cmp(Query a,Query b){
return a.val<b.val;
}
bool cmp2(Query a,Query b){
return a.id<b.id;
}
long long X(Node a, Node b, Node c){
return (b.x-a.x)*(c.y-a.y)-(c.x-a.x)*(b.y-a.y);
}
bool pd(Node a,Node b, long long c){
return (a.y-b.y)<=c*(b.x-a.x);
}
int main()
{
//freopen("1.txt","r",stdin);
int Case;
scanf("%d",&Case);
while (Case--){
int n;
scanf("%d",&n);
int q;
scanf("%d",&q);
for (int i=2;i<=n;i++){
scanf("%d",&a[i]);
}
for (int i=2;i<=n-1;i++){
scanf("%d",&aa[i]);
}
aa[1] = a[2];
sum[n]=0;
for (int i=n-1;i>=1;i--) sum[i]=sum[i+1]+aa[i];
cnt=0;
for (int i=1;i<=n-1;i++){
dp[i] = a[n-i+1] + sum[n-i+1];
if (cnt && T[cnt-1].y<=dp[i]) continue;
while (cnt>=2 && X(T[cnt-2],T[cnt-1],Node(i,dp[i]))<=0) cnt--;
T[cnt++] = Node(i,dp[i]);
}
for (int i=1;i<=q;i++){
que[i].id = i;
scanf("%d",&que[i].val);
}
sort(que+1,que+q+1,cmp);
for (int i=1;i<=q;i++){
while (cnt>=2 && pd(T[cnt-2],T[cnt-1],que[i].val)) cnt--;
int last = T[cnt-1].x;
que[i].ans = dp[last]+que[i].val*last;
}
sort(que+1,que+q+1,cmp2);
for (int i=1;i<=q;i++){
printf("%lld",que[i].ans);
if (i==q) printf("\n");else printf(" ");
}
}
return 0;
}