递归+优化 一维数组实现矩阵乘法

1.

计算T(n)队列,规则是T(n) = T(n - 1) + T(n - 2) + T(n -3),其中T(0) = T(1) = 1,T(2) = 2。 

函数定义: 
int Tribonaci(int n) { 

备注,不考虑证整数溢出,尽可能优化算法。 

  这一题我一看就知道要考什么,很显然的递归定义,但也是很显然的,这里所谓的优化是指不要重复计算。 

  简单的说,在计算T(n)的时候要用到T(n - 1)、T(n - 2)和T(n - 3)的结果,在计算T(n - 1)的时候也要用到T(n - 2)和T(n - 3)的结果,所以在各项计算的时候必须把以前计算的结果记录下来,去掉重复计算。这里用到的一点小技巧就是要新写一个函数用来做这种事情,嗯,看看我写的代码吧! 

/** 
  Get the value of T(n - 1), and retrieve the result of 
  T(n - 2) and T(n - 3). 
  @param[in] n The n in T(n). 
  @param[out] mid Value of T(n - 2). 
  @param[out] right Value of T(n - 3). 
  @return Value of T(n - 1). 
*/ 
int find_trib(int n, int & mid, int & right) 

    if (3 == n) 
    { 
        mid = 1; 
        right = 1; 
        return 2; 
    } 
    else 
    { 
        int temp; 
        mid = find_trib(n - 1, right, temp); 
        return mid + right + temp; 
    } 


/** 
  Find value of T(n). 
  @param[in] The n in T(n). 
  @return Value of T(n). 
  @note T(n) = T(n - 1) + T(n - 2) + T(n - 3) (n > 2) 
        T(0) = T(1) = 1, T(2) = 2. 
*/ 
int tribonaci(int n) 

    if (n < 0) 
    { 
        // Undefined feature. 
        return 0; 
    } 

    if (0 == n || 1 == n) 
    { 
        return 1; 
    } 

    if (2 == n) 
    { 
        return 2; 
    } 

    int mid, right; 
    int left = find_trib(n, mid, right); 
    return left + mid + right; 


2.

. 实现两个N*N矩阵的乘法,矩阵由一维数组表示。

void matrix_multiply(int A[], int B[], int result[], int N)
{
    for (int i = 0; i < N; ++i)
    {
        for (int j = 0; j < N; ++j)
        {
            result[i*N+j] = 0;
        }
    }
    for (int i = 0; i < N; ++i)
    {
        for (int j = 0; j < N; ++j)
        {
            for (int k = 0; k < N; ++k)
            {
                result[i*N+j] += A[i*N+k] * B[k*N+j];
            }
        }
    }
}


/* 略微优化 */
void matrix_multiply(int A[], int B[], int result[], int N)
{
    for (int i = 0; i < N * N; i += N)
    {
        for (int j = 0; j < N; ++j)
        {
            result[i+j] = 0;
        }
    }
    for (i = 0; i < N * N; i += N)
    {
        for (j = 0; j < N; ++j)
        {
            int t = 0;
            for (ia = i, ib = j; ib < N * N; ++ia, ib += N)
            {
                t += a[ia] * b[ib];
            }
            result[i+j] = t;
        }
    }
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章