C中的稀疏矩阵存储

时间:2010-08-12 18:43:45

标签: c matrix sparse-matrix

我有一个非对称的稀疏矩阵。稀疏性有点随机,我不能指望距离对角线一定距离的所有值。

然而,它仍然稀疏,我想减少矩阵的存储要求。因此,我试图找出如何按顺序存储从第一个非零开始的每一行,直到我到达最后一个非零。

也就是说,如果行m的第一个非零出现在第2列,而最后一个非零出现在第89列,我想存储在A [m]行2->中。 89。

由于每行没有相同数量的非零,我将使A的所有行具有相同数量的元素,并且对于具有较小数量的非行的行,将零填充到行的末尾 - 零素。

如何在C中进行此翻译?我实际上没有原始的完整矩阵来复制值(原始矩阵以CSR形式出现在我身上)。如果我在fortran中这样做,我可以将我的数组定义为二维,并且通过跟踪非零列的开始/停止值并将其存储为每个行,每行都是可变长度。

我将尝试演示如下:

这是我知道的值的矩阵表示 - 对于每个值,我知道行和列的位置

  [1    2    3    4                   ]
  [   5    6    7    8                ]
  [       10    11    12    13        ]
 m[   14    15    16    17       18   ]
  [         19    20    21         22 ]

现在这一行m在第一个非零和最后一个非零之间具有最大的“跨度” 所以我的新矩阵将是5x[span of row m]

  [1     2     3     4          ]
  [5     6     7     8          ]
  [10    11    12    13         ]
 m[14    15    16    17       18]
  [19    20    21       22      ] 

正如您所看到的,行m不需要填充零,因为它是最长的“跨度”

其他行现在都将第0行作为第一个非零值,并在每个非零之间保持零列的间距。

3 个答案:

答案 0 :(得分:3)

我将它实现为一个参差不齐的数组,A [n] [0]总是返回对角线上的元素。 A [n] [1]将项目返回到对角线的右侧,A [n] [2]将项目返回到对角线的左侧,等等。然后,你只需要一个将矩阵索引[i,j]映射到不规则数组索引[r] [s]的函数。

这具有稀疏性的优点,如果您的值保持接近对角线,则阵列不会很长。


或者,你可以有这个定义:

struct Row
{
    int InitialOffset;
    int NumElements;
    int[] Values;
}

然后你会有一个Row []。检索基于矩阵索引的值如下所示:

//matrix is merely an array of rows...
int GetValue(*matrix this, int i, int j)
{
    Row CurrentRow = (*this)[i];
    if (CurrentRow.InitialOffset > j)
        return 0;
    else if (CurrentRow.InitialOffset + CurrentRow.NumElements < j)
        return 0; 
    return CurrentRow.Values[j - CurrentRow.InitialOffset]
}

我的C语法有点朦胧,但你应该明白这一点。


根据您的演示,我建议:

struct Matrix
{
    int[,] Data
    int[] StartOffset;
    int[] NumberElements;
}

使用如下......

int GetValue(*Matrix this, int i, int j)
{
    if (this.StartOffset[i] > j)
        return 0;
    else if (this.StartOffset[i] + this.NumberElements[i] < j)
        return 0; 
    return this.Data[i, j-this.StartOffset[i]];
}

您的初始化过程看起来像这样

//Data is a struct that holds row index, col index, and value
Matrix* InitMatrix (*Data values, int numVals)
{
    //loop through values to find longest row and number of rows
    //create new matrix, malloc matrix for longrow * numRows
    //malloc numrows elements for StartOffset and NumItems
    //foreach row, find min() and max()-min() of col indexs and 
    //store in StartOffset and NumItems
}

您需要进行一些处理,但数据压缩并不便宜。

答案 1 :(得分:2)

另一种方法是使用链接结构(如果矩阵非常稀疏,那么效率非常高,不如填充更多)。 I hinted at the implementation in a earlier answer

我将继续执行连续运行,我不确定您是否真的需要/需要使用等长行。为什么不使用参差不齐的数组?

答案 2 :(得分:1)

Derek,您在其中一条评论中提到过要使用单个malloc。这意味着您知道您有多少非空元素。鉴于此,tt可以将稀疏矩阵存储在一个数组中,该数组将每个元素保持矩阵元素的值和“位置增量”到下一个元素。类似的东西:

struct melem {
    int value; // value of data
    int offset; // offset to next element
}

struct melem matrix[num_nonempty_elements];

...

// Note: this is pseudocode!
matrix[row*COLS + col].value = a[row][col];
matrix[row*COLS + col].offset = (row*COLS + col)_[i] - (row*COLS + col)_[i-1];

编辑:考虑一下,这与链表方法非常相似,但需要1次分配。 OTOH,可能需要更多计算才能访问所需的单元格。