您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

如何获取先验未知数组作为Fortran中函数的输出

如何获取先验未知数组作为Fortran中函数的输出

我希望有一个真正的Fortran程序员,但是在没有更好建议的情况下,我只会指定形状而不是大小x(:),使用一个临时数组temp(size(x)),并输出yallocatable。然后,在第一遍之后,allocate(y(j))从临时数组中复制值。但是我不能强调自己不是Fortran程序员,所以我不能说该语言是否具有可增长的数组或是否存在用于后者的库。

program test
    implicit none
    integer:: x(10) = (/1,0,2,0,3,0,4,0,5,0/)
    print "(10I2.1)", select(x)

contains

    function select(x) result(y)
        implicit none
        integer, intent(in):: x(:) 
        integer:: i, j, temp(size(x))
        integer, allocatable:: y(:)

        j = 0
        do i = 1, size(x)
            if (x(i) /= 0) then
                j = j + 1
                temp(j) = x(i)
            endif
        enddo

        allocate(y(j))
        y = temp(:j)
    end function select

end program test

编辑:

基于MSB的答案,这里是一种生长功能的修订版 温度 y与超额分配。 与之前一样,它将结果复制到y的末尾。 事实证明,我没有必要以最终大小显式分配新数组。相反,它可以通过分配自动完成。

    function select(x) result(y)
        implicit none
        integer, intent(in):: x(:) 
        integer:: i, j, dsize
        integer, allocatable:: temp(:), y(:)

        dsize = 0; allocate(y(0))

        j = 0
        do i = 1, size(x)
            if (x(i) /= 0) then
                j = j + 1

                if (j >= dsize) then         !grow y using temp
                    dsize = j + j / 8 + 8 
                    allocate(temp(dsize))
                    temp(:size(y)) = y
                    call move_alloc(temp, y) !temp gets deallocated
                endif

                y(j) = x(i)
            endif
        enddo
        y = y(:j)
    end function select
其他 2022/1/1 18:52:51 有442人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