Julia：带有包含复合类型的元组键的字典

Question

我在尝试使用其键是包含复合类型的元组的字典时遇到了问题。

这是一个复制我的问题的最小示例：

import Base: hash, isequal    

type T
    a :: Int
    b :: Int
end

function isequal(A::(T,Int), B::(T,Int))
    A[1].a == B[1].a && A[1].b == B[1].b && A[2] == B[2]
end
function hash(A::(T,Int))
    hash(A[1].a + A[1].b + A[2])
end

d = Dict{(T,Int),Int}()

d[(T(1,1),1)] = 1
d[(T(2,2),2)] = 2

r = (T(2,2),2)

for k in keys(d)
    println(isequal(r, k) && hash(r) == hash(k))
end
println(d[r])

运行此结果：

false
true
ERROR: key not found: (T(2,2),2)

所以isequal 和hash 有效，但由于某种原因，dict 无效。

有人知道发生了什么吗？谢谢。

Answer 1

在这种情况下，我不太了解元组的类型和调度，但基本上是you need to implement the two argument form of hash for this case。作为比较点，以下非常相似的代码按预期工作，没有两个参数形式：

type T
    a::Int
    b::Int
end

function Base.isequal(A::T, B::T)
    println("isequal", A, " ", B)
    A.a == B.a && A.b == B.b
end
function Base.hash(A::T)
    println("hash", A)
    hash(A.a + A.b)
end

d = Dict{T,Int}()

d[T(1,1)] = 1
d[T(2,2)] = 2

println("test")
r = T(2,2)
println(d[r])

有输出

isequalT(1,1) T(1,1)
hashT(1,1)
isequalT(2,2) T(2,2)
hashT(2,2)
test
hashT(2,2)
isequalT(2,2) T(2,2)
2

Answer 2

可以通过为每个复合类型而不是整个元组创建 hash 和 isequal 函数来解决问题。

如果你的密钥是(T,Int):

而不是这个：

function isequal(A::(T,Int), B::(T,Int))
    A[1].a == B[1].a && A[1].b == B[1].b && A[2] == B[2]
end
function hash(A::(T,Int))
    hash(A[1].a + A[1].b + A[2])
end

这样做：

function isequal(A::T, B::T)
    A.a == B.a && A.b == B.b
end
function hash(A::T)
    hash(A.a + A.b)
end

如果您希望原始公式起作用，则必须使用可选的第二个参数指定 Base.hash：h::Uint64:

function isequal(A::(T,Int), B::(T,Int))
    A[1].a == B[1].a && A[1].b == B[1].b && A[2] == B[2]
end
function hash(A::(T,Int))
    hash(A[1].a + A[1].b + A[2])
end
function hash(A::(T,Int), h::Uint64)
    hash(A[1].a + A[1].b + A[2] + h)
end