下三角矩阵-向量积

Question

在一次编程练习中，我得到了一个保存为数组的对称 3x3 矩阵的下三角元素

|1 * *|

|2 4 *| => [1,2,3,4,5,6]

|3 5 6|

我需要制作乘积 C(i)=C(i)+M(i,j)V(j)，其中 M 是对称矩阵，V 是向量。

V =>[A,B,C]
C(1)=1*A + 2*B + 3*C
C(2)=2*A + 4*B + 5*C
C(3)=3*A + 5*B + 6*C

我正在尝试制作一个可以执行此产品的高效算法

我可以轻松生成 C(3) 所需的所有产品 但是，当我尝试生成值 C(1)、C(2) 时遇到问题，我不知道如何解决这个问题无需使用额外的内存。

这就是我所做的

k=6
n=3

  DO 1 j = n,1,-1
    l= k
    DO 2 i = n,j + 1,-1
       C(i) = C(i) + V(j)*M(l)
       l = l - 1           
2   enddo             
    C(j) = V(j)*M(k-n+j)
    k = k - (n-j+1)
1 enddo

我遇到的问题是我无法为 C(1) 生成和添加 2*B，为 C(2) 生成和添加 5*C。练习的目标是使用尽可能少的步骤和尽可能小的数组空间。

有什么建议吗？

Answer 1

您的代码存在多个问题：

• 在外循环中，您分配C(n)（可能是对角线条目），因此根本不使用内循环的计算
• 您正在从后到前循环左下三角形，如果您反转它，矢量化矩阵内的索引会简单得多
• 矩阵内部位置（k和l）的计算错误
• 您不计算镜像元素的乘积

这是我尊重以上几点的解决方案：

  ! Loop over all elements in the lower left triangle
  k = 0
  do j=1,n
    ! Increment the position inside the unrolled matrix
    k = k+1
    ! diagonal entries, i = j
    c(j) = c(j) + v(j)*M(k)

    ! off-diagonal entries
    do i=j+1,n
      ! Increment the position inside the unrolled matrix
      k = k+1
      ! Original entry
      c(i) = c(i) + v(j)*M(k)
      ! Mirrored one
      c(j) = c(j) + v(i)*M(k)
    enddo !i
  enddo !j