【洛谷p3994】Highway 二分+斜率优化DP

题目大意:给你一颗$n$个点的有根树,相邻两个点之间有距离,我们可以从$x$乘车到$x$的祖先,费用为$dis\times P[x]+Q[x]$,问你除根以外每个点到根的最小花费。

数据范围:$n≤10^6$。

此题我们显然$dp$,列出方程为$f[x]=min\{f[y]+dis(x,y)\times P[x]+Q[x]\}$,其中$y$为$x$的祖先。

不难看出可能是一个斜率优化的式子,我们往下推推

设$i$是$j$的祖先,且从$i$出转移比从$j$处转移劣,不难列出:

$f[i]+dis(i,x)P[x]+Q[x]>f[j]+dis(j,x)P[x]+Q[x]$

化简化简,令$D_i$表示从根到$i$的距离,继续移项化简

$f[i]-f[j]>P[x]((D_x-D_j)-(D_x-D_i))$

$f[i]-f[j]>P[x](D_i-D_j)$

$\dfrac{f[i]-f[j]}{D_i-D_j}<P[x]$(注意大于符号变成小于,因为$D_i<D_j$,重要!,我被坑了)

考虑到$P$随着点深度递增,于是就可以快乐斜率优化了。

 

不过这题需要在树上转移,当遍历完一个子树后,我们要把单调队列恢复,恢复操作显然可以用可持久化线段树来搞。

其实不用这么复杂,考虑到只存在移动队头,移动队尾&在队尾加一个数的操作,我们只需要记录下以前的队头/尾的情况,还有被覆盖的数原先是啥,就可以在常数时间内恢复回去。

然而这么做的话无法保证移动队头尾的次数是线性的(我们搞一个链+链底菊花树),所以在找队头和队尾时,需要用二分。

于是时间复杂度就变成$O(n\log\ n)$了,二分的常数很小,问题不大。

(我比较懒写了暴力移动的,二分的话自己脑补吧qwq)

 1 #include<bits/stdc++.h>
 2 #define M 1000005
 3 #define L long long
 4 using namespace std;
 5 
 6 struct edge{L u,v,next;}e[M]={0}; L head[M]={0},use=0;
 7 void add(L x,L y,L z){use++;e[use].u=y;e[use].v=z;e[use].next=head[x];head[x]=use;}
 8 L n,f[M]={0},P[M]={0},Q[M]={0},dis[M]={0},q[M]={0},h=1,t=0;
 9 
10 double slope(L i,L j){return 1.*(f[i]-f[j])/(dis[i]-dis[j]);}
11 
12 void dfs(L x,L D){
13     L H=h,T=t;
14     while(h<t&&P[x]>slope(q[h],q[h+1])) h++;
15     f[x]=f[q[h]]+((dis[x]=D)-dis[q[h]])*P[x]+Q[x];
16     while(h<t&&slope(q[t-1],q[t])>slope(q[t],x)) t--;
17     L lastT=q[++t]; q[t]=x;
18     for(L i=head[x];i;i=e[i].next) dfs(e[i].u,D+e[i].v);
19     q[t]=lastT; h=H; t=T;
20 }    
21 
22 main(){
23     scanf("%lld",&n);
24     for(L i=2;i<=n;i++){
25         L fa,dis; scanf("%lld%lld%lld%lld",&fa,&dis,P+i,Q+i);
26         add(fa,i,dis);
27     }
28     q[t=1]=1; for(L i=head[1];i;i=e[i].next) dfs(e[i].u,e[i].v);
29     for(L i=2;i<=n;i++) printf("%lld\n",f[i]);
30 }
posted @ 2019-02-23 21:01  AlphaInf  阅读(242)  评论(0编辑  收藏  举报