FORTRAN 函数子程序求二维数组主对角元素和

最新推荐文章于 2023-02-05 07:45:00 发布

zhuliting

最新推荐文章于 2023-02-05 07:45:00 发布

阅读量3.3k

点赞数

分类专栏： FORTRAN 文章标签： fortran integer subroutine function fun input

本文链接：https://blog.csdn.net/zhuliting/article/details/6432066

版权

FORTRAN 专栏收录该内容

21 篇文章 2 订阅

订阅专栏

写一个函数子程序，求nxn二维数组全体主对角元素的和。

program main implicit none integer,parameter::imax = 4, jmax = 4, N = imax*jmax integer i,j real y(imax,imax),arr(N),crosssum !1-dim array arr = (/(i, i = 1, N)/) print *, 'arr:' print '(4f6.1)', ((arr(j),j=i,N,imax),i=1,imax) !2-dim array y = reshape(source = (/1,2,3,4,5,6,7,8,9,10,11, & 12,13,14,15,16/),shape = (/4,4/)) print *,'y:' print '(4f6.1)',(y(i,:),i = 1,imax) print *, 'crosssum of y =', crosssum(y,imax) print *,'corsssum of arr = ',crosssum(arr,imax) stop end function crosssum(x, n) result(s) implicit none integer n, i real x(n,n), s s = 0.0 do i = 1, n s = s + x(i,i) enddo end

result:

[root@c0109 zlt]# ./abc arr: 1.0 5.0 9.0 13.0 2.0 6.0 10.0 14.0 3.0 7.0 11.0 15.0 4.0 8.0 12.0 16.0 y: 1.0 5.0 9.0 13.0 2.0 6.0 10.0 14.0 3.0 7.0 11.0 15.0 4.0 8.0 12.0 16.0 crosssum of y = 34.00000 corsssum of arr = 34.00000

附：将一维数组以二维的形式显示：

program main integer,parameter::imax = 4, jmax = 5, N = imax*jmax integer i,j real arr(N) do i = 1, N arr(i) = i enddo do i = 1, imax do j = i, N, imax write(*, '(f6.1)', advance = 'NO') arr(j) enddo write(*,*) enddo print * print '(5f6.1)', ((arr(j),j=i,N,imax),i=1,imax) end

result:

1.0 5.0 9.0 13.0 17.0 2.0 6.0 10.0 14.0 18.0 3.0 7.0 11.0 15.0 19.0 4.0 8.0 12.0 16.0 20.0

另外，用户自身定义的函数子程序也可以作为参数进行传递，下面看一个实际的例子：

PROGRAM main implicit none integer N real, external::Funx print *, 'please input the counts N:' read(*,*) N call Calculate(N, Funx) END PROGRAM subroutine Calculate(N, Fun) implicit none integer N,I real Time1, Time2, S real Fun s = 0.0 call CPU_TIME(Time1) do I = 1, N S = Fun(N) + S enddo call CPU_TIME(Time2) write(*, '(1x, "Result = ", E10.5)') s write(*, '(1x, "Time = ", F10.2)') Time2 - Time1 end subroutine Calculate ! external function , the parameter real function Funx(N) implicit none integer N, I Funx = 0.0 do I = 1, N Funx = Funx + SQRT(2.0) enddo end function Funx

result:

[root@c0108 ~]# gfortran a.F -o a [root@c0108 ~]# ./a please input the counts N: 5 Result = .35355E+02 Time = 0.00 [root@c0108 ~]#