optimize the vec dis joint

PennyLaneAI · Oct 18, 2024 · 3816dae · 3816dae
1 parent 29fff3f
commit 3816dae
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pennylane_lightning/core/src/utils/Util.hpp b/pennylane_lightning/core/src/utils/Util.hpp
@@ -585,7 +585,11 @@ std::vector<T1> cast_vector(const std::vector<T0> &vec) {
 template <typename T = std::size_t>
 bool areVecsDisjoint(const std::vector<T> &v1, const std::vector<T> &v2) {
     std::set<T> s0(v1.begin(), v1.end());
-    s0.insert(v2.begin(), v2.end());
-    return s0.size() == v1.size() + v2.size();
+    for (const auto &element : v2) {
+        if (s0.find(element) != s0.end()) {
+            return false;
+        }
+    }
+    return true;
 }
 } // namespace Pennylane::Util