将较小的向量与较大的向量进行比较,以检查它在较小向量的末尾是否不同

Compare smaller vector to bigger to check if it differs at the end of smaller

本文关键字:向量 检查 是否 比较      更新时间:2023-10-16

我们有两个大小的向量,取决于运行时,需要检查它们是否相等 - 只有在较小尺寸的向量结束后才不同元素。我使用了 std::equal,但问题是我需要首先找到哪个向量尺寸较小,这会导致额外的代码行:

#include <vector>
#include <iostream>
int main()
{
  std::vector<int> a(1000, 3);
  std::vector<int> a1(100, 3);
  if(a.size() > a1.size())
  {
    if(std::equal(a1.begin(), a1.end(), a.begin()))
    {
      std::cout << "Same a gt a1" << std::endl;
    }
  }
  if(a1.size() > a.size())
  {
    if(std::equal(a.begin(), a.end(), a1.begin()))
    {
      std::cout << "Same a1 gt a" << std::endl;
    }
  }
  if(a1.size() == a.size())
  {
    if(std::equal(a.begin(), a.end(), a1.begin()))
    {
       std::cout << "Same a = a1" << std::endl;
    }
  }
}

是否可以改进比较两个向量或仅在较小向量末尾不同的代码?

从 C++14 开始,您可以使用 std::mismatch 并对照每个范围的末尾检查返回的一对迭代器:

auto it = std::mismatch(a.begin(), a.end(), a1.begin(), a1.end());
if (it.first == a.end() || it.second == a1.end()) {
    // Equality
}

还需要知道元素从哪里开始不同,如果没有,则在这一点上,较大的向量更大(您不想比较的子范围的开始)。

如果您事先计算较小的大小,则只需要一次调用std::equal。我会像这样重构代码:

#include <vector>
#include <iostream>
#include <algorithm>
int main()
{
    std::vector<int> a(1000, 3);
    std::vector<int> a1(100, 3);
    if (std::equal(a1.begin(), a1.begin() + std::min(a.size(), a1.size()), a.begin())) 
    {
        std::cout << "Same" << std::endl;
    }
    return 0;
}

如果你需要保留关于哪个向量更大的第二个信息,你可以像这样实现它,例如:

std::cout << "Same " << ((a.size() == a1.size())? "a = a1" : ((a.size() > a1.size())? "a gt a1" : "a1 gt a")) << std::endl;

这是一个纯 C++11 解决方案,应该适用于任何顺序容器(例如 std::vectorstd::liststd::deque)。它使用自定义返回类型,因为您在原始代码段中显示的比较类型包含的信息比简单的布尔值可以包含的信息更多。

enum class CombinedCompareResult {
   NotEqual, EqualAndFirstLarger, EqualAndSecondLarger, EqualIncludingSize
};
template <class Rng1, class Rng2>
CombinedCompareResult combinedCompare(const Rng1& rng1, const Rng2& rng2)
{
   using std::begin;
   const auto elementsToCompare = std::min(rng1.size(), rng2.size());
   if (!std::equal(begin(rng1), std::next(begin(rng1), elementsToCompare), begin(rng2)))
      return CombinedCompareResult::NotEqual;
   else if (rng1.size() == rng2.size())
      return CombinedCompareResult::EqualIncludingSize;
   else if (rng1.size() > rng2.size())
      return CombinedCompareResult::EqualAndFirstLarger;
   else
      return CombinedCompareResult::EqualAndSecondLarger;
}

这可以像下面一样使用,并且应该导致与问题中的代码相同的行为。

const auto cmp = combinedCompare(lst, a);
if (cmp == CombinedCompareResult::EqualIncludingSize)
    std::cout << "Same a = a1" << std::endl;
else if (cmp == CombinedCompareResult::EqualAndFirstLarger)
    std::cout << "Same a gt a1" << std::endl;
else if (cmp == CombinedCompareResult::EqualAndSecondLarger)
    std::cout << "Same a1 gt a" << std::endl;