首页 > > 程序设计 > C/C++ >

bzoj1492 [ NOI2007 ] --斜率优化DP+cdq分治

2018-06-17 23:20:39来源：未知阅读 ()

显然在某一天要么花完所有钱，要么不花钱。

所以首先想到O(n^2)DP：

f[i]=max{f[i-1],(f[j]*r[j]*a[i]+f[j]*b[i])/(a[j]*r[j]+b[j])},j<i

其中f[j]*r[j]/(a[j]*r[j]+b[j])是第j天最多能买多少A券，B类似。

假如我们已经找到了最优的j，那么有f[i]=(f[j]*r[j]*a[i]+f[j]*b[i])/(a[j]*r[j]+b[j])

令x[j]=f[j]*r[j]/(a[j]*r[j]+b[j])，y[j]=f[j]/(a[j]*r[j]+b[j])

那么有f[i]=x[j]*a[i]+y[j]*b[i]

y[j]=(-a[i]/b[i])*x[j]+f[i]/b[i]
这是一条斜率为-a[i]/b[i]的直线。

由于b[i]不变，而我们要最大化f[i]，所以现在我们可以将问题转化为从一些点中选出一个点，使经过这点的已知斜率的直线的截距最大。

但因为斜率和坐标都是无序的，我们用splay维护一个上凸壳，在求f[i]时从上凸壳中选出一个点j，使j点左右两边的斜率刚好将-a[i]/b[i]夹住，那么i就从j转移。

但这题还有更神奇的cdq分治做法。

我们可以想想，为什么斜率和坐标都是无序的呢？因为我们从1~n依次求解。那能否按照斜率排序后一一求解呢？答案是能。

处理区间[l,r]时先将[l,mid]建成一个上凸壳，然后将[mid+1,r]按斜率排序并更新就可以了。

时间复杂度O(n*logn)

代码:

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<cmath>
 5 #include<algorithm>
 6 using namespace std;
 7 #define N 100001
 8 #define Eps 1e-9
 9 #define INF 0x3f3f3f3f3f3f3f3f
10 struct Node{
11   double x,y,A,B,r,k;
12   int Id;
13 }a[N],t[N];
14 double f[N];
15 int i,j,St[N],Top,n,m;
16 inline bool Cmp(Node a,Node b){return a.k<b.k;}
17 inline double _Max(double x,double y){return x<y?y:x;}
18 inline double Slop(int x,int y){
19   if(!y)return -1e20;
20   if(fabs(a[x].x-a[y].x)<Eps)return 1e20;
21   return (a[x].y-a[y].y)/(a[x].x-a[y].x);
22 }
23 inline void Solve(int l,int r){
24   if(l==r){
25     f[l]=_Max(f[l],f[l-1]);
26     a[l].y=f[l]/(a[l].A*a[l].r+a[l].B);
27     a[l].x=a[l].y*a[l].r;
28     return;
29   }
30   int Mid=l+r>>1;
31   int l1=l,l2=Mid+1;
32   for(int i=l;i<=r;i++)
33     if(a[i].Id<=Mid)t[l1++]=a[i];else t[l2++]=a[i];
34   for(int i=l;i<=r;i++)a[i]=t[i];
35   Solve(l,Mid);
36   Top=0;
37   for(int i=l;i<=Mid;i++){
38     while(Top>1&&Slop(St[Top-1],i)+Eps>Slop(St[Top-1],St[Top]))Top--;
39     St[++Top]=i;
40   }
41   int j=1;
42   St[++Top]=0;
43   for(int i=r;i>Mid;i--){
44     while(j<Top&&a[i].k<Slop(St[j],St[j+1])+Eps)j++;
45     f[a[i].Id]=_Max(f[a[i].Id],a[St[j]].x*a[i].A+a[St[j]].y*a[i].B);
46   }
47   Solve(Mid+1,r);
48   l1=l,l2=Mid+1;
49   for(int i=l;i<=r;i++)
50     if(l1>Mid)t[i]=a[l2++];else
51       if(l2>r)t[i]=a[l1++];else
52     if(a[l1].x<a[l2].x||(fabs(a[l1].x-a[l2].x)<Eps&&a[l1].y<a[l2].y))t[i]=a[l1++];else t[i]=a[l2++];
53   for(int i=l;i<=r;i++)a[i]=t[i];
54 }
55 int main()
56 {
57   scanf("%d%lf",&n,&f[0]);
58   for(i=1;i<=n;i++){
59     scanf("%lf%lf%lf",&a[i].A,&a[i].B,&a[i].r);
60     a[i].Id=i;a[i].k=-a[i].A/a[i].B;
61   }
62   sort(a+1,a+n+1,Cmp);
63   Solve(1,n);
64   printf("%.3lf",f[n]);
65   return 0;
66 }