1. 程式人生 > >C++求矩陣最小路徑和進階方法空間複雜度O(min {row, col})

C++求矩陣最小路徑和進階方法空間複雜度O(min {row, col})

#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;
}