JD2022-TU1/main/tools/framework/JD.Collections/Set.cs

740 lines
31 KiB
C#

using System;
using System.Collections.Generic;
using JD.Collections.Details;
using JD.Collections.Extensions;
using JD.Kernel.StaticHelpers;
namespace JD.Collections
{
/// <summary>
/// <see cref="Set{T}"/> is a collection that contains items of type T.
/// The item are maintained in a haphazard, unpredictable order, and duplicate items are not allowed.
/// </summary>
/// <remarks>
/// <p>The items are compared in one of two ways. If T implements <see cref="IComparable{T}"/>
/// then the Equals method of that interface will be used to compare items, otherwise the Equals
/// method from Object will be used. Alternatively, an instance of <see cref="IEqualityComparer{T}"/> can be passed
/// to the constructor to compare items.</p>
/// <p>Set is implemented as a hash table. Inserting, deleting, and looking up an
/// an element are all done in approximately constant time, regardless of the number of items in the Set.</p>
///</remarks>
[Serializable]
public class Set<T> : SetBase<T>, ICloneable
{
#region Variables
// The comparer used to hash/compare items.
private readonly IEqualityComparer<T> m_equalityComparer;
// The hash table that actually does the work of storing the items.
private IVariableSizeHash<T> m_hash;
#endregion
#region Constructors
/// <summary>
/// Creates a new Set. The Equals method and GetHashCode method on T
/// will be used to compare items for equality.
/// </summary>
///<remarks>
/// Items that are null are permitted, and will be sorted before all other items.
///</remarks>
public Set()
: this(null as IEqualityComparer<T>)
{
}
/// <summary>
/// Creates a new Set. The Equals and GetHashCode method of the passed comparer object
/// will be used to compare items in this set.
/// </summary>
/// <param name="equalityComparer">An instance of IEqualityComparer&lt;T&gt; that will be used to compare items.</param>
public Set(IEqualityComparer<T> equalityComparer)
{
m_equalityComparer = equalityComparer ?? EqualityComparer<T>.Default;
m_hash = new ShortHash<T>(m_equalityComparer);
}
/// <summary>
/// Creates a new Set. The Equals method and GetHashCode method on T
/// will be used to compare items for equality.
/// </summary>
///<remarks>
/// Items that are null are permitted.
///</remarks>
/// <param name="collection">A collection with items to be placed into the Set.</param>
public Set(IEnumerable<T> collection)
: this(collection, EqualityComparer<T>.Default)
{
}
/// <summary>
/// Creates a new Set. The Equals and GetHashCode method of the passed comparer object
/// will be used to compare items in this set. The set is
/// initialized with all the items in the given collection.
/// </summary>
/// <param name="collection">A collection with items to be placed into the Set.</param>
/// <param name="equalityComparer">An instance of IEqualityComparer&lt;T&gt; that will be used to compare items.</param>
public Set(IEnumerable<T> collection, IEqualityComparer<T> equalityComparer)
: this(equalityComparer)
{
this.AddRange(collection);
}
/// <summary>
/// Creates a new Set given a comparer and a tree that contains the data. Used
/// internally for Clone.
/// </summary>
/// <param name="equalityComparer">EqualityComparer for the set.</param>
/// <param name="hash">Data for the set.</param>
private Set(IEqualityComparer<T> equalityComparer, IVariableSizeHash<T> hash)
{
m_equalityComparer = equalityComparer;
m_hash = hash;
}
#endregion Constructors
#region Methods
#region Cloning
/// <summary>
/// Makes a shallow clone of this set; i.e., if items of the
/// set are reference types, then they are not cloned. If T is a value type,
/// then each element is copied as if by simple assignment.
/// </summary>
/// <remarks>Cloning the set takes time O(N), where N is the number of items in the set.</remarks>
/// <returns>The cloned set.</returns>
object ICloneable.Clone()
{
return Clone();
}
/// <summary>
/// Makes a shallow clone of this set; i.e., if items of the
/// set are reference types, then they are not cloned. If T is a value type,
/// then each element is copied as if by simple assignment.
/// </summary>
/// <remarks>Cloning the set takes time O(N), where N is the number of items in the set.</remarks>
/// <returns>The cloned set.</returns>
public Set<T> Clone()
{
return new Set<T>(m_equalityComparer, m_hash.Clone());
}
#endregion Cloning
#region Basic collection containment
/// <summary>
/// Returns the <see cref="IEqualityComparer{T}"/> used to compare items in this set.
/// </summary>
/// <value>If the set was created using a comparer, that comparer is returned. Otherwise
/// the default comparer for T (<see cref="EqualityComparer{T}.Default"/>) is returned.</value>
public IEqualityComparer<T> Comparer
{
get
{
return m_equalityComparer;
}
}
/// <summary>
/// Returns the number of items in the set.
/// </summary>
/// <remarks>The size of the set is returned in constant time.</remarks>
/// <value>The number of items in the set.</value>
public sealed override int Count
{
get
{
return m_hash.Count;
}
}
/// <summary>
/// Returns an enumerator that enumerates all the items in the set.
/// The items are enumerated in sorted order.
/// </summary>
/// <remarks>
/// <p>Typically, this method is not called directly. Instead the "foreach" statement is used
/// to enumerate the items, which uses this method implicitly.</p>
/// <p>If an item is added to or deleted from the set while it is being enumerated, then
/// the enumeration will end with an InvalidOperationException.</p>
/// <p>Enumerating all the items in the set takes time O(N), where N is the number
/// of items in the set.</p>
/// </remarks>
/// <returns>An enumerator for enumerating all the items in the Set.</returns>
public sealed override IEnumerator<T> GetEnumerator()
{
return m_hash.GetEnumerator();
}
/// <summary>
/// Determines if this set contains an item equal to <paramref name="item"/>. The set is not changed.
/// </summary>
/// <remarks>Searching the set for an item takes approximately constant time, regardless of the number of items in the set.</remarks>
/// <param name="item">The item to search for.</param>
/// <returns>True if the set contains <paramref name="item"/>. False if the set does not contain <paramref name="item"/>.</returns>
public sealed override bool Contains(T item)
{
return m_hash.Contains(item);
}
/// <summary>
/// <para>Determines if this set contains an item equal to <paramref name="item"/>, according to the comparison mechanism that was used when the set was created. The set is not changed.</para>
/// <para>If the set does contain an item equal to <paramref name="item"/>, then the item from the set is returned.</para>
/// </summary>
/// <remarks>Searching the set for an item takes approximately constant time, regardless of the number of items in the set.</remarks>
/// <example>
/// In the following example, the set contains strings which are compared in a case-insensitive manner.
/// <code>
/// Set&lt;string&gt; set = new Set&lt;string&gt;(StringComparer.CurrentCultureIgnoreCase);
/// set.Add("HELLO");
/// string s;
/// bool b = set.TryGetItem("Hello", out s); // b receives true, s receives "HELLO".
/// </code>
/// </example>
/// <param name="item">The item to search for.</param>
/// <param name="foundItem">Returns the item from the set that was equal to <paramref name="item"/>.</param>
/// <returns>True if the set contains <paramref name="item"/>. False if the set does not contain <paramref name="item"/>.</returns>
public bool TryGetItem(T item, out T foundItem)
{
return m_hash.TryGetValue(item, out foundItem);
}
#endregion
#region Adding elements
/// <summary>
/// Adds a new item to the set. If the set already contains an item equal to
/// <paramref name="item"/>, that item is replaced with <paramref name="item"/>.
/// </summary>
/// <remarks>
/// <para>Equality between items is determined by the comparison instance or delegate used to create the set.</para>
/// <para>Adding an item takes approximately constant time, regardless of the number of items in the set.</para></remarks>
/// <param name="item">The item to add to the set.</param>
/// <returns>True if it added the value (if it wasn't present).</returns>
public new virtual bool Add(T item)
{
return m_hash.Add(item, CollisionBehavior.Replace, ref m_hash);
}
/// <summary>
/// Redirect the void Add(T) of the base class to the bool Add() whose hides void Add().
/// </summary>
/// <param name="item"></param>
protected override void AddImpl(T item)
{
Add(item);
}
#endregion Adding elements
#region Removing elements
/// <summary>
/// Searches the set for an item equal to <paramref name="item"/>, and if found,
/// removes it from the set. If not found, the set is unchanged.
/// </summary>
/// <remarks>
/// <para>Equality between items is determined by the comparison instance or delegate used to create the set.</para>
/// <para>Removing an item from the set takes approximately constant time, regardless of the size of the set.</para></remarks>
/// <param name="item">The item to remove.</param>
/// <returns>True if <paramref name="item"/> was found and removed. False if <paramref name="item"/> was not in the set.</returns>
public override bool Remove(T item)
{
return m_hash.Remove(item);
}
/// <summary>
/// Removes all the items in <paramref name="collection"/> from the set.
/// </summary>
/// <remarks>
/// <para>Equality between items is determined by the comparison instance or delegate used to create the set.</para>
/// <para>Removing the collection takes time O(M), where M is the
/// number of items in <paramref name="collection"/>.</para></remarks>
/// <param name="collection">A collection of items to remove from the set.</param>
/// <returns>The number of items removed from the set.</returns>
/// <exception cref="ArgumentNullException"><paramref name="collection"/> is null.</exception>
public int RemoveRange(IEnumerable<T> collection)
{
int count = 0;
if (collection != null)
{
if (collection == this)
{
count = Count;
Clear(); // special case, otherwise we will throw.
}
else
{
foreach (T item in collection)
{
if (Remove(item))
{
++count;
}
}
}
}
return count;
}
/// <summary>
/// Removes all items from the set.
/// </summary>
/// <remarks>Clearing the set takes a constant amount of time, regardless of the number of items in it.</remarks>
public override void Clear()
{
m_hash.Clear();
}
#endregion Removing elements
#region Set operations
/// <summary>
/// Check that this set and another set were created with the same comparison
/// mechanism. Throws exception if not compatible.
/// </summary>
/// <param name="otherSet">Other set to check comparision mechanism.</param>
/// <exception cref="InvalidOperationException">If otherSet and this set don't use the same method for comparing items.</exception>
private void CheckConsistentComparison(Set<T> otherSet)
{
Throw.IfNull(otherSet, "otherSet");
Throw.InvalidOperationIf(!Equals(m_equalityComparer, otherSet.m_equalityComparer), Strings.InconsistentComparisons);
}
/// <summary>
/// Determines if this set is a superset of another set. Neither set is modified.
/// This set is a superset of <paramref name="otherSet"/> if every element in
/// <paramref name="otherSet"/> is also in this set.
/// <remarks>IsSupersetOf is computed in time O(M), where M is the size of the
/// <paramref name="otherSet"/>.</remarks>
/// </summary>
/// <param name="otherSet">Set to compare to.</param>
/// <returns>True if this is a superset of <paramref name="otherSet"/>.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public bool IsSupersetOf(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
if (otherSet.Count > Count)
{
return false; // Can't be a superset of a bigger set
}
// Check each item in the other set to make sure it is in this set.
foreach (T item in otherSet)
{
if (!Contains(item))
{
return false;
}
}
return true;
}
/// <summary>
/// Determines if this set is a proper superset of another set. Neither set is modified.
/// This set is a proper superset of <paramref name="otherSet"/> if every element in
/// <paramref name="otherSet"/> is also in this set.
/// Additionally, this set must have strictly more items than <paramref name="otherSet"/>.
/// </summary>
/// <remarks>IsProperSubsetOf is computed in time O(M), where M is the size of
/// <paramref name="otherSet"/>.</remarks>
/// <param name="otherSet">Set to compare to.</param>
/// <returns>True if this is a proper superset of <paramref name="otherSet"/>.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public bool IsProperSupersetOf(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
if (otherSet.Count >= Count)
{
return false; // Can't be a proper superset of a bigger or equal set
}
return IsSupersetOf(otherSet);
}
/// <summary>
/// Determines if this set is a subset of another set. Neither set is modified.
/// This set is a subset of <paramref name="otherSet"/> if every element in this set
/// is also in <paramref name="otherSet"/>.
/// </summary>
/// <remarks>IsSubsetOf is computed in time O(N), where N is the size of the this set.</remarks>
/// <param name="otherSet">Set to compare to.</param>
/// <returns>True if this is a subset of <paramref name="otherSet"/>.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public bool IsSubsetOf(Set<T> otherSet)
{
Throw.IfNull(otherSet, "otherSet");
return otherSet.IsSupersetOf(this);
}
/// <summary>
/// Determines if this set is a proper subset of another set. Neither set is modified.
/// This set is a subset of <paramref name="otherSet"/> if every element in this set
/// is also in <paramref name="otherSet"/>. Additionally, this set must have strictly
/// fewer items than <paramref name="otherSet"/>.
/// </summary>
/// <remarks>IsProperSubsetOf is computed in time O(N), where N is the size of the this set.</remarks>
/// <param name="otherSet">Set to compare to.</param>
/// <returns>True if this is a proper subset of <paramref name="otherSet"/>.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public bool IsProperSubsetOf(Set<T> otherSet)
{
Throw.IfNull(otherSet, "otherSet");
return otherSet.IsProperSupersetOf(this);
}
/// <summary>
/// Determines if this set is equal to another set. This set is equal to
/// <paramref name="otherSet"/> if they contain the same items.
/// </summary>
/// <remarks>IsEqualTo is computed in time O(N), where N is the number of items in
/// this set.</remarks>
/// <param name="otherSet">Set to compare to</param>
/// <returns>True if this set is equal to <paramref name="otherSet"/>, false otherwise.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public bool IsEqualTo(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
// Must be the same size.
if (otherSet.Count != Count)
{
return false;
}
// Check each item in the other set to make sure it is in this set.
foreach (T item in otherSet)
{
if (!Contains(item))
{
return false;
}
}
return true;
}
/// <summary>
/// Determines if this set is disjoint from another set. Two sets are disjoint
/// if no item from one set is equal to any item in the other set.
/// </summary>
/// <remarks>
/// <para>The answer is computed in time O(N), where N is the size of the smaller set.</para>
/// </remarks>
/// <param name="otherSet">Set to check disjointness with.</param>
/// <returns>True if the two sets are disjoint, false otherwise.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public bool IsDisjointFrom(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
Set<T> smaller, larger;
if (otherSet.Count > Count)
{
smaller = this; larger = otherSet;
}
else
{
smaller = otherSet; larger = this;
}
foreach (T item in smaller)
{
if (larger.Contains(item))
{
return false;
}
}
return true;
}
/// <summary>
/// Computes the union of this set with another set. The union of two sets
/// is all items that appear in either or both of the sets. This set receives
/// the union of the two sets, the other set is unchanged.
/// </summary>
/// <remarks>
/// <para>If equal items appear in both sets, the union will include an arbitrary choice of one of the
/// two equal items.</para>
/// <para>The union of two sets is computed in time O(M + N), where M is the size of the
/// larger set, and N is the size of the smaller set.</para>
/// </remarks>
/// <param name="otherSet">Set to union with.</param>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public void UnionWith(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
this.AddRange(otherSet);
}
/// <summary>
/// Computes the union of this set with another set. The union of two sets
/// is all items that appear in either or both of the sets. A new set is
/// created with the union of the sets and is returned. This set and the other set
/// are unchanged.
/// </summary>
/// <remarks>
/// <para>If equal items appear in both sets, the union will include an arbitrary choice of one of the
/// two equal items.</para>
/// <para>The union of two sets is computed in time O(M + N), where M is the size of the
/// one set, and N is the size of the other set.</para>
/// </remarks>
/// <param name="otherSet">Set to union with.</param>
/// <returns>The union of the two sets.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public Set<T> Union(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
Set<T> smaller, larger;
if (otherSet.Count > Count)
{
smaller = this; larger = otherSet;
}
else
{
smaller = otherSet; larger = this;
}
Set<T> result = larger.Clone();
result.AddRange(smaller);
return result;
}
/// <summary>
/// Computes the intersection of this set with another set. The intersection of two sets
/// is all items that appear in both of the sets. This set receives
/// the intersection of the two sets, the other set is unchanged.
/// </summary>
/// <remarks>
/// <para>When equal items appear in both sets, the intersection will include an arbitrary choice of one of the
/// two equal items.</para>
/// <para>The intersection of two sets is computed in time O(N), where N is the size of the smaller set.</para>
/// </remarks>
/// <param name="otherSet">Set to intersection with.</param>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public void IntersectionWith(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
Set<T> smaller, larger;
if (otherSet.Count > Count)
{
smaller = this; larger = otherSet;
}
else
{
smaller = otherSet; larger = this;
}
IVariableSizeHash<T> newHash = new ShortHash<T>(m_equalityComparer);
foreach (T item in smaller)
{
if (larger.Contains(item))
{
newHash.Add(item, CollisionBehavior.Throw, ref newHash);
}
}
m_hash = newHash;
}
/// <summary>
/// Computes the intersection of this set with another set. The intersection of two sets
/// is all items that appear in both of the sets. A new set is
/// created with the intersection of the sets and is returned. This set and the other set
/// are unchanged.
/// </summary>
/// <remarks>
/// <para>When equal items appear in both sets, the intersection will include an arbitrary choice of one of the
/// two equal items.</para>
/// <para>The intersection of two sets is computed in time O(N), where N is the size of the smaller set.</para>
/// </remarks>
/// <param name="otherSet">Set to intersection with.</param>
/// <returns>The intersection of the two sets.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public Set<T> Intersection(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
Set<T> smaller, larger;
if (otherSet.Count > Count)
{
smaller = this; larger = otherSet;
}
else
{
smaller = otherSet; larger = this;
}
Set<T> result = new Set<T>(m_equalityComparer);
foreach (T item in smaller)
{
if (larger.Contains(item))
{
result.Add(item);
}
}
return result;
}
/// <summary>
/// Computes the difference of this set with another set. The difference of these two sets
/// is all items that appear in this set, but not in <paramref name="otherSet"/>. This set receives
/// the difference of the two sets; the other set is unchanged.
/// </summary>
/// <remarks>
/// <para>The difference of two sets is computed in time O(N), where N is the size of the smaller set.</para>
/// </remarks>
/// <param name="otherSet">Set to difference with.</param>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public void DifferenceWith(Set<T> otherSet)
{
// Difference with myself is nothing. This check is needed because the
// main algorithm doesn't work correctly otherwise.
if (this == otherSet)
{
Clear();
}
CheckConsistentComparison(otherSet);
foreach (T item in otherSet)
{
Remove(item);
}
}
/// <summary>
/// Computes the difference of this set with another set. The difference of these two sets
/// is all items that appear in this set, but not in <paramref name="otherSet"/>. A new set is
/// created with the difference of the sets and is returned. This set and the other set
/// are unchanged.
/// </summary>
/// <remarks>
/// <para>The difference of two sets is computed in time O(N), where N is the size of the smaller set.</para>
/// </remarks>
/// <param name="otherSet">Set to difference with.</param>
/// <returns>The difference of the two sets.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public Set<T> Difference(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
Set<T> result = Clone();
result.DifferenceWith(otherSet);
return result;
}
/// <summary>
/// Computes the symmetric difference of this set with another set. The symmetric difference of two sets
/// is all items that appear in either of the sets, but not both. This set receives
/// the symmetric difference of the two sets; the other set is unchanged.
/// </summary>
/// <remarks>
/// <para>The symmetric difference of two sets is computed in time O(N), where N is the size of the smaller set.</para>
/// </remarks>
/// <param name="otherSet">Set to symmetric difference with.</param>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public void SymmetricDifferenceWith(Set<T> otherSet)
{
// main algorithm doesn't work correctly otherwise.
if (this == otherSet)
{
Clear();
}
CheckConsistentComparison(otherSet);
if (otherSet.Count > Count)
{
m_hash.StopEnumerations();
IVariableSizeHash<T> newHash = otherSet.m_hash.Clone();
foreach (T item in this)
{
if (!newHash.Remove(item))
{
newHash.Add(item, CollisionBehavior.Throw, ref newHash);
}
}
m_hash = newHash;
}
else
{
foreach (T item in otherSet)
{
if (Contains(item))
{
Remove(item);
}
else
{
Add(item);
}
}
}
}
/// <summary>
/// Computes the symmetric difference of this set with another set. The symmetric difference of two sets
/// is all items that appear in either of the sets, but not both. A new set is
/// created with the symmetric difference of the sets and is returned. This set and the other set
/// are unchanged.
/// </summary>
/// <remarks>
/// <para>The symmetric difference of two sets is computed in time O(N), where N is the size of the smaller set.</para>
/// </remarks>
/// <param name="otherSet">Set to symmetric difference with.</param>
/// <returns>The symmetric difference of the two sets.</returns>
/// <exception cref="InvalidOperationException">This set and <paramref name="otherSet"/> don't use the same method for comparing items.</exception>
public Set<T> SymmetricDifference(Set<T> otherSet)
{
CheckConsistentComparison(otherSet);
Set<T> smaller, larger;
if (otherSet.Count > Count)
{
smaller = this; larger = otherSet;
}
else
{
smaller = otherSet; larger = this;
}
Set<T> result = larger.Clone();
foreach (T item in smaller)
{
if (result.Contains(item))
{
result.Remove(item);
}
else
{
result.Add(item);
}
}
return result;
}
#endregion Set operations
#endregion
}
}