大小为K的滑动窗口在算法题中,通常用于解决查找满足一定要求的两个元素,且两个元素的位置差异不超过K。
常见的问题如Letecode上的问题Contains Duplicate III。
问题描述:Given an array of integers, find out whether there are two distinct indices i and j in the array such that the difference between nums[i] and nums[j] is at most t and the difference between i and j is at most k.
解题思路:维护一个大小为 k 的二叉搜索树,来一个新的元素时,在BST上二分搜索有没有符合条件的数对,动态更新这个BST。因为BST的大小为 k 或不超过 k,所以这里面的数下标的差值一定是符合条件的。还有几点要注意的就是nums[i]与nums[j]的差值的是绝对值,数据比较坑爹,为了防止溢出,数据用 long类型的。
可以看出,此处的K大小的BST即为一种滑动窗口,在Java中,实现了SortedSet接口的TreeSet即可满足题目要求。通过返回满足题目要求的子set,若子set不为空,则找到。
栈和队列通常用于解决要求空间复杂度为o(k)的问题。此类问题常见于二叉树相关的题目,如Binary Search Tree Iterator。
问题如下:
Implement an iterator over a binary search tree (BST). Your iterator will be initialized with the root node of a BST.
Calling next()
will return the next smallest number in the BST.
Note: next()
and hasNext()
should run in average O(1) time and uses O(h) memory, where h is the height of the tree.
此题可用中序遍历解决,但不符合Note要求。根据二叉搜索树的特点,一个节点的下一个节点为其父节点或其右子树的最左侧节点。所以通过一个栈存储从根节点到最左侧节点的路径,栈最上面的节点是最小节点,每次调用next()方法,都是取栈最上面的结点,然后把该节点的右子树(若有的话)的根节点到最左侧节点的路径放入栈中。