public int countElements(int[] arr) {
        int count = 0;
        Set<Integer> mySet = new HashSet<>();

        for (int i = 0; i < arr.length; i++) {
            mySet.add(arr[i]);

            if (mySet.contains(arr[i] + 1) || mySet.contains(arr[i] - 1)) {
                count++;
            }
        }

        return count;
    }