HDU2993 MAX average problem [斜率dp]

题意：给你一段长度为n的数列, 求其长度不小于 K 的平均值最大的子串。

解析：抄的解析http://www.docin.com/p-469236754.html 例二

[分析]

简单的枚举算法可以这样描述：每次枚举一对满足条件的(a, b)，即a≤b-F+1，检查ave(a, b)，并更新当前最大值。

然而这题中N很大，N²的枚举算法显然不能使用，但是能不能优化一下这个效率不高的算法呢？答案是肯定的。

目标图形化

首先一定会设序列a_i的部分和：S_i=a₁+a₂+…+a_i，，特别的定义S₀=0。

这样可以很简洁的表示出目标函数！

如果将S函数绘在平面直角坐标系内，这就是过点S_j和点S_i-1直线的斜率！

于是问题转化为：平面上已知N+1个点，P_i(i, S_i)，0≤i≤N，求横向距离大于等于F的任意两点连线的最大斜率。

构造下凸折线

有序化一下，规定对i<j，只检查P_j向P_i的连线，对P_i不检查与P_j的连线。也就是说对任意一点，仅检查该点与在其前方的点的斜率。于是我们定义点P_i的检查集合为

G_i = {P_j, 0≤j≤i-F}

特别的，当i<F时，G_i为空集。

其明确的物理意义为：在平方级算法中，若要检查ave(a, b)，那么一定有P_a∈G_b；因此平方级的算法也可以这样描述，首先依次枚举P_b点，再枚举P_a∈G_b，同时检查k(P_aP_b)。

若将P_i和G_i同时列出，则不妨称P_i为检查点，G_i中的元素都是P_i的被检查点。

当我们考察一个点P_t时，朴素的平方级算法依次选取G_t中的每一个被检查点p，考察直线pP_t的斜率。但仔细观察，若集合内存在三个点P_i, P_j, P_k，且i<j<k，三个点形成如下图所示的的关系，即P_j点在直线P_iP_k的上凸部分：k(P_i, P_j)>k(P_j, P_k)，就很容易可以证明P_j点是多余的。

图 2

若k(P_t, P_j) > k(P_t, P_i)，那么可以看出，P_t点一定要在直线P_iP_j的上方，即阴影所示的1号区域。同理若k(P_t, P_j) > k(P_t, P_k)，那么P_t点一定要在直线P_jP_k的下方，即阴影所示的2号区域。

综合上述两种情况，若P_tP_j的斜率同时大于P_tP_i和P_tP_k的，P_t点一定要落在两阴影的重叠部分，但这部分显然不满足开始时t>j的假设。于是，P_t落在任何一个合法的位置时，P_tP_j的斜率要么小于P_tP_i，要么小于P_tP_k，即不可能成为最大值，因此P_j点多余，完全可以从检查集合中删去。

这个结论告诉我们，任何一个点P_t的检查集合中，不可能存在一个对最优结果有贡献的上凸点，因此我们可以删去每一个上凸点，剩下的则是一个下凸折线。最后需要在这个下凸折线上找一点与P_t点构成的直线斜率最大——显然这条直线是在与折线相切时斜率最大，如图所示。

图 3

维护下凸折线

这一小节中，我们的目标是：用尽可能少的时间得到每一个检查点的下凸折线。

算法首先从P_F开始执行：它是检查集合非空的最左边的一个点，集合内仅有一个元素P₀，而这显然满足下凸折线的要求，接着向右不停的检查新的点：P_F+1, P_F+2, …, P_N。

检查的过程中，维护这个下凸折线：每检查一个新的点P_t，就可以向折线最右端加入一个新的点P_t-F，同时新点的加入可能会导致折线右端的一些点变成上凸点，我们用一个类似于构造凸包的过程依次删去这些上凸点，从而保证折线的下凸性。由于每个点仅被加入和删除一次，所以每次维护下凸折线的平摊复杂度为O(1)，即我们用O(N)的时间得到了每个检查集合的下凸折线。

最后的优化：利用图形的单调性

