【Leetcode】1891. Cutting Ribbons

记录算法题解

已于 2024-01-06 13:10:12 修改

阅读量464

点赞数 8

分类专栏： LC 二分、位运算与数学文章标签： leetcode 算法 c++

于 2024-01-05 17:08:28 首次发布

本文链接：https://blog.csdn.net/qq_46105170/article/details/135413805

版权

LC 二分、位运算与数学专栏收录该内容

397 篇文章 2 订阅

订阅专栏

文章讨论了解决LeetCode中关于切割木棍的问题，通过二分查找算法确定是否存在能用给定木棍截出至少k个长度为x的小木棍的方法，分析了时间复杂度为O(nlogmax(a))和空间复杂度为O(1)的解决方案。

摘要由CSDN通过智能技术生成

题目地址：

https://leetcode.com/problems/cutting-ribbons/description/

给定 $n$ 条木棍及其长度，以数组 $a$ 形式给出。再给定一个正整数 $k$ 。对于某个长度 $x$ ，问是否可以将那些木棍截取出至少 $k$ 个长 $x$ 的小木棍。木棍不能拼接，只能剪断。

思路是二分。最短的长度是 $1$ ，最长的长度是 $\max a$ （否则切不出长度了），对于长度 $x$ ，切出的最多数量为 $\sum_i \lfloor\frac{a[i]}{x}\rfloor$ ，其关于 $x$ 是单调的，从而可以二分。代码如下：

class Solution {
 public:
  int maxLength(vector<int>& rs, int k) {
    int l = 1, r = 0;
    for (int x : rs) r = max(r, x);
    auto f = [&](int x) {
      int cnt = 0;
      for (int r : rs) cnt += r / x;
      return cnt >= k;
    };
    while (l < r) {
      int mid = l + (r - l + 1 >> 1);
      if (f(mid)) l = mid;
      else r = mid - 1;
    }

    return f(l) ? l : 0;
  }
};