如何在Fortran中用IF结构判断两个数组的整数匹配？

嘿，我完全懂你遇到的困扰——想用IF...THEN结构判断数组A的元素是否和数组B存在匹配，但之前的写法没法直接适配IF的要求对吧？咱们来拆解问题，看看哪里出了问题，以及怎么解决。

问题出在哪？ #

你原来的代码：

DO I = 1,SIZE(A)
  Match(J)=(ANY(A(I)==B))
END DO

这段代码是在给Match数组的第J个元素赋值，把A(I)是否在B中的判断结果存进去。但IF...THEN需要的是单个逻辑标量值（也就是.TRUE.或.FALSE.），如果直接把数组或者未明确索引的Match放进去，Fortran编译器肯定会报错，因为它没法把数组当成IF的判断条件。

解决方案1：直接在IF中使用ANY判断单个元素 #

如果你想逐个检查A的元素是否在B中匹配，直接把ANY(A(I) == B)作为IF的条件就行，不需要先赋值给Match数组：

INTEGER, DIMENSION(:), ALLOCATABLE :: A, B
! 假设这里已经给A和B赋值并分配了内存

DO I = 1, SIZE(A)
  ! 直接用ANY的结果作为IF的标量逻辑条件
  IF (ANY(A(I) == B)) THEN
    PRINT *, "A(", I, ") = ", A(I), " 与B中的元素匹配"
    ! 这里添加匹配后的处理逻辑
  END IF
END DO

解决方案2：检查两个数组是否存在任意匹配（无需循环） #

如果只是想判断A和B之间是否存在至少一个匹配元素，不需要逐个循环，直接嵌套ANY或者用COUNT函数：

! 方法1：嵌套ANY，DIM=1指定按列（也就是A的每个元素和B比较）
IF (ANY(ANY(A == B, DIM=1))) THEN
  PRINT *, "A和B中存在匹配的元素"
END IF

! 方法2：用COUNT统计匹配次数，只要大于0就说明有匹配
IF (COUNT(A == B) > 0) THEN
  PRINT *, "A和B中存在匹配的元素"
END IF

解决方案3：先记录所有匹配结果，再用IF判断 #

如果你需要后续多次使用匹配结果，可以先把每个A元素的匹配状态存入逻辑数组，之后再用这个数组做IF判断：

LOGICAL, DIMENSION(:), ALLOCATABLE :: matches
INTEGER, DIMENSION(:), ALLOCATABLE :: A, B

! 初始化数组
ALLOCATE(matches(SIZE(A)))

DO I = 1, SIZE(A)
  matches(I) = ANY(A(I) == B)
END DO

! 判断是否存在至少一个匹配
IF (ANY(matches)) THEN
  ! 提取所有匹配的位置
  PRINT *, "匹配的位置为：", PACK([(I, I=1,SIZE(A))], matches)
END IF

! 也可以判断某个特定位置是否匹配
IF (matches(3)) THEN
  PRINT *, "A的第3个元素在B中存在匹配"
END IF

额外提示 #

因为你提到两个数组都包含唯一的正数值，所以不用担心重复匹配的问题，上面的方法都可以直接使用。如果后续需要更高效的匹配（比如数组很大），可以先对其中一个数组排序，再用二分查找，但对于一般规模的数组，ANY已经足够好用啦。

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