Question

我在PowerShell中寻找高效的查找表

朴素方法对大数据无效

$data = `
    @(
        @{
            A = 1;
            B = 2;
            C = 3;
        },
        @{
            A = 4;
            B = 5;
            C = 6;
        },
        @{
            A = 7;
            B = 8;
            C = 9;
        }
    )

# looking for value C base on pair (A, B)

function FindC
{
    param
    (
        [int] $A,
        [int] $B
    )

    $data | `
        Where-Object -FilterScript { ($_.A -eq $A) -and ($_.B -eq $B) } | `
        Select-Object -ExpandProperty C
}

我认为，有效的方法是使用哈希表，但不清楚如何处理复合键。我们不能使用hashtables，PSObject或PSCustomObject，因为它们没有实现相等

> @{ A = 1; B = 2; } -eq @{ A = 1; B = 2; }
False
> [PSObject] @{ A = 1; B = 2; } -eq [PSObject] @{ A = 1; B = 2; }
False
> [PSCustomObject] @{ A = 1; B = 2; } -eq [PSCustomObject] @{ A = 1; B = 2; }
False

我不想为这种简单的情况实现我自己的数据类。所以我到目前为止唯一发现的是元组

> [Tuple]::Create(1, 2) -eq [Tuple]::Create(1, 2)
True

所以我们可以使用元组

$hashtable = @{}
foreach ($entry in $data)
{
    $hashtable[[Tuple]::Create($entry.A, $entry.B)] = $entry.C
}

function FindC
{
    param
    (
        [int] $A,
        [int] $B
    )

    $hashtable[[Tuple]::Create($A, $B)]
}

任何人都知道更优雅的方式来实现这一目标吗？

Answer 1

如果要使用Hashtables，则可以覆盖Equals方法并提供自己的相等定义。请检查此问题：How to overload the PowerShell inbuilt class's methods

具有复合键

1 个答案: