sparse稀疏矩阵—COO—排序—最终生成CSR格式 matlab

% 创建一个稀疏矩阵
A =  [1 0 0; 1 2 0; 2 3 0];                     %%@@@@@@@@@@@@@@@@@@@@@@@@@

% 将稀疏矩阵转换为 COO 格式
[rows, cols, values] = find(A);

% 将 COO 格式按行排序后，再在每行中按列排序value
n = size(A, 1);
nnz = length(values);

row_col_varlue=[rows,cols,values];
[sortedValues, sortOrder] = sortrows(row_col_varlue, 1);

% 根据排序顺序重新排列矩阵的行
row_col_varlue = row_col_varlue(sortOrder, :);

% 输出排列后的行、列和value
rows=row_col_varlue(:,1);
cols=row_col_varlue(:,2);
values=row_col_varlue(:,3);


% COO向CSR格式转换
% 计算行偏移值
row_offset = zeros(1, n+1);
count = zeros(1, n);
for i = 1:nnz
    count(rows(i)) = count(rows(i)) + 1;
end
row_offset(1) = 1;
for i = 1:n
    row_offset(i+1) = row_offset(i) + count(i);
end

csr_values=values';
csr_col=cols';


%打印结果
disp('CSR 格式稀疏矩阵：');
disp('非零元素的值：');
disp(csr_values);
disp('列索引：');
disp(csr_col);
disp('行偏移值：');
disp(row_offset);