将结构字段的类型声明为与 Julia 中的结构本身相同的类型

Question

如果我在 Julia 中定义一个结构，是否可以将其中一个字段的类型声明为与结构本身的类型相同？例如，我想定义一个结构 NodeStruct 有两个字段，data 和 next 其中data 是一些标准类型（例如AbstractFloat）和next 是NodeStruct 类型或nothing .考虑这个给出错误LoadError: UndefVarError: NodeStruct not defined的示例代码。

import Base.@kwdef

@kwdef mutable struct NodeStruct{A<:AbstractFloat, B<:Union{NodeStruct, Nothing}}
    data ::A
    next ::B
end

有可能以某种方式做到这一点吗？我希望能够以参数方式执行此操作，因为performance tips 建议这样做，并且计算时间对我来说非常重要。我在types 或constructors 的文档中找不到任何解决此问题的信息。

Answer 1

既然你知道你的结构元素是一个具体的类型，这将是：

@kwdef mutable struct NodeStruct{A<:AbstractFloat}
    data ::A
    next ::Union{NodeStruct, Nothing}
end

现在你只需使用它

julia> NodeStruct(3.5, nothing)
NodeStruct{Float64}(3.5, nothing)

如果您需要更抽象的结构，您可以这样做：

abstract type AbstractNS end

@kwdef mutable struct NodeStruct2{A<:AbstractFloat, B<:AbstractNS} <: AbstractNS
    data ::A
    next ::Union{Nothing, B}
end

为这个抽象结构创建根时，您需要提供类型：

julia> root = NodeStruct2{Float64, NodeStruct2}(3.5, nothing)
NodeStruct2{Float64, NodeStruct2}(3.5, nothing)

但是对于叶子，它会起作用：

julia> leaf = NodeStruct2(5.6, root)
NodeStruct2{Float64, NodeStruct2{Float64, NodeStruct2}}(5.6, NodeStruct2{Float64, NodeStruct2}(3.5, nothing))