bzoj 2876: [Noi2012]骑行川藏 拉格朗日乘子法

题意

蛋蛋非常热衷于挑战自我，今年暑假他准备沿川藏线骑着自行车从成都前往拉萨。川藏线的沿途有着非常美丽的风景，但在这一路上也有着很多的艰难险阻，路况变化多端，而蛋蛋的体力十分有限，因此在每天的骑行前设定好目的地、同时合理分配好自己的体力是一件非常重要的事情。
由于蛋蛋装备了一辆非常好的自行车，因此在骑行过程中可以认为他仅在克服风阻做功（不受自行车本身摩擦力以及自行车与地面的摩擦力影响）。某一天他打算骑N段路，每一段内的路况可视为相同：对于第i段路，我们给出有关这段路况的3个参数 si , ki , vi’ ，其中 si 表示这段路的长度， ki 表示这段路的风阻系数， vi’ 表示这段路上的风速（表示在这段路上他遇到了顺风，反之则意味着他将受逆风影响）。若某一时刻在这段路上骑车速度为v，则他受到的风阻大小为 F = ki ( v - vi’ )2（这样若在长度为s的路程内保持骑行速度v不变，则他消耗能量（做功）E = ki ( v - vi’ )2 s）。
设蛋蛋在这天开始时的体能值是 Eu ，请帮助他设计一种行车方案，使他在有限的体力内用最短的时间到达目的地。请告诉他最短的时间T是多少。

【评分方法】
本题没有部分分，你程序的输出只有和标准答案的差距不超过0.000001时，才能获得该测试点的满分，否则不得分。

【数据规模与约定】
对于10%的数据，N=1；
对于40%的数据，N<=2；
对于60%的数据，N<=100；
对于80%的数据，N<=1000；
对于所有数据，N <= 10000，0 <= Eu <= 108，0 < si <= 100000，0 < ki <= 1，-100 < vi’ < 100。数据保证最终的答案不会超过105。

【提示】
必然存在一种最优的体力方案满足：蛋蛋在每段路上都采用匀速骑行的方式。

分析

我们可以列出这样的一个东西

$$\begin{aligned} \sum\limits_{i=1}^{n} s_ik_i(v_i-v_i')^2 = E \\ \min(\sum\limits_{i=1}^{n} \frac{s_i}{v_i}) \end{aligned}$$

这相当于把下面的多元函数看成f,上面的多元函数看成g，要找到

$$f(x_1,x_2,x_3,...,x_n)$$
的极值点，使得满足
$$g(x_1,x_2,x_3,...,x_n) = m$$

于是根据拉格朗日乘子法，有

$$\nabla{f} = \lambda\nabla{g}$$

我们对每个变量求一维偏导，对某个xi求偏导就是说把其他变量看成常量来求导
就有了

$$\begin{aligned} -\frac{s_i}{v_i^2} = \lambda 2s_ik_i(v_i-v_i')\\ \sum\limits_{i=1}^{n}s_ik_i(v_i-v_i') = E \end{aligned}$$
化简一下：

$$\begin{aligned} 2\lambda k_i(v_i-v_i')v_i^2 = -1\\ \sum\limits_{i=1}^{n}s_ik_i(v_i-v_i') = E \end{aligned}$$
讨论一波
首先第一点，无论顺风和逆风，vi > vi’,所以的话lambda肯定小于0，然后lambda的绝对值越大，vi的绝对值越小，导致E变小，我们可以发现这样满足了二分性。
我们可以二分lambda，再对每个i二分速度vi‘，就可以完成这道题了

时间复杂度

$$O(Nlog^2x)$$
x取决于你的上界和精度

代码

#include <bits/stdc++.h>
#define db long double
using namespace std;
const int N = 100010;
const db eps = 1e-16;
const db inf = 2e9;
db s[N],k[N],v[N],e,v1[N]; int n;
bool check(db lambda)
{
  db tmp = 0.0;
  for(int i=1;i<=n;i++)
  {
    db l = max((db)0.0,v[i]); db r = inf; db mid;
    while(r-l>=eps)
    {
      mid = (l+r)/2.0;
      if(lambda * 2 * k[i] * (mid - v[i]) * mid * mid <= -1) r=mid;
      else l=mid;
    }
    // printf("%.f\n",mid) ;
    v1[i] = mid; tmp += s[i] * k[i] * (mid - v[i]) * (mid - v[i]);
  }
  // printf("%.10f\n",tmp);
  return (tmp <= e);
}
int main()
{
  cin >> n >> e;
  for(int i=1;i<=n;i++) cin >> s[i] >> k[i] >> v[i];
  db l = -inf,r = 0,mid;
  while(r-l>=eps)
  {
    mid = (l+r) / 2.0;
    if(check(mid)) l=mid;
    else r=mid;
  }
  db ans=0.0; for(int i=1;i<=n;i++) ans = ans + s[i] / v1[i]; 
  return printf("%.8Lf\n",ans),0;
}