add assign operation of initializer_list

1030130376 · Apr 11, 2019 · 10fefb8 · 10fefb8
1 parent 0b3989e
commit 10fefb8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Associative containers/Set/stl_set.h b/Associative containers/Set/stl_set.h
@@ -71,6 +71,11 @@ class set {
     t = rhs.t;
     return *this;
   }
+  set& operator=(std::initializer_list<value_type> ils) {
+    clear();
+    insert(ils.begin(), ils.end());
+    return *this;
+  }
 
  public:  // move operation
   set(set&& rhs) noexcept : t(std::move(rhs.t)) {}
@@ -95,7 +100,7 @@ class set {
   const_reverse_iterator crend() const noexcept { return t.crend(); }
 
  public:  // swap
-  void swap(set<Key, Compare, Alloc>& rhs) noexcept { t.swap(rhs.t); }
+  void swap(set& rhs) noexcept { t.swap(rhs.t); }
 
  public:  // insert
   pair<iterator, bool> insert(const value_type& val) {