
题目描述 伐木工人米尔科需要砍倒M米长的木材。这是一个对米尔科来说很容易的工作,因为他有一个漂亮的新伐木机,可以像野火一样砍倒森林。不过,米尔科只被允许砍倒单行树木。 米尔科的伐木机工作过程如下:米尔科设置一个高度参数H(米),伐木机升起一个巨大的锯片到高度H,并锯掉所有的树比H高的部分(当然,树木不高于H米的部分保持不变)。米尔科就行到树木被锯下的部分。 例如,如果一行树的高度分别为20,15,10和17,米尔科把锯片升到15米的高度,切割后树木剩下的高度将是15,15,10和15,而米尔科将从第1棵树得到5米,从第4棵树得到2米,共得到7米木材。 米尔科非常关注生态保护,所以他不会砍掉过多的木材。这正是他为什么尽可能高地设定伐木机锯片的原因。帮助米尔科找到伐木机锯片的最大的整数高度H,使得他能得到木材至少为M米。换句话说,如果再升高1米,则他将得不到M米木材。 输入输出格式 输入格式: 第1行:2个整数N和M,N表示树木的数量(1<=N<=1000000),M表示需要的木材总长度(1<=M<=2000000000) 第2行:N个整数表示每棵树的高度,值均不超过1000000000。所有木材长度之和大于M,因此必有解。 输出格式: 第1行:1个整数,表示砍树的最高高度。 输入输出样例 输入样例#1: 5 20 4 42 40 26 46 输出样例#1: 36
解:对于此题,直接二分答案,然后判断当前答案
是否合法即可
复杂度O(n * logm)

1 #include<iostream> 2 #include<cstdio> 3 #include<algorithm> 4 #include<cmath> 5 #include<cstring> 6 #include<string> 7 #define ll long 8 #define DB double 9 using namespace std; 10 const int N=1e6+10; 11 int n,h[N]; 12 ll m,sum; 13 int L,R,mid,ans; 14 bool ok(int x) 15 { 16 sum=0; 17 for(int i=1;i<=n;++i) 18 if(h[i]-x>0) sum+=h[i]-x; 19 return sum>=m; 20 } 21 int main() 22 { 23 scanf("%d%lld",&n,&m); 24 for(int i=1;i<=n;++i) scanf("%d",&h[i]),R=max(R,h[i]); 25 sort(h+1,h+n+1); 26 while(L<=R) 27 { 28 mid=(L+R)>>1; 29 if(ok(mid)) ans=mid,L=mid+1; 30 else R=mid-1; 31 } 32 printf("%d",ans); 33 return 0; 34 }
一个优化,将树木高度从小到大排序,记录一个前缀和
对于当前答案mid,二分找到第一个>=他的位置,
判断 sum[n]-sum[p-1]-(n-p+1)*1ll*mid>=m;
复杂度O(logn * logm)

1 #include<iostream> 2 #include<cstdio> 3 #include<algorithm> 4 #include<cmath> 5 #include<cstring> 6 #include<string> 7 #define ll long 8 #define DB double 9 using namespace std; 10 const int N=1e6+10; 11 int n,h[N],p; 12 ll m,sum[N],tmp; 13 int L,R,mid,ans; 14 bool ok(int x) 15 { 16 p=lower_bound(h+1,h+n+1,x)-h; 17 tmp=sum[n]-sum[p-1];tmp-=(n-p+1)*1ll*x; 18 return tmp>=m; 19 } 20 int main() 21 { 22 scanf("%d%lld",&n,&m); 23 for(int i=1;i<=n;++i) scanf("%d",&h[i]),R=max(R,h[i]); 24 sort(h+1,h+n+1); 25 for(int i=1;i<=n;++i) sum[i]=sum[i-1]+h[i]; 26 while(L<=R) 27 { 28 mid=(L+R)>>1; 29 if(ok(mid)) ans=mid,L=mid+1; 30 else R=mid-1; 31 } 32 printf("%d",ans); 33 return 0; 34 }