找到数据流中的连续整数

For a stream of integers, implement a data structure that checks if the last k integers parsed in the stream are equal to value.

Implement the DataStream class:

• DataStream(int value, int k) Initializes the object with an empty integer stream and the two integers value and k.
• boolean consec(int num) Adds num to the stream of integers. Returns true if the last k integers are equal to value, and false otherwise. If there are less than k integers, the condition does not hold true, so returns false.

 

Example 1:

Input
["DataStream", "consec", "consec", "consec", "consec"]
[[4, 3], [4], [4], [4], [3]]
Output
[null, false, false, true, false]

Explanation
DataStream dataStream = new DataStream(4, 3); //value = 4, k = 3 
dataStream.consec(4); // Only 1 integer is parsed, so returns False. 
dataStream.consec(4); // Only 2 integers are parsed.
                      // Since 2 is less than k, returns False. 
dataStream.consec(4); // The 3 integers parsed are all equal to value, so returns True. 
dataStream.consec(3); // The last k integers parsed in the stream are [4,4,3].
                      // Since 3 is not equal to value, it returns False.

 

Constraints:

• 1 <= value, num <= 109
• 1 <= k <= 105
• At most 105 calls will be made to consec.

/*
题目思路：
1. 创建一个 DataStream 类，该类有两个成员变量：一个用来存储整数数据流的列表，一个用来存储目标值 value 和 k 的值。
2. 构造函数初始化 value 和 k。
3. consec 方法使用 Stream API 将整数添加到数据流中，并检查最后 k 个整数是否都等于 value。
*/
import java.util.*;
import java.util.stream.*;

public class DataStream {
    private int value;
    private int k;
    private List<Integer> stream;

    public DataStream(int value, int k) {
        this.value = value;
        this.k = k;
        this.stream = new ArrayList<>();
    }

    public boolean consec(int num) {
        stream.add(num);
        if (stream.size() < k) {
            return false;
        }
        return stream.subList(stream.size() - k, stream.size()).stream().allMatch(n -> n == value);
    }
}

此题解采用的是一种计数方法来检查数据流中最后 k 个整数是否等于初始化时指定的 value。在构造函数中，初始化 value 和 k 的值，并设置一个计数器 cnt 用于记录连续等于 value 的整数数量。在 consec 方法中，每次调用时都会将传入的整数 num 与 value 进行比较。如果 num 等于 value，则增加计数器 cnt；否则，重置计数器 cnt 为 0。最后，如果 cnt 的值大于等于 k，则返回 true 表示最后 k 个整数都等于 value，否则返回 false。

O(1)

O(1)