2526. 找到数据流中的连续整数
给你一个整数数据流，请你实现一个数据结构，检查数据流中最后 k 个整数是否 等于 给定值 value 。
请你实现 DataStream 类：
DataStream(int value, int k) 用两个整数 value 和 k 初始化一个空的整数数据流。boolean consec(int num) 将 num 添加到整数数据流。如果后 k 个整数都等于 value ，返回 true ，否则返回 false 。如果少于 k 个整数，条件不满足，所以也返回 false 。
示例 1：
输入：
["DataStream", "consec", "consec", "consec", "consec"]
[[4, 3], [4], [4], [4], [3]]
输出：
[null, false, false, true, false]
解释：
DataStream dataStream = new DataStream(4, 3); // value = 4, k = 3
dataStream.consec(4); // 数据流中只有 1 个整数，所以返回 False 。
dataStream.consec(4); // 数据流中只有 2 个整数
// 由于 2 小于 k ，返回 False 。
dataStream.consec(4); // 数据流最后 3 个整数都等于 value， 所以返回 True 。
dataStream.consec(3); // 最后 k 个整数分别是 [4,4,3] 。
// 由于 3 不等于 value ，返回 False 。
提示：
1 <= value, num <= 1091 <= k <= 105consec 次数为 105 次。原站题解
golang 解法, 执行用时: 292 ms, 内存消耗: 37.2 MB, 提交时间: 2023-01-09 10:46:59
type DataStream struct{ value, k, cnt int }
func Constructor(value, k int) DataStream {
return DataStream{value, k, 0}
}
func (d *DataStream) Consec(num int) bool {
if num == d.value {
d.cnt++
} else {
d.cnt = 0
}
return d.cnt >= d.k
}
/**
* Your DataStream object will be instantiated and called as such:
* obj := Constructor(value, k);
* param_1 := obj.Consec(num);
*/
python3 解法, 执行用时: 424 ms, 内存消耗: 45 MB, 提交时间: 2023-01-09 10:46:43
class DataStream:
def __init__(self, value: int, k: int):
self.value = value
self.k = k
self.cnt = 0
def consec(self, num: int) -> bool:
self.cnt = 0 if num != self.value else self.cnt + 1
return self.cnt >= self.k
# Your DataStream object will be instantiated and called as such:
# obj = DataStream(value, k)
# param_1 = obj.consec(num)