nyoj 盖伦的告白（线段树||双端队列）

盖伦的告白

题目描述
盖伦和赵信这对基友又在打赌，谁输了就去向卡特琳娜告白。。

这一季中盖伦一如既往的怂。赵信为了帮他一把，故意表现出很厉害的样子，结果成功激活了盖伦的智商。

赵信手中有n张牌，每张牌上面有一个数，这些牌是非递减的。现在盖伦从他手中抽取k张牌，如果盖伦抽取后剩下的牌中相邻两张的差值的最大值最小，就算他赢。

结果当然是盖伦赢了。那么问题来了！

输入
第一行两个正整数n和k。

第二行n个数代表n张牌。

3≤n≤105

1≤k≤n-2

-109≤Ai≤109

输出
输出只有一个数，上述中的最小值。
样例输入
5 1
1 2 4 7 8
8 2
1 2 3 5 8 13 17 18
5 1
1 2 4 6 9
样例输出
3
5
2
题目链接：盖伦的告白

思路：
注意牌的大小是非递减的，当我们从中间删除元素的话将会产生更大的差值即：Ai+1-Ai-1>=max(Ai-Ai-1,Ai+1-Ai).

所以最优解是从两端删除元素，剩下的n-k个元素中查询区间最值即可。复杂度O(nlogn).

代码（线段树）：

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;

const int inf=0x3f3f3f3f;
const int maxn=1e5+2;
struct node
{
    int val,le,ri;
    int mid()
    {
        return (le+ri)>>1;
    }
} tree[maxn<<2];
int a[maxn],b[maxn];
int n,k;

void Build(int rt,int le,int ri)
{
    tree[rt].le=le;
    tree[rt].ri=ri;
    if(le==ri)
    {
        tree[rt].val=b[le];
        return ;
    }
    int mid=tree[rt].mid();
    Build(rt<<1,le,mid);
    Build(rt<<1|1,mid+1,ri);
    tree[rt].val=max(tree[rt<<1].val,tree[rt<<1|1].val);
}

int Query(int rt,int le,int ri)
{
    if(le<=tree[rt].le&&tree[rt].ri<=ri)
        return tree[rt].val;
    int mid=tree[rt].mid();
    if(ri<=mid)
        return Query(rt<<1,le,ri);
    else if(le>mid)
        return Query(rt<<1|1,le,ri);
    else
        return max(Query(rt<<1,le,mid),Query(rt<<1|1,mid+1,ri));
}

int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=1; i<=n; ++i)
        {
            scanf("%d",&a[i]);
            if(i>1)
                b[i-1]=a[i]-a[i-1];
        }
        Build(1,1,n-1);
        int ans=inf,len=n-k-1;
        for(int i=1; i+len<=n; ++i)
            ans=min(ans,Query(1,i,i+len-1));
        printf("%d\n",ans);
    }
    return 0;
}

代码（双端队列）：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

const int maxn=1e5+10;
int a[maxn],b[maxn],ser[maxn],max_num[maxn];//max_num存储的是非递增的最大值
int n,k;

int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=1; i<=n; ++i)
        {
            scanf("%d",&a[i]);
            if(i>1)
                b[i-1]=a[i]-a[i-1];
        }
        int head=0,tail=0;
        for(int i=1; i<n-k; ++i)
        {
            while(tail>head&&max_num[tail-1]<=b[i])
                --tail;
            max_num[tail]=b[i];
            ser[tail++]=i;
        }
        int ans=max_num[head];
        for(int i=n-k; i<n; ++i)
        {
            while(tail>head&&max_num[tail-1]<=b[i])
                --tail;
            max_num[tail]=b[i];
            ser[tail++]=i;
            while(head<tail&&ser[head]<=i-(n-k)+1)
                ++head;
            ans=min(ans,max_num[head]);
        }
        printf("%d\n",ans);
    }
    return 0;
}

ps：双端队列的思想真是厉害了，每次用到的max_num都是当前区间的最大值