|
6 | 6 |
|
7 | 7 | 远征队即将开启未知的冒险之旅,不过在此之前,将对补给车队进行最后的检查。`supplies[i]` 表示编号为 `i` 的补给马车装载的物资数量。
|
8 | 8 | 考虑到车队过长容易被野兽偷袭,他们决定将车队的长度变为原来的一半(向下取整),计划为:
|
9 |
| -- 找出车队中 **物资之和最小** 两辆 **相邻** 马车,将它们车辆的物资整合为一辆。若存在多组物资之和相同的马车,则取编号最小的两辆马车进行整合; |
10 |
| -- 重复上述操作直到车队长度符合要求。 |
| 9 | + |
| 10 | +- 找出车队中 **物资之和最小** 两辆 **相邻** 马车,将它们车辆的物资整合为一辆。若存在多组物资之和相同的马车,则取编号最小的两辆马车进行整合; |
| 11 | +- 重复上述操作直到车队长度符合要求。 |
11 | 12 |
|
12 | 13 | 请返回车队长度符合要求后,物资的分布情况。
|
13 | 14 |
|
14 | 15 | **示例 1:**
|
15 |
| ->输入:`supplies = [7,3,6,1,8]` |
| 16 | + |
| 17 | +> 输入:`supplies = [7,3,6,1,8]` |
16 | 18 | >
|
17 |
| ->输出:`[10,15]` |
| 19 | +> 输出:`[10,15]` |
18 | 20 | >
|
19 |
| ->解释: |
| 21 | +> 解释: |
20 | 22 | > 第 1 次合并,符合条件的两辆马车为 6,1,合并后的车队为 [7,3,7,8];
|
21 | 23 | > 第 2 次合并,符合条件的两辆马车为 (7,3) 和 (3,7),取编号最小的 (7,3),合并后的车队为 [10,7,8];
|
22 | 24 | > 第 3 次合并,符合条件的两辆马车为 7,8,合并后的车队为 [10,15];
|
23 |
| ->返回 `[10,15]` |
| 25 | +> 返回 `[10,15]` |
24 | 26 |
|
25 | 27 | **示例 2:**
|
26 |
| ->输入:`supplies = [1,3,1,5]` |
| 28 | + |
| 29 | +> 输入:`supplies = [1,3,1,5]` |
27 | 30 | >
|
28 |
| ->输出:`[5,5]` |
| 31 | +> 输出:`[5,5]` |
29 | 32 |
|
30 | 33 | **解释:**
|
31 |
| -- `2 <= supplies.length <= 1000` |
32 |
| -- `1 <= supplies[i] <= 1000` |
| 34 | + |
| 35 | +- `2 <= supplies.length <= 1000` |
| 36 | +- `1 <= supplies[i] <= 1000` |
33 | 37 |
|
34 | 38 | ## 解法
|
35 | 39 |
|
36 | 40 | <!-- 这里可写通用的实现逻辑 -->
|
37 | 41 |
|
| 42 | +**方法一:模拟** |
| 43 | + |
| 44 | +根据题目描述,我们每次遍历 `supplies`,找到物资之和最小的两辆相邻马车,将它们车辆的物资整合为一辆,重复上述操作直到车队长度符合要求。 |
| 45 | + |
| 46 | +时间复杂度 $O(n^2)$,空间复杂度 $O(n)$。其中 $n$ 为 `supplies` 的长度。 |
| 47 | + |
38 | 48 | <!-- tabs:start -->
|
39 | 49 |
|
40 | 50 | ### **Python3**
|
41 | 51 |
|
42 | 52 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
43 | 53 |
|
44 | 54 | ```python
|
45 |
| - |
| 55 | +class Solution: |
| 56 | + def supplyWagon(self, supplies: List[int]) -> List[int]: |
| 57 | + for _ in range((len(supplies) + 1) >> 1): |
| 58 | + n = len(supplies) |
| 59 | + mi = inf |
| 60 | + k = 0 |
| 61 | + for i in range(n - 1): |
| 62 | + x = supplies[i] + supplies[i + 1] |
| 63 | + if mi > x: |
| 64 | + mi = x |
| 65 | + k = i |
| 66 | + t = [] |
| 67 | + i = 0 |
| 68 | + while i < n: |
| 69 | + if i == k: |
| 70 | + t.append(mi) |
| 71 | + i += 2 |
| 72 | + else: |
| 73 | + t.append(supplies[i]) |
| 74 | + i += 1 |
| 75 | + supplies = t |
| 76 | + return supplies |
46 | 77 | ```
|
47 | 78 |
|
48 | 79 | ### **Java**
|
49 | 80 |
|
50 | 81 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
51 | 82 |
|
52 | 83 | ```java
|
| 84 | +class Solution { |
| 85 | + public int[] supplyWagon(int[] supplies) { |
| 86 | + for (int h = (supplies.length + 1) >> 1; h > 0; --h) { |
| 87 | + int n = supplies.length; |
| 88 | + int mi = 1 << 30; |
| 89 | + int k = 0; |
| 90 | + for (int i = 0; i < n - 1; ++i) { |
| 91 | + int x = supplies[i] + supplies[i + 1]; |
| 92 | + if (mi > x) { |
| 93 | + mi = x; |
| 94 | + k = i; |
| 95 | + } |
| 96 | + } |
| 97 | + int[] t = new int[n - 1]; |
| 98 | + for (int i = 0, j = 0; i < n; ++i, ++j) { |
| 99 | + if (i == k) { |
| 100 | + t[j] = mi; |
| 101 | + ++i; |
| 102 | + } else { |
| 103 | + t[j] = supplies[i]; |
| 104 | + } |
| 105 | + } |
| 106 | + supplies = t; |
| 107 | + } |
| 108 | + return supplies; |
| 109 | + } |
| 110 | +} |
| 111 | +``` |
| 112 | + |
| 113 | +### **C++** |
| 114 | + |
| 115 | +```cpp |
| 116 | +class Solution { |
| 117 | +public: |
| 118 | + vector<int> supplyWagon(vector<int>& supplies) { |
| 119 | + for (int h = (supplies.size() + 1) >> 1; h; --h) { |
| 120 | + int n = supplies.size(); |
| 121 | + int mi = 1 << 30; |
| 122 | + int k = 0; |
| 123 | + for (int i = 0; i < n - 1; ++i) { |
| 124 | + int x = supplies[i] + supplies[i + 1]; |
| 125 | + if (mi > x) { |
| 126 | + mi = x; |
| 127 | + k = i; |
| 128 | + } |
| 129 | + } |
| 130 | + vector<int> t(n - 1); |
| 131 | + for (int i = 0, j = 0; i < n; ++i, ++j) { |
| 132 | + if (i == k) { |
| 133 | + t[j] = mi; |
| 134 | + ++i; |
| 135 | + } else { |
| 136 | + t[j] = supplies[i]; |
| 137 | + } |
| 138 | + } |
| 139 | + supplies = move(t); |
| 140 | + } |
| 141 | + return supplies; |
| 142 | + } |
| 143 | +}; |
| 144 | +``` |
| 145 | +
|
| 146 | +### **Go** |
| 147 | +
|
| 148 | +```go |
| 149 | +func supplyWagon(supplies []int) []int { |
| 150 | + for h := (len(supplies) + 1) >> 1; h > 0; h-- { |
| 151 | + n := len(supplies) |
| 152 | + mi := 1 << 30 |
| 153 | + k := 0 |
| 154 | + for i := 0; i < n-1; i++ { |
| 155 | + x := supplies[i] + supplies[i+1] |
| 156 | + if mi > x { |
| 157 | + mi = x |
| 158 | + k = i |
| 159 | + } |
| 160 | + } |
| 161 | + t := make([]int, n-1) |
| 162 | + for i, j := 0, 0; i < n; i, j = i+1, j+1 { |
| 163 | + if i == k { |
| 164 | + t[j] = mi |
| 165 | + i++ |
| 166 | + } else { |
| 167 | + t[j] = supplies[i] |
| 168 | + } |
| 169 | + } |
| 170 | + supplies = t |
| 171 | + } |
| 172 | + return supplies |
| 173 | +} |
| 174 | +``` |
53 | 175 |
|
| 176 | +### **TypeScript** |
| 177 | + |
| 178 | +```ts |
| 179 | +function supplyWagon(supplies: number[]): number[] { |
| 180 | + for (let h = (supplies.length + 1) >> 1; h > 0; --h) { |
| 181 | + const n = supplies.length; |
| 182 | + let mi = 1 << 30; |
| 183 | + let k = 0; |
| 184 | + for (let i = 0; i < n - 1; ++i) { |
| 185 | + const x = supplies[i] + supplies[i + 1]; |
| 186 | + if (mi > x) { |
| 187 | + mi = x; |
| 188 | + k = i; |
| 189 | + } |
| 190 | + } |
| 191 | + const t: number[] = new Array(n - 1); |
| 192 | + for (let i = 0, j = 0; i < n; ++i, ++j) { |
| 193 | + if (i === k) { |
| 194 | + t[j] = mi; |
| 195 | + ++i; |
| 196 | + } else { |
| 197 | + t[j] = supplies[i]; |
| 198 | + } |
| 199 | + } |
| 200 | + supplies = t; |
| 201 | + } |
| 202 | + return supplies; |
| 203 | +} |
54 | 204 | ```
|
55 | 205 |
|
56 | 206 | ### **...**
|
|
0 commit comments