一、环境说明
- 本文是 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

六、复杂度分析
- 时间复杂度:O(n) ,n是fruits数组的大小。最坏时间复杂度是O(2n),对应左右窗口都移动到了fruits最右边的情况。
- 空间复杂度:O(|C|),|C|是kind的大小,|C|=2。