1. 程式人生 > >poj2456(貪心+二分)

poj2456(貪心+二分)

/*
題意:
    有n個牛欄,選m個放進牛,相當於一條線段上有 n 個點,選取 m 個點,
使得相鄰點之間的最小距離值最大

思路:貪心+二分
    二分列舉相鄰兩牛的間距,判斷大於等於此間距下能否放進所有的牛。
*/
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int N = 1e6+10;
int a[N],n,m;
bool judge(int k)//列舉間距k,看能否使任意兩相鄰牛
{
    int cnt = a[0], num = 1;//num為1表示已經第一頭牛放在a[0]牛欄中
    for(int i = 1; i < n; i ++)//列舉剩下的牛欄
    {
        if(a[i] - cnt >= k)//a[i]這個牛欄和上一個牛欄間距大於等於k,表示可以再放進牛
        {
            cnt = a[i];
            num ++;//又放進了一頭牛
        }
        if(num >= m) return true;//所有牛都放完了
    }
    return false;
}
void solve()
{
    int l = 1, r = a[n-1] - a[0];//最小距離為1,最大距離為牛欄編號最大的減去編號最小的
    while(l < r)
    {
        int mid = (l+r) >> 1;
        if(judge(mid)) l = mid + 1;
        else r = mid;
    }
    printf("%d\n",r-1);
}
int main()
{
    int i;
    while(~scanf("%d%d",&n,&m))
    {
        for(i = 0; i < n; i ++)
            scanf("%d",&a[i]);
        sort(a, a+n);//對牛欄排序
        solve();
    }
    return 0;
}