所以我真的很挣扎,甚至现在我对我的解决方案不满意。
我的set
至少包含0,并且可能包含其他正int
个。我需要在set
中找到第一个正数不。
所以写一个标准的while
- 循环来实现这一点很容易。
i = foo.begin();
while (i != prev(foo.end()) && *i + 1 == *next(i)){
++i;
}
cout << "First number that cannot be formed " << *i + 1 << endl;
但是当我尝试编写循环的STL算法版本时,我得到的东西就像这个循环一样失败了:
auto i = foo.begin();
while (++i != prev(foo.end()) && *prev(i) + 1 == *i);
cout << "First number that cannot be formed " << *prev(i) + 1 << endl;
在以下情况下,这两个循环都正确地产生 3
set<int> foo{0, 1, 2, 4};
但在这种情况下,第二个循环错误地产生3而不是 4 :
set<int> foo{0, 1, 2, 3};
如何使用STL算法编写此内容并完成第一个循环的行为?
修改
在看到一些答案后,我想增加难度。我真正想要的是不需要临时变量的东西,可以放在cout
语句中。
答案 0 :(得分:2)
你的循环问题是你过早地停止了一个元素。这有效:
while (++i != foo.end() && *prev(i) + 1 == *i);
第一个循环的差异是条件*prev(i) + 1 == *i)
而不是*i + 1 == *next(i)
;你检查的范围必须相应地移动。
您也可以使用std::adjacent_find
:
auto i = std::adjacent_find(begin(s), end(s), [](int i, int j) { return i + 1 != j; });
if(i == s.end()) {
std::cout << *s.rbegin() + 1 << std::endl;
} else {
std::cout << *i + 1 << std::endl;
}
对编辑的回应:使其可以精确地内联的一种方法是
std::cout << std::accumulate(begin(s), end(s), 0,
[](int last, int x) {
return last + 1 == x ? x : last;
}) + 1 << '\n';
...但效率较低,因为它在发现间隙时不会短路。另一种做短路的方法是
std::cout << *std::mismatch(begin (s),
prev(end (s)),
next(begin(s)),
[](int lhs, int rhs) {
return lhs + 1 == rhs;
}).first + 1 << '\n';
答案 1 :(得分:2)
您是否尝试过adjacent_find
?
#include <algorithm>
#include <iostream>
#include <set>
int main()
{
std::set<int> foo{0, 1, 2, 4};
auto it = std::adjacent_find(begin(foo), end(foo),
[](auto e1, auto e2){ return (e2 - e1) > 1; });
// precondition: foo is not empty
if (it == end(foo)) --it;
std::cout << *it+1;
}
编辑:好的,如果你认为Boost足够标准,你可以做到这一点,这真是太棒了:
#include <algorithm>
#include <boost/iterator/counting_iterator.hpp>
#include <set>
#include <iostream>
int main()
{
std::set<int> foo{0, 1, 2, 4};
auto it =
std::mismatch(
begin(foo), end(foo),
boost::counting_iterator<int>(*begin(foo))
);
std::cout << *it.second;
}
编辑2:我在阅读另一个问题时想到的另一个问题:
int i = 0;
std::find_if(begin(foo), end(foo),
[&i](int e){ return e != i++; });
std::cout << i;
这只是带有mismatch
的另一种方式。
答案 2 :(得分:0)
你正在打一个边缘案件。一旦i ==位置在集合的末尾,你的while循环失败。在这种情况下,它以i == 3结束。你需要让我走过数组的边界才能使这个工作。
您可以通过将第2行更改为:
来完成此操作while (++i **<**= prev(foo.end()) && *prev(i) + 1 == *i);
通过使它&lt; =,一旦超过foo结束时的值,我将失败。
以下是其他一些需要考虑的事项: 1)不保证对集合进行排序。 2)set foo(0,1,1)的情况会发生什么?即复制失败的那个是正确的,但它也是集合末尾的那个?
你需要一个稍微复杂的算法来捕捉所有这些情况。