diff --git a/library/alloc/src/collections/btree/set.rs b/library/alloc/src/collections/btree/set.rs index caa629cf4e65e..31655c118072d 100644 --- a/library/alloc/src/collections/btree/set.rs +++ b/library/alloc/src/collections/btree/set.rs @@ -1093,7 +1093,13 @@ impl FromIterator for BTreeSet { // use stable sort to preserve the insertion order. inputs.sort(); - let iter = inputs.into_iter().map(|k| (k, ())); + BTreeSet::from_sorted_iter(inputs.into_iter()) + } +} + +impl BTreeSet { + fn from_sorted_iter>(iter: I) -> BTreeSet { + let iter = iter.map(|k| (k, ())); let map = BTreeMap::bulk_build_from_sorted_iter(iter); BTreeSet { map } } @@ -1258,11 +1264,10 @@ impl Sub<&BTreeSet> for &BTreeSet { /// let b = BTreeSet::from([3, 4, 5]); /// /// let result = &a - &b; - /// let result_vec: Vec<_> = result.into_iter().collect(); - /// assert_eq!(result_vec, [1, 2]); + /// assert_eq!(result, BTreeSet::from([1, 2])); /// ``` fn sub(self, rhs: &BTreeSet) -> BTreeSet { - self.difference(rhs).cloned().collect() + BTreeSet::from_sorted_iter(self.difference(rhs).cloned()) } } @@ -1281,11 +1286,10 @@ impl BitXor<&BTreeSet> for &BTreeSet { /// let b = BTreeSet::from([2, 3, 4]); /// /// let result = &a ^ &b; - /// let result_vec: Vec<_> = result.into_iter().collect(); - /// assert_eq!(result_vec, [1, 4]); + /// assert_eq!(result, BTreeSet::from([1, 4])); /// ``` fn bitxor(self, rhs: &BTreeSet) -> BTreeSet { - self.symmetric_difference(rhs).cloned().collect() + BTreeSet::from_sorted_iter(self.symmetric_difference(rhs).cloned()) } } @@ -1304,11 +1308,10 @@ impl BitAnd<&BTreeSet> for &BTreeSet { /// let b = BTreeSet::from([2, 3, 4]); /// /// let result = &a & &b; - /// let result_vec: Vec<_> = result.into_iter().collect(); - /// assert_eq!(result_vec, [2, 3]); + /// assert_eq!(result, BTreeSet::from([2, 3])); /// ``` fn bitand(self, rhs: &BTreeSet) -> BTreeSet { - self.intersection(rhs).cloned().collect() + BTreeSet::from_sorted_iter(self.intersection(rhs).cloned()) } } @@ -1327,11 +1330,10 @@ impl BitOr<&BTreeSet> for &BTreeSet { /// let b = BTreeSet::from([3, 4, 5]); /// /// let result = &a | &b; - /// let result_vec: Vec<_> = result.into_iter().collect(); - /// assert_eq!(result_vec, [1, 2, 3, 4, 5]); + /// assert_eq!(result, BTreeSet::from([1, 2, 3, 4, 5])); /// ``` fn bitor(self, rhs: &BTreeSet) -> BTreeSet { - self.union(rhs).cloned().collect() + BTreeSet::from_sorted_iter(self.union(rhs).cloned()) } }