mirror of
https://github.com/nim-lang/Nim.git
synced 2026-04-18 21:40:32 +00:00
symmetric difference operation for sets via xor (#24286)
closes https://github.com/nim-lang/RFCs/issues/554
Adds a symmetric difference operation to the language bitset type. This
maps to a simple `xor` operation on the backend and thus is likely
faster than the current alternatives, namely `(a - b) + (b - a)` or `a +
b - a * b`. The compiler VM implementation of bitsets already
implemented this via `symdiffSets` but it was never used.
The standalone binary operation is added to `setutils`, named
`symmetricDifference` in line with [hash
sets](https://nim-lang.org/docs/sets.html#symmetricDifference%2CHashSet%5BA%5D%2CHashSet%5BA%5D).
An operator version `-+-` and an in-place version like `toggle` as
described in the RFC are also added, implemented as trivial sugar.
(cherry picked from commit ae9287c4f3)
This commit is contained in:
@@ -75,3 +75,31 @@ func `[]=`*[T](t: var set[T], key: T, val: bool) {.inline.} =
|
||||
s[a3] = true
|
||||
assert s == {a2, a3}
|
||||
if val: t.incl key else: t.excl key
|
||||
|
||||
when defined(nimHasXorSet):
|
||||
func symmetricDifference*[T](x, y: set[T]): set[T] {.magic: "XorSet".} =
|
||||
## This operator computes the symmetric difference of two sets,
|
||||
## equivalent to but more efficient than `x + y - x * y` or
|
||||
## `(x - y) + (y - x)`.
|
||||
runnableExamples:
|
||||
assert symmetricDifference({1, 2, 3}, {2, 3, 4}) == {1, 4}
|
||||
else:
|
||||
func symmetricDifference*[T](x, y: set[T]): set[T] {.inline.} =
|
||||
result = x + y - (x * y)
|
||||
|
||||
proc `-+-`*[T](x, y: set[T]): set[T] {.inline.} =
|
||||
## Operator alias for `symmetricDifference`.
|
||||
runnableExamples:
|
||||
assert {1, 2, 3} -+- {2, 3, 4} == {1, 4}
|
||||
result = symmetricDifference(x, y)
|
||||
|
||||
proc toggle*[T](x: var set[T], y: set[T]) {.inline.} =
|
||||
## Toggles the existence of each value of `y` in `x`.
|
||||
## If any element in `y` is also in `x`, it is excluded from `x`;
|
||||
## otherwise it is included.
|
||||
## Equivalent to `x = symmetricDifference(x, y)`.
|
||||
runnableExamples:
|
||||
var x = {1, 2, 3}
|
||||
x.toggle({2, 3, 4})
|
||||
assert x == {1, 4}
|
||||
x = symmetricDifference(x, y)
|
||||
|
||||
Reference in New Issue
Block a user