一、环境说明
- 本文是 LeetCode 904题 : 水果成篮,使用c语言实现
- 滑动窗口+哈希集合。
- 测试环境:Visual Studio 2019
二、代码展示
int totalFruit(int* fruits, int fruitsSize) {
int hash[100001] = { 0 };
if (1 == fruitsSize) {
return 1;
}
else if (2 == fruitsSize) {
return 2;
}
int left = 0, right = 1;
int kind[2] = { 0 };
kind[0] = fruits[left];
hash[kind[0]]++;
kind[1] = fruits[right];
hash[kind[1]]++;
int ans = 0;
while (right < fruitsSize - 1) {
hash[fruits[++right]]++;
if (fruits[right] != kind[0] && fruits[right] != kind[1]) {
if (kind[0] == kind[1]) {
kind[1] = fruits[right];
}
else {
while (1) {
hash[fruits[left++]]--;
if (0 == hash[fruits[left - 1]]) {
if (fruits[left - 1] == kind[1]) {
kind[1] = kind[0];
kind[0] = fruits[right];
}
else {
kind[0] = kind[1];
kind[1] = fruits[right];
}
break;
}
}
}
}
if (kind[0] == kind[1]) {
ans = ans < hash[kind[0]] ? hash[kind[0]] : ans;
}
else {
ans = ans < hash[kind[0]] + hash[kind[1]] ? hash[kind[0]] + hash[kind[1]] : ans;
}
}
return ans;
}
三、思路分析
- 滑动窗口+hash思想,思路清晰。
- 本题水果种类用非负数表示,hash数组更简单。
- kind[2]表示当前的水果种类,可能有1~2种。
- 遍历fruits,每次循环右窗口右移,新水果hash值++。当右窗口遇到第3种水果,左窗口循环右移,左水果hash值- -,直到窗口内只剩两种水果。
- 思路就这么简单。
四、代码分析
- 理解思路很重要!
- 代码实现只能说,顺着思路,添加细节,看注释吧。
- 博主欢迎读者在评论区留言,作为日更博主,看到就会回复的。
五、AC
![AC](https://i-blog.csdnimg.cn/blog_migrate/b6758238e2bb4ed7302542b8347fa6d6.png)
六、复杂度分析
- 时间复杂度:O(n) ,n是fruits数组的大小。最坏时间复杂度是O(2n),对应左右窗口都移动到了fruits最右边的情况。
- 空间复杂度:O(|C|),|C|是kind的大小,|C|=2。