利用动态规划的思想改进杨辉三角，并进一步优化；c++实现

最新推荐文章于 2024-09-07 23:15:53 发布

乱敲的码农

最新推荐文章于 2024-09-07 23:15:53 发布

阅读量551

点赞数 8

分类专栏： c++ 文章标签： c++ 动态规划算法数据结构

本文链接：https://blog.csdn.net/2302_80272644/article/details/140858206

版权

c++ 专栏收录该内容

40 篇文章 0 订阅

订阅专栏

杨辉三角（Pascal's Triangle）是一个经典的数学问题，也可以用递归算法来生成。下面是一个使用C++递归计算杨辉三角指定行列值的代码示例。

普通递归

#include <iostream>
using namespace std;

// 计算杨辉三角中指定行列的值
int pascal(int row, int col) {
    // 递归基例：如果是第一列或对角线上的元素，值为1
    if (col == 0 || col == row) {
        return 1;
    }
    // 递归关系：每个位置的值等于它上面两个位置的值之和
    return pascal(row - 1, col - 1) + pascal(row - 1, col);
}

int main() {
    int rows;
    cout << "请输入杨辉三角的行数: ";
    cin >> rows;

    // 打印杨辉三角
    for (int i = 0; i < rows; ++i) {
        for (int j = 0; j <= i; ++j) {
            cout << pascal(i, j) << " ";
        }
        cout << endl;
    }

    return 0;
}

代码解析：

函数pascal：
- 递归基例：如果col为0（第一列）或col == row（对角线上的元素），返回1。
- 递归关系：每个位置的值等于它上面两个位置的值之和，即pascal(row - 1, col - 1) + pascal(row - 1, col)。
主函数main：
- 用户输入杨辉三角的行数rows，然后通过双重循环调用pascal函数来生成并打印每一行的值。

运行结果示例：

假设输入5行，输出为：

记忆化递归（二维）

使用记忆化递归或动态规划可以显著优化计算杨辉三角的效率，避免重复计算相同的子问题。下面是改进后的C++代码，使用动态规划思想来生成杨辉三角。

#include <iostream>
#include <vector>
using namespace std;

// 计算杨辉三角中指定行列的值，使用动态规划进行优化
int pascal(int row, int col, vector<vector<int>>& dp) {
    // 如果已经计算过该值，直接返回
    if (dp[row][col] != -1) {
        return dp[row][col];
    }

    // 递归基例：如果是第一列或对角线上的元素，值为1
    if (col == 0 || col == row) {
        dp[row][col] = 1;
    } else {
        // 递归关系：每个位置的值等于它上面两个位置的值之和
        dp[row][col] = pascal(row - 1, col - 1, dp) + pascal(row - 1, col, dp);
    }

    return dp[row][col];
}

int main() {
    int rows;
    cout << "请输入杨辉三角的行数: ";
    cin >> rows;

    // 初始化二维向量用于存储已经计算过的值，初始值设为-1表示未计算
    vector<vector<int>> dp(rows, vector<int>(rows, -1));

    // 打印杨辉三角
    for (int i = 0; i < rows; ++i) {
        for (int j = 0; j <= i; ++j) {
            cout << pascal(i, j, dp) << " ";
        }
        cout << endl;
    }

    return 0;
}

代码解析：

动态规划数组dp：
- dp是一个二维向量，大小为rows x rows，初始值设为-1表示该位置尚未计算。
函数pascal：
- 在递归计算之前，首先检查dp[row][col]是否已经计算过。如果已经计算过，直接返回该值，避免重复计算。
- 如果还未计算，则按照递归关系计算，并将结果存储在dp[row][col]中。
主函数main：
- 用户输入行数rows，然后初始化dp数组。
- 通过双重循环调用pascal函数生成并打印杨辉三角。

运行结果示例：

同样的输入5行，输出结果仍然是：

优化效果：

通过引入记忆化（使用dp数组存储已经计算过的值），大大减少了递归的重复计算，提升了算法的效率，尤其在处理较大规模的杨辉三角时效果更为明显。

记忆化递归（一维）

通过只存储当前行和前一行的数据，可以减少空间复杂度，并且在每次计算新的一行时更新数组中的值。下面是使用一维数组进行优化的C++代码：

#include <iostream>
#include <vector>
using namespace std;

int main() {
    int rows;
    cout << "请输入杨辉三角的行数: ";
    cin >> rows;

    // 初始化一维向量用于存储当前行的值
    vector<int> dp(rows, 1);  // 每行的第一个元素始终为1

    // 打印杨辉三角
    for (int i = 0; i < rows; ++i) {
        // 从后往前更新数组中的值，避免覆盖未更新的值
        for (int j = i - 1; j > 0; --j) {
            dp[j] = dp[j] + dp[j - 1];
        }

        // 打印当前行
        for (int j = 0; j <= i; ++j) {
            cout << dp[j] << " ";
        }
        cout << endl;
    }

    return 0;
}

代码解析：

一维向量dp：
- dp数组用于存储杨辉三角当前行的数据，初始化大小为rows，所有元素初始值为1，因为每行的第一个和最后一个元素总是1。
更新dp数组：
- 计算每一行时，从后往前更新dp数组，计算新的dp[j]值时，依赖于前一行的dp[j]和dp[j-1]。这样可以避免覆盖未更新的值。
主函数main：
- 用户输入行数rows，然后通过循环依次计算每一行的值并打印。