From 3816dae6c0603b2b30a8da370b56554db97b76c8 Mon Sep 17 00:00:00 2001 From: Shuli Shu <08cnbj@gmail.com> Date: Fri, 18 Oct 2024 18:00:22 +0000 Subject: [PATCH] optimize the vec dis joint --- pennylane_lightning/core/src/utils/Util.hpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/pennylane_lightning/core/src/utils/Util.hpp b/pennylane_lightning/core/src/utils/Util.hpp index 0186064f1..6f512ac40 100644 --- a/pennylane_lightning/core/src/utils/Util.hpp +++ b/pennylane_lightning/core/src/utils/Util.hpp @@ -585,7 +585,11 @@ std::vector cast_vector(const std::vector &vec) { template bool areVecsDisjoint(const std::vector &v1, const std::vector &v2) { std::set 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