题目:
给定两个数组 nums1
和 nums2
,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2] 输出:[2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出:[9,4] 解释:[4,9] 也是可通过的
提示:
1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 1000
代码:
#滑动窗口
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
vector<int> ans;
int len_s = s.size();
int len_p = p.size();
if(len_s < len_p) return ans;
// int s_count[26];
// int p_count[26];
vector<int> p_count(26);
vector<int> s_count(26);
// 把p的长度作为滑动窗口的大小,第一次以s的首个字母为起始点开始
for (int i = 0; i < len_p; i++){
s_count[s[i] - 'a']++;
p_count[p[i] - 'a']++;
}
// 如果两个容器相等,记录下0,为第一个满足条件的下标
if (s_count == p_count) ans.push_back(0);
// 接着向后滑动窗口,窗口向后移动的位置最多为s和p的长度差
// 窗口最长可以移动len_s-len_p
for (int i = 0; i < len_s-len_p; i++){
s_count[s[i] - 'a']--;
s_count[s[i+len_p] - 'a']++;
if(s_count == p_count) ans.push_back(i+1);
}
return ans;
}
};
关键思路:
- 有一个需要注意的点是,vector容器可以比较是否相等。不能定义为数组然后直接去比较数组是否相等。
- 注意最后的i+1,i初始为0,窗口的起始位置在1处,所以是+1的关系。