Question

我正在尝试实现solLoops函数的设置版本：

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <ctime>
#include <vector>
#include <algorithm>
#include <unordered_set>

int solLoops(std::vector<int> A)
{

    int count = 0;
    for (int i = 0; i < A.size() - 1; ++i)
    {
        if (A[i + 1] - A[i] <= 1)
        {
            count++;
        }
    }
    return count;
}

int solSet(std::vector<int> A)
{
    std::unordered_set<int> S;

    for (int i = 0; i  < A.size(); ++i)
    {
        S.insert(A[i]);
    }

    int count = 0;
    for (auto it = S.begin(); it != std::prev(S.end()); ++it)
    {
        auto it2 = std::next(it, 1);
        if (*it2 - *it <= 1)
        {   
            count++;
        }
    }

    return count;
}



int main() {

    const unsigned int N = 4;

    std::vector<int> A;
    A.reserve(N);
    A.emplace_back(11);
    A.emplace_back(12);
    A.emplace_back(13);
    A.emplace_back(1);
    A.emplace_back(2);

    //int result = solLoops(A);
    int result = solSet(A);
    std::cout << "Result is: " << result << "\n";
    std::cout << "\n";

    return 0;
}

但是我遇到了一个段错误。我正在尝试实现循环：

for (int i = 0; i < A.size() - 1; ++i)

通过

for (auto it = S.begin(); it != std::prev(S.end()); ++it)

并进行减法：

if (A[i + 1] - A[i] <= 1)

as

auto it2 = std::next(it, 1); if (*it2 - *it <= 1)

（我知道这个集合包含了随机顺序的元素。我只想弄清楚如何在这里进行减法）

Answer 1

std::unordered_set有forward iterators（只能前进）。但std::prev需要bidirectional iterator。

您可以通过检查 next 迭代器是否结束来解决您的问题：std::next(it) != S.end()。或者使用std::set代替拥有双向迭代器。

减去连续的设定元素

1 个答案: