下面是一个在Haskell中返回True的函数,如果存在一个由三个相等的连续对角线元素组成的列表:
import Data.List (transpose)
checkDiagonal :: Eq a => [[a]] -> Bool
checkDiagonal matrix = any hasThreeEqualDiagonal (matrix ++ transpose matrix)
hasThreeEqualDiagonal :: Eq a => [a] -> Bool
hasThreeEqualDiagonal [] = False
hasThreeEqualDiagonal [_, _] = False
hasThreeEqualDiagonal [x, y, z, _] = x == y && y == z
hasThreeEqualDiagonal (x : y : z : xs) = x == y && y == z || hasThreeEqualDiagonal (y : z : xs)
main :: IO ()
main = do
let matrix = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
putStrLn $ show $ checkDiagonal matrix
在上面的代码中,checkDiagonal
函数接受一个二维列表作为输入,它将输入的列表与其转置连接在一起,以便同时检查行和列的对角线。然后,它使用hasThreeEqualDiagonal
函数来检查每个列表是否存在三个连续相等的元素。如果有任何一个列表满足条件,那么checkDiagonal
函数将返回True,否则返回False。
在hasThreeEqualDiagonal
函数中,我们递归地检查每个列表的连续三个元素是否相等。如果列表的长度小于3,则直接返回False。如果列表的前三个元素相等,则返回True。否则,递归地检查列表的剩余部分。
在主函数中,我们定义了一个示例矩阵,并调用checkDiagonal
函数来检查是否存在三个连续相等的对角线元素。然后,我们使用putStrLn
函数将结果打印到控制台上。