Java总哈希表的运用，判断一个数组中是否存在相同的元素之间的距离在k以内！

最新推荐文章于 2025-01-25 23:16:48 发布

原创最新推荐文章于 2025-01-25 23:16:48 发布 · 1.3k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#java #哈希表

Java 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用哈希表的方法来检查给定数组中是否存在距离不超过k的重复元素，通过Java和C++两种实现方式展示了如何在O(n)的时间复杂度内解决问题。

Check if a given array contains duplicate elements within k distance from each other

用哈希表，复杂度O（n），直接嵌套循环复杂度O（n*k）。

import java.io.BufferedInputStream;
import java.util.HashSet;
import java.util.Scanner;

public class Main {
    static boolean checkDuplicateWithinK(int arr[], int k) {
        HashSet<Integer> set = new HashSet<>();
        for(int i = 0; i < arr.length; ++i) {
            if(set.contains(arr[i])) {
                return true;
            }
            set.add(arr[i]);
            if(i >= k) {
                set.remove(arr[i-k]);
            }
        }
        return false;
    }
    public static void main(String []args) {
        int T, n, k;
        Scanner cin = new Scanner(new BufferedInputStream(System.in));
        while (cin.hasNext()) {
            T = cin.nextInt();
            while (T-- != 0) {
                k = cin.nextInt();
                n = cin.nextInt();
                int arr[] = new int[n];
                for (int i = 0; i < n; ++i) {
                    arr[i] = cin.nextInt();
                }
                if (checkDuplicateWithinK(arr, k) == true) {
                    System.out.println("True");
                } else {
                    System.out.println("False");
                }
            }
        }
    }
}

当然还可以用C++的STL来解决：

#include <bits/stdc++.h>

using namespace std;
const int MAXN = 284;
int arr[MAXN];

int main() {
    ios::sync_with_stdio(false);
    int T, n, k;
    cin >> T;
    while(T--) {
        cin >> k >> n;
        multiset<int> s;
        bool ans = false;
        for(int i = 0; i < n; ++i) {
            cin >> arr[i];
            if(s.find(arr[i]) != s.end()) {
                ans = true;   
            }
            s.insert(arr[i]);
            if(i >= k) {
                s.erase(s.find(arr[i-k]));
            }
        }
        cout << (ans ? "True" : "False") << endl;
    }
    return 0;
}