最后一个问题就是如何求过P_t点，且与折线相切的直线了。一种直接的方法就是二分，每次查找的复杂度是O(log₂N)。但是从图形的性质上很容易得到另一种更简便更迅速的方法：由于折线上过每一个点切线的斜率都是一定的[1]，而且根据下凸函数斜率的单调性，如果在检查点P_t时找到了折线上的已知一个切点A，那么A以前的所有点都可以删除了：过这些点的切线斜率一定小于已知最优解，不会做出更大的贡献了。

于是另外保留一个指针不回溯的向后移动以寻找切线斜率即可，平摊复杂度为为O(1)。

至此，此题算法时空复杂度均为O(N)，得到了圆满的解决。

 

---

[1] 由于折线没有连续性，因此更准确的应该说，过每一个点切线斜率的范围都一定的。

 

N(log₂N) 和  O(N)的两种算法的代码如下

N(log₂N) 算法

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

const int N = 100005;
int sum[N];

struct point{
    point(){};
    point(int x, int y) : x(x),y (y){}
    int x, y;
}s[N];

long long cross(point a, point b, point c) { //ab * bc
    return (long long)(b.x - a.x) * (c.y - b.y) - (long long)(c.x - b.x) * (b.y - a.y);
}

int bsearch(int l, int r, point &now) {
    while(l < r){
        int mid = (l + r) >> 1;
        if(cross(s[mid], s[mid + 1], now) < 0)  r = mid;
        else l = mid + 1;
    }
    return l;
}

double solve(int n, int m){
    double maxx = 0;
    int p = -1;
    for(int i = m; i <= n; i ++){
        point r(i - m, sum[i - m]), now(i, sum[i]);
        while(p > 0 && cross(s[p - 1], s[p], r) < 0) p --;
        s[++p] = r;
        int opt = bsearch(0, p, now);
        double ans = (double)( sum[i] - s[opt].y ) / (i - s[opt].x);
        if(ans > maxx) maxx = ans;
    }
    return maxx;
}
void read(int &r) {
    char c;
    while (c = getchar(), c < '0' || c > '9');
    r = c - '0';
    while (c = getchar(), c >= '0' && c <= '9') r = r * 10 + c - '0';
}
int main(){
    int n, m, i;
    while(~scanf("%d%d", &n, &m)){
        sum[0] = 0;
        for(i = 1; i <= n; i ++){
            //read(sum[i]);
            scanf("%d", &sum[i]);
            sum[i] += sum[i - 1];
        }
        printf("%.2f\n", solve(n, m));
    }
}

O(N)算法

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

const int N = 100005;
int sum[N];

struct point{
    point(){};
    point(int x, int y) : x(x),y (y){}
    int x, y;
}s[N];

long long cross(point a, point b, point c) { //ab * bc
    return (long long)(b.x - a.x) * (c.y - b.y) - (long long)(c.x - b.x) * (b.y - a.y);
}

double solve(int n, int m){
    double maxx = 0;
    int p = -1, h = 0;
    for(int i = m; i <= n; i ++){
        point r(i - m, sum[i - m]), now(i, sum[i]);
        while(p > h && cross(s[p - 1], s[p], r) < 0) p --;
        s[++p] = r;
        while(h < p && (long long)(sum[i] - s[h].y) * (i - s[h + 1].x) < (long long)(sum[i] - s[h + 1].y) * (i - s[h].x) ) h ++;
        double ans = (double)( sum[i] - s[h].y ) / (i - s[h].x);
        if(ans > maxx) maxx = ans;
    }
    return maxx;
}
void read(int &r) {
    char c;
    while (c = getchar(), c < '0' || c > '9');
    r = c - '0';
    while (c = getchar(), c >= '0' && c <= '9') r = r * 10 + c - '0';
}
int main(){
    int n, m, i;
    while(~scanf("%d%d", &n, &m)){
        sum[0] = 0;
        for(i = 1; i <= n; i ++){
            read(sum[i]);
            //scanf("%d", &sum[i]);
            sum[i] += sum[i - 1];
        }
        printf("%.2f\n", solve(n, m));
    }
}