如何从另一个数组返回具有某个int的索引的数组

Question

我是一名 Java 初学者，我一直在尝试编写一种方法来列出数组中某个 int 的所有索引。到目前为止，我所做的是将该 int 的值存储在另一个数组中的相应索引处，但我能做的最好的事情就是将所有其他不等于原始 int 的索引的值设置为 -1。

我想我需要在数组中存储值I，并删除所有的-1，但是我不知道怎么做。顺便说一下，这些值是-1，因为这个程序中所有的数组都包含0-100之间的整数。如果数组中的整数可以是任意数，我该怎么办？

当然，也有一种更容易或更有效的方法来做到这一点。

public static int[] maxValueIndex(int[] arr, int targetValue, int x) {
    
    int[] maxValue = new int[x];
    
    
    for (int i = 0; i < arr.length; i++) {
        if (arr[i] == targetValue) {
            maxValue[i] = arr[i];
        } else {
            maxValue[i] = -1;
        }
        
    }       
    return maxValue;
    }

Answer 1

如果只需要使用数组来解决此任务，则可能需要两次才能创建仅包含有效索引的压缩数组：

1. 查找匹配项数，然后创建并填充紧凑数组

public static int[] getTargetIndexes(int targetValue, int ... arr) {
    int n = arr.length;
    int targetCount = 0;
    for (int i = 0; i < n; i++) {
        if (arr[i] == targetValue) {
            targetCount++;
        }
    } 
    
    int[] indexes = new int[targetCount];

    for (int i = 0, j = 0; j <targetCount && i < n; i++) {
        if (arr[i] == targetValue) {
            indexes[j++] = i;
        }
    }
    
    return indexes;
}

public static int[] getTargetIndexes(int targetValue, int ... arr) {
    int n = arr.length;
    int[] indexes = new int[n];

    int targetCount = 0;
    for (int i = 0; i < n; i++) {
        if (arr[i] == targetValue) {
            indexes[i] = i;
            targetCount++;
        } else {
            indexes[i] = -1;
        }
    }
    
    for (int i = 0, j = 0; j < targetCount && i < n; i++) {
        if (indexes[i] > -1) {
            indexes[j++] = i;
        }
    }
    
    return Arrays.copyOf(indexes, targetCount); // truncate bad indexes
}

此外，该方法的签名使用vararg将输入数组作为int的序列传递——那么vararg参数int…arr应该是最后一个。

如果可以使用Stream API，则可以通过声明性方式方便地解决任务：

public static int[] getTargetIndexes(int targetValue, int ... arr) {   
    return IntStream.range(0, arr.length) // get stream of indexes
            .filter(i -> arr[i] == targetValue) // keep only matching indexes
            .toArray(); // build output array
}

Answer 2

如果我正确理解了您的查询，那么您需要一个包含所有索引 i 的数组，以便 arr[i]==targetValue。我们可以使用任何动态数据结构有效地实现这一目标。例如，使用 ArrayList 并继续逐个添加所有所需的索引，然后将 List 转换为数组并返回它。
像这样：

List<Integer> index = new ArrayList<Integer>();
for (int i = 0; i < arr.length; i++)
{
    if (arr[i] == targetValue)
        index.add( i );
}
int[] maxValue = index.stream().mapToInt(Integer::intValue).toArray();
return maxValue;