|
59 | 59 |
|
60 | 60 | <!-- 这里可写通用的实现逻辑 -->
|
61 | 61 |
|
| 62 | +**方法一:贪心枚举** |
| 63 | + |
| 64 | +我们贪心地枚举 `nums` 中每一个数 $nums[j]$ 作为子数组的开始,判断其是否与当前 $groups[i]$ 匹配,是则将指针 $i$ 往后移一位,将指针 $j$ 往后移动 $groups[i].length$ 位,否则将指针 $j$ 往后移动一位。 |
| 65 | + |
| 66 | +如果 $i$ 走到了 $groups.length$,说明所有的子数组都匹配上了,返回 `true`,否则返回 `false`。 |
| 67 | + |
| 68 | +时间复杂度 $O(n \times m),空间复杂度 O(1)$。其中 $n$ 和 $m$ 分别为 `groups` 和 `nums` 的长度。 |
| 69 | + |
62 | 70 | <!-- tabs:start -->
|
63 | 71 |
|
64 | 72 | ### **Python3**
|
65 | 73 |
|
66 | 74 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
67 | 75 |
|
68 | 76 | ```python
|
69 |
| - |
| 77 | +class Solution: |
| 78 | + def canChoose(self, groups: List[List[int]], nums: List[int]) -> bool: |
| 79 | + n, m = len(groups), len(nums) |
| 80 | + i = j = 0 |
| 81 | + while i < n and j < m: |
| 82 | + g = groups[i] |
| 83 | + if g == nums[j : j + len(g)]: |
| 84 | + j += len(g) |
| 85 | + i += 1 |
| 86 | + else: |
| 87 | + j += 1 |
| 88 | + return i == n |
70 | 89 | ```
|
71 | 90 |
|
72 | 91 | ### **Java**
|
73 | 92 |
|
74 | 93 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
75 | 94 |
|
76 | 95 | ```java
|
| 96 | +class Solution { |
| 97 | + public boolean canChoose(int[][] groups, int[] nums) { |
| 98 | + int n = groups.length, m = nums.length; |
| 99 | + int i = 0; |
| 100 | + for (int j = 0; i < n && j < m;) { |
| 101 | + if (check(groups[i], nums, j)) { |
| 102 | + j += groups[i].length; |
| 103 | + ++i; |
| 104 | + } else { |
| 105 | + ++j; |
| 106 | + } |
| 107 | + } |
| 108 | + return i == n; |
| 109 | + } |
| 110 | + |
| 111 | + private boolean check(int[] a, int[] b, int j) { |
| 112 | + int m = a.length, n = b.length; |
| 113 | + int i = 0; |
| 114 | + for (; i < m && j < n; ++i, ++j) { |
| 115 | + if (a[i] != b[j]) { |
| 116 | + return false; |
| 117 | + } |
| 118 | + } |
| 119 | + return i == m; |
| 120 | + } |
| 121 | +} |
| 122 | +``` |
| 123 | + |
| 124 | +### **C++** |
| 125 | + |
| 126 | +```cpp |
| 127 | +class Solution { |
| 128 | +public: |
| 129 | + bool canChoose(vector<vector<int>>& groups, vector<int>& nums) { |
| 130 | + auto check = [&](vector<int>& a, vector<int>& b, int j) { |
| 131 | + int m = a.size(), n = b.size(); |
| 132 | + int i = 0; |
| 133 | + for (; i < m && j < n; ++i, ++j) { |
| 134 | + if (a[i] != b[j]) { |
| 135 | + return false; |
| 136 | + } |
| 137 | + } |
| 138 | + return i == m; |
| 139 | + }; |
| 140 | + int n = groups.size(), m = nums.size(); |
| 141 | + int i = 0; |
| 142 | + for (int j = 0; i < n && j < m;) { |
| 143 | + if (check(groups[i], nums, j)) { |
| 144 | + j += groups[i].size(); |
| 145 | + ++i; |
| 146 | + } else { |
| 147 | + ++j; |
| 148 | + } |
| 149 | + } |
| 150 | + return i == n; |
| 151 | + } |
| 152 | +}; |
| 153 | +``` |
77 | 154 |
|
| 155 | +### **Go** |
| 156 | +
|
| 157 | +```go |
| 158 | +func canChoose(groups [][]int, nums []int) bool { |
| 159 | + check := func(a, b []int, j int) bool { |
| 160 | + m, n := len(a), len(b) |
| 161 | + i := 0 |
| 162 | + for ; i < m && j < n; i, j = i+1, j+1 { |
| 163 | + if a[i] != b[j] { |
| 164 | + return false |
| 165 | + } |
| 166 | + } |
| 167 | + return i == m |
| 168 | + } |
| 169 | + n, m := len(groups), len(nums) |
| 170 | + i := 0 |
| 171 | + for j := 0; i < n && j < m; { |
| 172 | + if check(groups[i], nums, j) { |
| 173 | + j += len(groups[i]) |
| 174 | + i++ |
| 175 | + } else { |
| 176 | + j++ |
| 177 | + } |
| 178 | + } |
| 179 | + return i == n |
| 180 | +} |
78 | 181 | ```
|
79 | 182 |
|
80 | 183 | ### **...**
|
|
0 commit comments