108 lines
3.9 KiB
VB.net
108 lines
3.9 KiB
VB.net
Imports System.Runtime.CompilerServices
|
||
Imports Microsoft.VisualBasic.Language
|
||
|
||
Namespace ComponentModel.Algorithm.BinaryTree
|
||
|
||
Public Class ClusterKey(Of K)
|
||
|
||
ReadOnly members As New List(Of K)
|
||
ReadOnly views As Func(Of K, String)
|
||
|
||
Public ReadOnly Property NumberOfKey As Integer
|
||
Get
|
||
Return members.Count
|
||
End Get
|
||
End Property
|
||
|
||
Default Public ReadOnly Property Item(index As Integer) As K
|
||
Get
|
||
Return members(index)
|
||
End Get
|
||
End Property
|
||
|
||
Sub New([single] As K, views As Func(Of K, String))
|
||
Me.views = views
|
||
|
||
' Add a initial single member object
|
||
Call members.Add([single])
|
||
End Sub
|
||
|
||
<MethodImpl(MethodImplOptions.AggressiveInlining)>
|
||
Public Sub Add(newMember As K)
|
||
Call members.Add(newMember)
|
||
End Sub
|
||
|
||
Public Overrides Function ToString() As String
|
||
If members.Count = 1 Then
|
||
Return views(members(Scan0))
|
||
Else
|
||
Return views(members(Scan0)) & $", and with {members.Count} cluster members.."
|
||
End If
|
||
End Function
|
||
|
||
''' <summary>
|
||
''' 在这里应该是多个key比较一个query
|
||
''' </summary>
|
||
''' <param name="compares"></param>
|
||
''' <returns></returns>
|
||
Public Shared Function DoComparison(compares As Comparison(Of K)) As Func(Of ClusterKey(Of K), K, Integer)
|
||
Return Function(cluster, key) As Integer
|
||
Dim compareVal As Value(Of Integer) = -100
|
||
Dim values As New List(Of Integer)
|
||
|
||
For Each index As K In cluster.members
|
||
If (compareVal = compares(index, key)) = 0 Then
|
||
Return 0
|
||
Else
|
||
values += compareVal
|
||
End If
|
||
Next
|
||
|
||
If values.Where(Function(x) x = 1).Count > values.Count / 2 Then
|
||
Return 1
|
||
Else
|
||
Return -1
|
||
End If
|
||
End Function
|
||
End Function
|
||
End Class
|
||
|
||
Public Class AVLClusterTree(Of K) : Implements IEnumerable(Of ClusterKey(Of K))
|
||
|
||
ReadOnly avltree As AVLTree(Of ClusterKey(Of K), K)
|
||
ReadOnly views As Func(Of K, String)
|
||
|
||
Sub New(compares As Comparison(Of K), Optional views As Func(Of K, String) = Nothing)
|
||
Me.views = views
|
||
Me.avltree = New AVLTree(Of ClusterKey(Of K), K)(doCompares(compares), Function(c) c.ToString)
|
||
End Sub
|
||
|
||
Private Shared Function doCompares(compares As Comparison(Of K)) As Comparison(Of ClusterKey(Of K))
|
||
Dim unsymmetricalCompares = ClusterKey(Of K).DoComparison(compares)
|
||
|
||
' 在AVL数据模块中，比较代码中，待插入的key都是放在左边，即X参数部分
|
||
Return Function(x, y)
|
||
Return unsymmetricalCompares(y, x(Scan0))
|
||
End Function
|
||
End Function
|
||
|
||
<MethodImpl(MethodImplOptions.AggressiveInlining)>
|
||
Public Sub Add(key As K)
|
||
Call avltree.Add(New ClusterKey(Of K)(key, views), key, Sub(node, null) node.Key.Add(key))
|
||
End Sub
|
||
|
||
Public Sub Clear()
|
||
Call avltree.Clear()
|
||
End Sub
|
||
|
||
Public Iterator Function GetEnumerator() As IEnumerator(Of ClusterKey(Of K)) Implements IEnumerable(Of ClusterKey(Of K)).GetEnumerator
|
||
For Each cluster In avltree.root.PopulateNodes
|
||
Yield cluster.Key
|
||
Next
|
||
End Function
|
||
|
||
Private Iterator Function IEnumerable_GetEnumerator() As IEnumerator Implements IEnumerable.GetEnumerator
|
||
Yield GetEnumerator()
|
||
End Function
|
||
End Class
|
||
End Namespace |