From dea452c71ac46de1f7394c8ef4f8907319a0ba8e Mon Sep 17 00:00:00 2001 From: Garvit244 Date: Sun, 17 Mar 2019 18:56:49 +0800 Subject: [PATCH] Solution of divide array into group with min sum group questions --- 1000-1100q/1014.py | 94 ++++++++++++++++++++++++++++++++++++++++++++++ 400-500Q/410.py | 50 ++++++++++++++++++++++++ README.md | 6 +++ 3 files changed, 150 insertions(+) create mode 100644 1000-1100q/1014.py create mode 100644 400-500Q/410.py diff --git a/1000-1100q/1014.py b/1000-1100q/1014.py new file mode 100644 index 0000000..5ee0a4d --- /dev/null +++ b/1000-1100q/1014.py @@ -0,0 +1,94 @@ +''' +A conveyor belt has packages that must be shipped from one port to another within D days. + +The i-th package on the conveyor belt has a weight of weights[i]. Each day, we load the ship with packages on the conveyor belt (in the order given by weights). We may not load more weight than the maximum weight capacity of the ship. + +Return the least weight capacity of the ship that will result in all the packages on the conveyor belt being shipped within D days. + + + +Example 1: + +Input: weights = [1,2,3,4,5,6,7,8,9,10], D = 5 +Output: 15 +Explanation: +A ship capacity of 15 is the minimum to ship all the packages in 5 days like this: +1st day: 1, 2, 3, 4, 5 +2nd day: 6, 7 +3rd day: 8 +4th day: 9 +5th day: 10 + +Note that the cargo must be shipped in the order given, so using a ship of capacity 14 and splitting the packages into parts like (2, 3, 4, 5), (1, 6, 7), (8), (9), (10) is not allowed. +Example 2: + +Input: weights = [3,2,2,4,1,4], D = 3 +Output: 6 +Explanation: +A ship capacity of 6 is the minimum to ship all the packages in 3 days like this: +1st day: 3, 2 +2nd day: 2, 4 +3rd day: 1, 4 + +Note: + +1 <= D <= weights.length <= 50000 +1 <= weights[i] <= 500 + +''' + + +class Solution(object): + def shipWithinDays(self, weights, D): + """ + :type weights: List[int] + :type D: int + :rtype: int + """ + high, low = sum(weights)+1, max(weights) + + while(low < high): + mid = (high+low)/2 + temp_left = mid + packet_at_left = D-1 + for weight in weights: + if weight <= mid: + if temp_left < weight: + if packet_at_left == 0: + low = mid+1 + break + packet_at_left -= 1 + temp_left = mid-weight + else: + temp_left -= weight + else: + high = mid + + return low + + +class Solution(object): + def shipWithinDays(self, weights, D): + """ + :type weights: List[int] + :type D: int + :rtype: int + """ + left, right = max(weights), sum(weights) + + while left < right: + curr_sum, groups, invalid = 0, 0, True + mid = left + ((right-left) >> 1) + for weight in weights: + if weight > mid: + invalid = False + break + if curr_sum + weight > mid: + groups += 1 + curr_sum = 0 + curr_sum += weight + if invalid and groups < D: + right = mid + else: + left = mid + 1 + return left \ No newline at end of file diff --git a/400-500Q/410.py b/400-500Q/410.py new file mode 100644 index 0000000..77fb28e --- /dev/null +++ b/400-500Q/410.py @@ -0,0 +1,50 @@ +''' +Given an array which consists of non-negative integers and an integer m, you can split the array into m non-empty continuous subarrays. Write an algorithm to minimize the largest sum among these m subarrays. + +Note: +If n is the length of array, assume the following constraints are satisfied: + +1 ≤ n ≤ 1000 +1 ≤ m ≤ min(50, n) +Examples: + +Input: +nums = [7,2,5,10,8] +m = 2 + +Output: +18 + +Explanation: +There are four ways to split nums into two subarrays. +The best way is to split it into [7,2,5] and [10,8], +where the largest sum among the two subarrays is only 18. +''' + +class Solution(object): + def splitArray(self, nums, m): + """ + :type nums: List[int] + :type m: int + :rtype: int + """ + + left, right = max(nums), sum(nums) + + while left < right: + mid = left + ((right-left) >> 1) + curr_sum, invalid, groups = 0, True, 0 + for num in nums: + if num > mid: + inalid = False + break + if num + curr_sum > mid: + groups += 1 + curr_sum = 0 + curr_sum += num + if invalid and groups < m: + right = mid + else: + left = mid + 1 + return left + \ No newline at end of file diff --git a/README.md b/README.md index a79d9e0..b08df14 100644 --- a/README.md +++ b/README.md @@ -8,6 +8,7 @@ Python solution of problems from [LeetCode](https://leetcode.com/). | # | Title | Solution | Difficulty | |---| ----- | -------- | ---------- | |1002|[Find Common Characters](https://leetcode.com/problems/find-common-characters)|[Python](./1000-1100q/1002.py)|Easy| +|1014|[Capacity To Ship Packages Within D Days](https://leetcode.com/problems/capacity-to-ship-packages-within-d-days/) | [Python](./1000-1100q/1014.py) | Medium| ##### [Problems 900-1000](./900-1000q/) | # | Title | Solution | Difficulty | @@ -29,6 +30,11 @@ Python solution of problems from [LeetCode](https://leetcode.com/). |977|[Squares of a Sorted Array](https://leetcode.com/problems/squares-of-a-sorted-array/) | [Python](./900-1000q/977.py)|Easy| +##### [Problems 400-500](./400-500q/) +| # | Title | Solution | Difficulty | +|---| ----- | -------- | ---------- | +|410|[Split Array Largest Sum](https://leetcode.com/problems/split-array-largest-sum/) | [Python](./400-500q/410.py)|Hard| + ##### [Problems 300-400](./300-400q/) | # | Title | Solution | Difficulty | |---| ----- | -------- | ---------- |