Làm cách nào để tìm chỉ mục tối đa trên tập con của các phần tử trong julialang?

Question

Tôi có một mảng A=[3, 5, 1, 2, 7, 9, 10, 2, 3] có độ dài length(A)=9 và một tập hợp S có chứa một tập con của các chỉ mục 1:9, ví dụ: S=Set([1, 3, 6, 8]). Tôi muốn tìm giá trị tối đa (giá trị và chỉ số) của A trên S. Tức là, tối đa là 9 và chỉ mục là 6.

tôi đã cố gắng để làm điều đó theo cách này

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); 

nhưng tôi cảm thấy như có một cách tốt hơn và thông minh.

Answer 1

Làm thế nào?

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

Điều này dựa vào thực tế là các bộ dữ liệu được sắp xếp theo thứ tự từ điển. Điều này chỉ lặp lại trên tập hợp con, vì vậy nó sẽ nhanh hơn khi S có ít phần tử đáng kể hơn A.

maximum là lựa chọn tốt hơn so với findmax tại đây như Dan Getz đã chỉ ra.

Answer 2

Nếu A sẽ không chứa các bản sao (hoặc ít nhất, tối đa là duy nhất), tôi muốn tìm ra sau dễ đọc hơn:

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

Các maximum công trình trên một máy phát điện, vì vậy không nên có bộ nhớ trên không. Nếu S nhỏ, kích thước của A chiếm ưu thế và bạn vẫn phải duyệt qua điều đó.

Answer 3

Đây là tiết hơn, nhưng trên máy tính của tôi khoảng 2.5x nhanh như các giải pháp máy phát điện (phiên bản 0.6):

function mymax(A, S) 
    val, ind = typemin(eltype(A)), -1 
    for i in S 
     val_ = A[i] 
     if val_ > val 
      val, ind = val_, i 
     end 
    end 
    return val, ind 
end 

Answer 4

Trong trường hợp có nhiều hơn mức tối đa bạn có thể làm:

function maximums(A,S) 
    maxvalue = maximum(A[i] for i in S) 
    (maxvalue, [i for i in S if A[i]==maxvalue]) 
end 

A = [3,3,3] 
S = Set([1,3]) 
maximums(A, S) # (3, [3, 1]) 

Nếu bạn muốn giữ gìn trật tự và Set là không cần thiết cho S thì bạn có thể sử dụng unique chức năng:

S = unique([1,3]) 
maximums(A, S) # (3, [1, 3])