题目描述
给你一个整数数组 cards
,其中 cards[i]
表示第 i
张卡牌的 值 。如果两张卡牌的值相同,则认为这一对卡牌 匹配 。
返回你必须拿起的最小连续卡牌数,以使在拿起的卡牌中有一对匹配的卡牌。如果无法得到一对匹配的卡牌,返回 -1
。
示例 1:
输入:cards = [3,4,2,3,4,7]
输出:4
解释:拿起卡牌 [3,4,2,3] 将会包含一对值为 3 的匹配卡牌。注意,拿起 [4,2,3,4] 也是最优方案。
示例 2:
输入:cards = [1,0,5,3]
输出:-1
解释:无法找出含一对匹配卡牌的一组连续卡牌。
提示:
1 <= cards.length <= 105
0 <= cards[i] <= 106
解法
方法一:哈希表
我们初始化答案为 $+\infty$,遍历数组,对于每个数字 $x$,如果 $last[x]$ 存在,则表示 $x$ 有一对匹配卡牌,此时更新答案为 $ans = min(ans, i - last[x] + 1)$,最后如果答案为 $+\infty$,则返回 $-1$,否则返回答案。
时间复杂度 $O(n)$,空间复杂度 $O(n)$。其中 $n$ 为数组长度。
| class Solution:
def minimumCardPickup(self, cards: List[int]) -> int:
last = {}
ans = inf
for i, x in enumerate(cards):
if x in last:
ans = min(ans, i - last[x] + 1)
last[x] = i
return -1 if ans == inf else ans
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14 | class Solution {
public int minimumCardPickup(int[] cards) {
Map<Integer, Integer> last = new HashMap<>();
int n = cards.length;
int ans = n + 1;
for (int i = 0; i < n; ++i) {
if (last.containsKey(cards[i])) {
ans = Math.min(ans, i - last.get(cards[i]) + 1);
}
last.put(cards[i], i);
}
return ans > n ? -1 : ans;
}
}
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 | class Solution {
public:
int minimumCardPickup(vector<int>& cards) {
unordered_map<int, int> last;
int n = cards.size();
int ans = n + 1;
for (int i = 0; i < n; ++i) {
if (last.count(cards[i])) {
ans = min(ans, i - last[cards[i]] + 1);
}
last[cards[i]] = i;
}
return ans > n ? -1 : ans;
}
};
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 | func minimumCardPickup(cards []int) int {
last := map[int]int{}
n := len(cards)
ans := n + 1
for i, x := range cards {
if j, ok := last[x]; ok && ans > i-j+1 {
ans = i - j + 1
}
last[x] = i
}
if ans > n {
return -1
}
return ans
}
|
1
2
3
4
5
6
7
8
9
10
11
12 | function minimumCardPickup(cards: number[]): number {
const n = cards.length;
const last = new Map<number, number>();
let ans = n + 1;
for (let i = 0; i < n; ++i) {
if (last.has(cards[i])) {
ans = Math.min(ans, i - last.get(cards[i]) + 1);
}
last.set(cards[i], i);
}
return ans > n ? -1 : ans;
}
|