遇到简单计数素数小于n问题的问题

时间:2019-06-08 18:34:55

标签: c++ primes heap-corruption

我一直在尝试leetcode问题204(素数小于n的素数),并且继续出现“释放后使用堆”错误。我的解决方案如下。我添加了一些打印语句,以了解发生了什么,但这并没有真正的帮助。

#include <iostream>
#include <vector>

using namespace std;

int countPrimes(int n) {
  if (n < 2) return 0;

  vector<int> primes;
  primes.push_back(2);
  for (int m = 3; m < n; m++) {
      cout << "m = " << m << endl;
      for (auto it = primes.begin(); it < primes.end(); it++) {
         cout << *it << ", ";
      }
      cout << endl;
      bool prime = true;
      for (int d : primes) {
          cout << "d = " << d << endl;
          if (m % d == 0) {
              prime = false;
              break;
          }
      cout << "end m = " << m << endl;
      if (prime) primes.push_back(m);
      } 
  }
  return primes.size();
}

int main() {

  cout << countPrimes(9) << endl;

}

打印输出:

m = 3
2,
d = 2
end m = 3
m = 4
2, 3,
d = 2
m = 5
2, 3,
d = 2
end m = 5
d = 0
exit status -1

任何想法我在做什么错。这让我觉得我对自己的工作有一些根本的误解...

1 个答案:

答案 0 :(得分:1)

问题在这里

bool prime = true;
for (int d : primes) {
    ...
if (prime) primes.push_back(m);
}

如果循环还改变了向量的大小,则不能在向量上使用基于范围的循环。基于范围的for循环取决于向量的迭代器,如果将元素添加到向量,则向量的所有迭代器都会失效。

当然你真的想写这段代码

bool prime = true;
for (int d : primes) {
    ...
}
if (prime)
    primes.push_back(m);

这很好,因为向量是在循环之后而不是在循环期间修改的。

关于缩进代码的课程非常严格。如果始终正确缩进,很容易发现错位的代码行。