julia - 如何找到julialang中元素子集的最大值的索引？

Question

我有一个长度为A=[3, 5, 1, 2, 7, 9, 10, 2, 3]的length(A)=9数组和一个包含索引子集S的子集1:9的集合，例如S=Set([1, 3, 6, 8])。我想找到A而不是S的最大值(值和索引)。也就是说，最大值为9，索引为6。

我试图这样做

A = [3, 5, 1, 2, 7, 9, 10, 2, 3];
S = Set([1, 3, 6, 8]);
B = [if i in S A[i] else 0 end for i in 1:length(A)];
findmax(B);

但是我觉得有更好更好的方法。

Answer 1

怎么样？

julia> maximum((A[i],i) for i in S)
(9, 6)

这依赖于元组按字典顺序排序的事实。这只会对子集进行迭代，因此当 S的元素数明显少于 A时，它应该更快。

正如Dan Getz所指出的， maximum比 findmax更好。