咨询MINLOC能否返回多最小值索引数组及索引最优循环方法

嘿，这个问题问到点子上了！在Fortran里处理数组中多个极值的索引确实是日常编码里常碰到的场景，我来给你一步步讲清楚～

一、MINLOC/MAXLOC能不能返回所有极值的索引数组？ #

先直接给结论：默认情况下，MINLOC和MAXLOC只会返回数组中第一个遇到的最小值/最大值的索引（如果是多维数组，会返回一个维度对应大小的数组，代表第一个极值的位置）。但如果我们想拿到所有满足条件的极值索引，就得稍微绕个弯——先找到极值本身，再筛选出所有等于这个值的位置。

举个一维数组的例子，假设我们有arr = [3, 1, 4, 1, 5]，要拿到所有最小值（也就是1）的索引：

program test_min_indices
    implicit none
    integer :: arr(5) = [3, 1, 4, 1, 5]
    integer :: min_val, all_min_indices(:), i

    ! 第一步：先拿到最小值
    min_val = MINVAL(arr)
    ! 第二步：用PACK函数收集所有等于最小值的索引
    all_min_indices = PACK([(i, i=1,size(arr))], arr == min_val)

    ! 输出结果：会得到 [2,4]
    print *, "所有最小值的索引：", all_min_indices
end program test_min_indices

如果是二维数组，我们可以用RESHAPE配合PACK来获取二维坐标：

program test_2d_min_indices
    implicit none
    integer :: arr(2,3) = reshape([5,2,2,7,2,9], [2,3])
    integer :: min_val, flat_indices(:), coords(2, size(flat_indices))
    integer :: i

    min_val = MINVAL(arr)
    ! 先拿到扁平化后的索引
    flat_indices = PACK([(i, i=1,size(arr))], arr == min_val)
    ! 把扁平化索引转换成二维坐标
    coords = reshape([(MODULO(flat_indices(i)-1, 2)+1, (flat_indices(i)-1)/2+1, i=1,size(flat_indices))], &
                    [2, size(flat_indices)])

    ! 输出结果：第一行是行号，第二列是列号，会得到 [[1,2,1],[2,2,3]]
    print *, "所有最小值的二维坐标："
    do i=1, size(coords, 2)
        print *, coords(:,i)
    end do
end program test_2d_min_indices

二、拿到索引数组后怎么最优遍历？ #

遍历的最优方式取决于你的索引存储形式和实际需求，这里给几个常用的高效方案：

1. 一维索引数组：直接DO循环（最直观高效） #

如果是用PACK得到的一维索引数组，直接用DO循环遍历就行，这种方式内存开销小，缓存命中率也高：

do i = 1, size(all_min_indices)
    ! 处理第all_min_indices(i)个元素
    print *, "处理元素：", arr(all_min_indices(i))
end do

2. 多维坐标数组：按列遍历+并行处理 #

因为Fortran是列主序语言，按列遍历多维坐标数组能获得更好的缓存性能：

do i = 1, size(coords, 2)
    j = coords(1, i)  ! 行号
    k = coords(2, i)  ! 列号
    print *, "处理元素(",j,",",k,")：", arr(j,k)
end do

如果你的编译器支持Fortran 2008及以上，还可以用DO CONCURRENT来并行处理这些索引，充分利用多核CPU的性能：

do concurrent(i = 1:size(coords, 2))
    j = coords(1, i)
    k = coords(2, i)
    ! 这里的操作是线程安全的，编译器会自动并行化
    arr(j,k) = arr(j,k) * 2
end do

3. 避免内存复制：用WHERE直接原地处理 #

如果你的需求是直接对所有极值元素做操作，甚至不需要单独提取索引数组——直接用WHERE语句原地处理，这是最优的方案，省去了索引数组的内存开销：

! 直接把所有最小值元素乘以2，不需要提取索引
WHERE(arr == min_val)
    arr = arr * 2
END WHERE

内容的提问来源于stack exchange，提问作者anfho