51NOD1281 山峰和旗子 【二分】

1281 山峰和旗子

题目来源：  Codility

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

用一个长度为N的整数数组A，描述山峰和山谷的高度。山峰需要满足如下条件， 0 < P < N - 1 且 A[P - 1] < A[P] > A[P + 1]。

现在要在山峰上插上K个旗子，并且每个旗子之间的距离 >= K，问最多能插上多少个旗子（即求K的最大值）。两个山峰之间的距离为|P - Q|。

以上图为例，高度为：1 5 3 4 3 4 1 2 3 4 6 2。其中可以作为山峰的点为：1 3 5 10。

放2面旗子, 可以放在1 和 5。

放3面旗子, 可以放在1 5 和 10。

放4面旗子, 可以放在1 5 和 10，之后就放不下了。

所以最多可以放3面旗子。

Input

第1行：一个数N，表示数组的长度(1 <= N <= 50000)。
第2 - N + 1行：每行1个数Ai(1 <= Ai <= 10^9)。

Output

输出最多能插上多少面旗子（即求K的最大值）。

Input示例

12
1 
5 
3 
4 
3 
4 
1 
2 
3 
4 
6 
2

Output示例

3

李陶冶  (题目提供者)

二分k即可

#include <bits/stdc++.h>

#define ll long long
#define pii pair<int,int>
#define MEM(a,x) memset(a,x,sizeof(a))

using namespace std;

const int inf=1e9+7;

const int N = 5e4+5;

int a[N];
vector<int>vec;

bool check(int k){
    if(k==0){
        return true;
    }
    if(k==1){
        return vec.size();
    }
    int st=0;
    for(int i=1;i<k;++i){
        auto it=lower_bound(vec.begin(),vec.end(),st+k);
        if(it==vec.end()){
            return false;
        }
        st=*it;
    }
    return true;
}

int slove(int n){
    vec.clear();
    for(int i=1;i<n-1;++i){
        if(a[i]>a[i-1]&&a[i]>a[i+1]){
            vec.push_back(i);
        }
    }
    int l=0,r=n,ans=0;
    while(l<=r){
        int mid=(l+r)/2;
        if(check(mid)){
            l=mid+1;
            ans=mid;
        }
        else{
            r=mid-1;
        }
    }
    return ans;
}

int main()
{
    //freopen("/home/lu/code/r.txt","r",stdin);
    int n;
    while(~scanf("%d",&n)){
        for(int i=0;i<n;++i){
            scanf("%d",&a[i]);
        }
        printf("%d\n",slove(n));
    }
    return 0;
}