1011. 在 D 天内送达包裹的能力

最新推荐文章于 2025-12-27 20:58:23 发布

原创最新推荐文章于 2025-12-27 20:58:23 发布 · 80 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

刷题心得专栏收录该内容

41 篇文章

订阅专栏

博客围绕LeetCode题目展开，作者提交的代码通过了所有测试用例，但出现超出时间限制的问题，原因是时间复杂度为O(n2)。作者认为可将复杂度降低到O(nlogn)，并提到使用二分法查找来解决，还提及了上下限。

题目链接：leetcode.

我就知道
84 / 84 个通过测试用例
状态：超出时间限制

class Solution {
public:
    int shipWithinDays(vector<int>& weights, int D) {
    	int N = weights.size();
        //先找出最小的数（即weights的最大值）
		int ans = 0;
		for(auto x : weights)
		{
			ans = max(ans, x);
		} 
		//一个一个往上加，看天数是否小于等于D
		while(true)
		{
			int cnt = 1, tmp = 0;
			for(int i = 0;i < N;++i)
			{
				tmp += weights[i];
				if(tmp > ans)
				{
					cnt++;
					tmp = weights[i];
				}
			}
            // cout << cnt << endl;
			if(cnt <= D)
			{
				return ans;
			}
			else
			{
				ans++;
			}
		} 
		return -1;
    }
};

呐，我这个超时呢，是因为O(n2)了，降低时间复杂度的话，只能O(nlogn)啊
二分法查找合适的ans啊，这太可以了
ans上限就是sum(weights)啊，下限就是max(weights)啊

/*
执行用时：64 ms, 在所有 C++ 提交中击败了54.82%的用户
内存消耗：25.5 MB, 在所有 C++ 提交中击败了22.40%的用户
*/
class Solution {
public:
    int shipWithinDays(vector<int>& weights, int D) {
    	int N = weights.size();
        //先找出左右边界
		int left = 0, right = 0;
		for(auto x : weights)
		{
			left = max(left, x);
			right += x;
		} 
		//二分法
		while(left < right)
		{
			int mid = left + (right - left) / 2;
			int cnt = 1, tmp = 0;
			for(int i = 0;i < N;++i)
			{
				tmp += weights[i];
				if(tmp > mid)
				{
					cnt++;
					tmp = weights[i];
				}
			}
			if(cnt <= D)
			{
				right = mid;
			}
			else
			{
				left = mid + 1;
			}
		} 
		return left;
    }
};