poj 3258 River Hopscotch(二分搜索)

又是一道二分搜索题,二分搜索思路就是通过二分假设1个答案,然后写一个函数验证这个答案是否可行,根据这个结果去继续二分求解最终答案。

本题先输入n个点,加上首尾两个点排序后产生n+1个间隔,记录下这些间隔。

验证答案时,考虑假设最小间隔是t,那么一旦遇到一个比t小的间隔,就要把它和相邻的间隔合并,如果前面的间隔都可以保证是大于等于t的,那么和前面的合并是没有意义的,所以要选择和后面的合并,并把可拆岩石数-1。这样扫过去是O(n)的。整体时间是nlogn的

判断答案不行有两种可能:1.需要拆除的岩石数大于给定的 2.累加到最后仍无法使某个间隔大于t


注意:二分时r等于的值是永远不可能被取到的!所以开始r应该等于L+1而不是L,wa了一次。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int L,n,m;
int a[50005],pos[50005];
bool C(int t){
    long long sum=0;
    int i=1,k=0;
    while(1){
        sum=0;
        k-=1;
        while(i<=n+1&&sum<t){
            sum+=a[i];
            ++k;
            if(k>m) return 0;
            ++i;
        }
        if(sum<t) return 0;
        if(i>n+1) break;
    }
    return 1;
}

int main(){
    scanf("%d%d%d",&L,&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&pos[i]);
    }
    pos[0]=0;pos[n+1]=L;
    sort(pos,pos+n+2);
    for(int i=1;i<=n+1;i++){
        a[i]=pos[i]-pos[i-1];
    }
    int l=0,r=L+1;
    int mid;
    while(r-l>1){
        mid=(r+l)/2;
        if(C(mid)) l=mid;
        else r=mid;
    }
    printf("%d",l);
    return 0;
}


你可能感兴趣的:(二分搜索)