CSC 转 CSR 的C++代码:

文章展示了C++中的模板函数`printfVector`以及`csc2csr`函数,用于处理向量数据结构,涉及CSC和CSR矩阵转换,展示了引用传递以避免拷贝操作。
template <typename T>
void printfVector(vector<T>&vec) {
  printf("输出:\n");
  for(int i=0;i<vec.size();i++)
  {
    printf("%ld ",vec[i]);
  }
  printf("\n");
}

// 如果函数需要修改 vector 的内容或者需要避免不必要的拷贝操作,可以使用引用传递
void csc2csr(int v_size, int e_size, vector<int>& cscColumnOffset, vector<int>& cscRowIndices, vector<int>& csrRowOffset, vector<int>& csrColumnIndices)
{
    csrRowOffset = vector<int>(v_size+1, 0);
    csrColumnIndices = vector<int>(e_size, 0);

    printfVector(cscColumnOffset);
    printfVector(cscRowIndices);
    printfVector(csrRowOffset);
    printfVector(csrColumnIndices);

    cout<<cscRowIndices[3]<<endl;

    for(int i = 0; i < e_size; i++){
        csrRowOffset[cscRowIndices[i] + 1]++;
    }


    for (int i = 1; i <= v_size; ++i) {
      csrRowOffset[i] += csrRowOffset[i - 1];
    }



    for (int i = 0; i < v_size; ++i) {
      for (int j = cscColumnOffset[i]; j < cscColumnOffset[i + 1]; ++j) {
        int local_src = cscRowIndices[j];
        csrColumnIndices[csrRowOffset[local_src]++] = i;
        cout<<csrColumnIndices[csrRowOffset[local_src]]<< endl;
      }
    }

    for (int i = v_size; i > 0; --i) {
      csrRowOffset[i] = csrRowOffset[i - 1];
    }
    csrRowOffset[0] = 0;

    printfVector(cscColumnOffset);
    printfVector(cscRowIndices);
    printfVector(csrRowOffset);
    printfVector(csrColumnIndices);
}

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值