计算双端队列中唯一值的最大数量的最快方法

Question

我正在尝试找到最快的方法来计算从控制台获取的一系列 n 个整数的 m 大小子数组中唯一值的最大数量。有没有办法进一步优化这段代码？提前致谢，亚历克斯

import java.util.*;
public class test {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        Deque deque = new ArrayDeque<>();
        int n = in.nextInt();
        int m = in.nextInt();
        long count = 0, c = 0;

        for (int i = 0; i < m; i++) deque.addFirst(in.nextInt());
        count = deque.stream().distinct().count();
        for (int i = 0; i < n - m; i++) {
            count = Math.max(count, deque.stream().distinct().count());

            deque.removeLast();
            deque.addFirst(in.nextInt());
        }
        System.out.println(count);
    }
}

Answer 1

这是 O(NM)（假设 distinct 是 O(M)）。通过跟踪当前窗口内具有非零计数的值的数量，应该可以在 O(N) 中执行此操作。

伪代码：

for (int i = M; i < N; i++) {
    counts[x[i-M]]--;
    if (counts[x[i-M]] == 0) {
        nonzero--;
    }

    if (counts[x[i]] == 0) {
        nonzero++;
    }
    counts[x[i]]++;

    maxNonzero = max(maxNonzero, nonzero);
}