开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第22天,点击查看活动详情
前言
从零开始学习c++,每天起码做一道leetcode题目,在此记录,希望最后能够有所收获!
一、题目描述
给你一个长度为 n 的二维整数数组 groups ,同时给你一个整数数组 nums 。
你是否可以从 nums 中选出 n 个 不相交 的子数组,使得第 i 个子数组与 groups[i] (下标从 0 开始)完全相同,且如果 i > 0 ,那么第 (i-1) 个子数组在 nums 中出现的位置在第 i 个子数组前面。(也就是说,这些子数组在 nums 中出现的顺序需要与 groups 顺序相同)
如果你可以找出这样的 n 个子数组,请你返回 true ,否则返回 false 。
如果不存在下标为 k 的元素 nums[k] 属于不止一个子数组,就称这些子数组是 不相交 的。子数组指的是原数组中连续元素组成的一个序列。
示例 1:
输入:groups = [[1,-1,-1],[3,-2,0]], nums = [1,-1,0,1,-1,-1,3,-2,0]
输出:true
解释:你可以分别在 nums 中选出第 0 个子数组 [1,-1,0,1,-1,-1,3,-2,0] 和第 1 个子数组 [1,-1,0,1,-1,-1,3,-2,0] 。
这两个子数组是不相交的,因为它们没有任何共同的元素。
示例 2:
输入:groups = [[10,-2],[1,2,3,4]], nums = [1,2,3,4,10,-2]
输出:false
解释:选择子数组 [1,2,3,4,10,-2] 和 [1,2,3,4,10,-2] 是不正确的,因为它们出现的顺序与 groups 中顺序不同。
[10,-2] 必须出现在 [1,2,3,4] 之前。
提示:
- groups.length == n
- 1 <= n <= 103
- 1 <= groups[i].length, sum(groups[i].length) <= 103
- 1 <= nums.length <= 103
- -107 <= groups[i][j], nums[k] <= 107
二、思路分析
采用双指针,用p1来记录nums中进行到了哪个元素,用p2来记录group中某个元素中进行到了哪个元素。
如果检测到p1和p2不相等的话,那就说明此时在group中遍历的这个元素在以最开始的p1的值为序号的元素是不相等的,所以需退回到之前的元素的下一个元素,继续进行。
如果检测到p1的值超出了nums的长度且p2不等于此时遍历的group的元素的长度,则返回false。
三、AC代码
class Solution {
public:
bool canChoose(vector<vector<int>>& groups, vector<int>& nums) {
int p1 = 0, p2, n = nums.size();
for (vector<int>& g : groups) {
p2 = 0;
while (p2 < g.size() && p1 < n) {
if (nums[p1] == g[p2]) p1++,p2++;
else {p1++,p1=p1-p2; p2 = 0;}
}
if (p1 >= n && p2 != g.size()) return false;
}
return true;
}
};
提交排名
四、总结
还需多加训练,这个双指针的用法我感觉就很难想到了。