这思路写得比较冗余,有兴趣可以优化一下
#include <iostream>
using namespace std;
#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<string.h>
#include<math.h>
#include <algorithm>
#include<cstdlib>
#include<vector>
// 到了(0,0),计算当前路径是否为答案
void countAns(vector<int> pathList, vector<int> &ansList) {
//算pathlist的和
int newSum = 0;
for (int i = 0; i < pathList.size(); i++) {
newSum += pathList[i];
}
//算anslist的和
int oldSum = 0;
for (int i = 0; i < ansList.size(); i++) {
oldSum += ansList[i];
}
//如果当前路径和大于等于原和,替换
if (oldSum <= newSum) {
ansList.clear();
for (int i = pathList.size() - 1; i >= 0; i--) {
ansList.push_back(pathList[i]);
}
}
}
void recursion(vector<vector<int>> nums, int i, int j, vector<int> pathList, vector<int> &ansList) {
// 如果是0,0 说明到了终点,算一下该点是否要替换答案
if (i == 0 && j == 0) {
countAns(pathList, ansList);
return;
}
// 如果j!=0,就是不等于第一列
if (j != 0) {
// 路径加上左上节点
pathList.push_back(nums[i - 1][j - 1]);
// 往左上走
recursion(nums, i - 1, j - 1, pathList, ansList);
// 弹出,回溯
pathList.pop_back();
}
// 如果j!=i,也就是不等于每行的最后一列
if (j != i) {
// 路径加上,上面一节点
pathList.push_back(nums[i - 1][j]);
// 往上面的节点走
recursion(nums, i - 1, j, pathList, ansList);
// 回溯
pathList.pop_back();
}
}
int main()
{
int n;
cin >> n ;
//二维数组
vector<vector<int>> nums (n,vector<int>(n,0));
for (int i = 0; i < n; i++) {
for (int j = 0; j <= i; j++) {
cin>> nums[i][j];
}
}
vector<int> pathList;
vector<int> ansList;
// 从最底一层出发,结果可能是到达最底一层的任何一个节点,所以要全试一次判断
for (int i = 0; i < n; i++) {
// 终点是n-1,i 走这个节点
pathList.push_back(nums[n - 1][i]);
recursion(nums, n - 1, i, pathList, ansList);
// 回溯
pathList.pop_back();
}
int sum = 0;
for (int i = 0; i < ansList.size(); i++) {
// 找出答案路径和
sum += ansList[i];
}
cout << sum << endl;
for (int i = 0; i < ansList.size(); i++) {
cout << ansList[i] << " ";
}
return 0;
}