C++求矩陣最小路徑和進階方法空間複雜度O(min {row, col})
阿新 • • 發佈:2019-02-06
#include <iostream>
int minPathSum_pro(const int(*pArr) [10], const int &row, const int &col);
int main()
{
int row = 0, col = 0, num = 0;
int arr[10][10];
std::cin >> row >> col;
for (int i = 0; i < row; ++i)
{
for (int j = 0; j < col; ++j)
{
std::cin >> num;
arr[i][j] = num;
}
}
std::cout << minPathSum_pro(arr, row, col);
system("pause");
return 0;
}
int minPathSum_pro(const int(*pArr)[10], const int &row, const int &col)
{
int dim = row <= col ? row : col;//dimension 維度
int large = dim == row ? col : row;
int *pArray = new int[dim];
int num = 0; //用於返回最終的最小路徑值
pArray[0] = pArr[0][0];
int flag = -1;
flag = dim == row ? 0 : 1; //flag = 0表示row <= col flag = 1 row > col
pArray[0] = pArr[0][0];
for (int i = 1; i < dim; ++i)//初始化第一維陣列,用來進行之後的翻滾
{
pArray[i] = pArray[i - 1] + (flag == 0 ? pArr[0][i] : pArr[i][0]);
}
for (int i = 1; i < dim; ++i)
{
pArray[0] = pArray[0] + (flag == 0 ? pArr[i][0] : pArr[0][i]);
for (int j = 1; j < large; ++j)
{
pArray[j] = (pArray[j - 1] < pArray[j] ? pArray[j - 1] : pArray[j]) + (flag == 0 ? pArr[i][j] : pArr[j][i]);
}
}
num = pArray[dim - 1];
delete[] pArray;
return num;
}
int minPathSum_pro(const int(*pArr) [10], const int &row, const int &col);
int main()
{
int row = 0, col = 0, num = 0;
int arr[10][10];
std::cin >> row >> col;
for (int i = 0; i < row; ++i)
{
for (int j = 0; j < col; ++j)
{
std::cin >> num;
arr[i][j] = num;
}
}
std::cout << minPathSum_pro(arr, row, col);
system("pause");
return 0;
}
int minPathSum_pro(const int(*pArr)[10], const int &row, const int &col)
{
int dim = row <= col ? row : col;//dimension 維度
int large = dim == row ? col : row;
int *pArray = new int[dim];
int num = 0; //用於返回最終的最小路徑值
pArray[0] = pArr[0][0];
int flag = -1;
flag = dim == row ? 0 : 1; //flag = 0表示row <= col flag = 1 row > col
pArray[0] = pArr[0][0];
for (int i = 1; i < dim; ++i)//初始化第一維陣列,用來進行之後的翻滾
{
pArray[i] = pArray[i - 1] + (flag == 0 ? pArr[0][i] : pArr[i][0]);
}
for (int i = 1; i < dim; ++i)
{
pArray[0] = pArray[0] + (flag == 0 ? pArr[i][0] : pArr[0][i]);
for (int j = 1; j < large; ++j)
{
pArray[j] = (pArray[j - 1] < pArray[j] ? pArray[j - 1] : pArray[j]) + (flag == 0 ? pArr[i][j] : pArr[j][i]);
}
}
num = pArray[dim - 1];
delete[] pArray;
return num;
}