遍历uBlas稀疏矩阵中非零元素

Question

遍历uBlas稀疏矩阵中非零元素

7

我有一个包含 O(N) 元素的稀疏矩阵。

boost::numeric::ublas::compressed_matrix<int> adjacency (N, N);

我可以写一个暴力双重循环来在O(N^2)时间内遍历所有条目，如下所示，但这样会太慢。

for(int i=0; i<N; ++i)
   for(int j=0; j<N; ++j)
       std::cout << adjacency(i,j) std::endl;

如何在O(N)时间内只循环非零条目？对于每个非零元素，我想访问其值和索引i,j。

- D R

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Gert · Accepted Answer

你可以在这个常见问题解答中找到答案：如何遍历所有非零元素？在你的情况下，应该是这样的：

typedef boost::numeric::ublas::compressed_matrix<int>::iterator1 it1_t;
typedef boost::numeric::ublas::compressed_matrix<int>::iterator2 it2_t;

for (it1_t it1 = adjacency.begin1(); it1 != adjacency.end1(); it1++)
{
  for (it2_t it2 = it1.begin(); it2 != it1.end(); it2++)
  {
    std::cout << "(" << it2.index1() << "," << it2.index2() << ") = ";
    std::cout << *it2 << std::endl;
  }
}