【二分】B027_NK_苹果树（二分找符合累加任务的最小索引）

一、Problem

牛牛有一个苹果园。又到了一年一度的收获季，牛牛现在要去采摘苹果买给市场的摊贩们。

牛牛的果园里面有 n 棵苹果树，第i棵苹果树上有 ai 个果子。
牛牛为了保证果子的新鲜程度，每天都会去苹果树上采摘果子。
牛牛特意安排一个计划表：

计划 m 天去采摘果子。

• 对于第 i 天，它会去所有果树上轮流采摘 bi 个果子。
• 如果对于第 i 天，某棵果树上没有 bi 个果子，那么它只会把当前果树上的果子采摘完。

牛牛想知道它每天能供应多少个苹果给市场的摊贩们

输入 [10,20,10],[5,7,2]
输出 [15,17,2]
说明 苹果树上的果子变化[10,20,10]–>[5,15,5]–>[0,8,0]–>[0,6,0]

$1<=a_i,b_i<=1e9$
$1<=n,m<=1e5$

二、Solution

TLE $O(n^2)$ 模拟摘苹果不难，故这里需要更聪明的解法

方法一：二分

思路

树的顺序不会影响结果，故可对 A 排序，然后用二分在 A 中找到一个能满足计划摘取的累积数量 now（即第 0 天到第 x 天的计划数量总和) 的最小索引 p：

• p ≠ n 时，则位置 p 后面的全部苹果树都可以在掉 now 那么多
• p ∈ [0, n) 时，证明有 [0, p] 位置这部分的树不可以被摘 now 个，但 [0, p] 位置这些数都可以被摘光

前缀和不能用 long 我也是服了，nlogn 也只能过 70.00% 我也服了

class Solution {
public:
	int binarySearch(vector<int>& A, int& p, int& now) {
		int n=A.size();
		if (now > A[n-1]) return n;
		if (now < A[0])   return 0;

		int l=p, r=n-1;
		while (l<r) {
			int m=l+(r-l)/2;
			if (A[m] >= now) r=m;
			else			 l=m+1;
		}
		return l;
	}
    vector<long> solve(vector<int>& A, vector<int>& B) {
    	sort(A.begin(), A.end());
    	int n=A.size(), m=B.size(), now=0, tot=0, p=0;
    	vector<long> ans(m, 0);
        vector<int> s(n+1); s[0]=0;
        for (int i = 0; i < n; ++i) s[i+1]=s[i]+A[i];

    	for (int i=0; i<m; i++) {
    		now+=B[i];
    		p=binarySearch(A, p, now);
    		ans[i] = (n-p)*now + s[p] - tot;
    		tot+=ans[i];
    	}
    	return ans;
    }
};

复杂度分析

• Time：$O(nlogn)$，
• Space：$O(n)$