vector容器基本用法:
C++ vector的用法(整理)_一个幽默且帅气的程序员的博客-CSDN博客
vector作为形参:c++中vector做形参传递问题_为初始化的vector不能传参___三思而先行的博客-CSDN博客
//输入参考:1 5 4 0 0 2 0 0 3 8 0 0 7 0 0
//输出参考:1 3 8
#include<iostream>
#include<vector>
using namespace std;
typedef struct Node{
int data;
struct Node *lchild, *rchild;
}*BiTree, BiTNode;
void InitBiTree(BiTree &T){
//先序创建二叉树
int n;
cin>>n;
if(n==0) T=NULL;//空结点输入0
else{
T=new BiTNode;
T->data=n;
InitBiTree(T->lchild);
InitBiTree(T->rchild);
}
return;
}
//vector容器作为辅助栈使用
vector<int> b;
void MaxPath(BiTree T, int sum, vector<int> &a){
static int max=0;//static变量记录路径和最大值
a.push_back(T->data);//结点进入辅助栈
sum+=T->data;//路径和, 累加上当前结点值
//某条路径遍历结束(到达叶子结点), 并且出现了新的最大路径和
if(T->lchild==NULL&&T->rchild==NULL&&max<sum){
max=sum;//更新最大值
b.assign(a.begin(), a.begin()+a.size());//保存最新路径
}
if(T->lchild!=NULL)//递归遍历左子树
MaxPath(T->lchild, sum, a);
if(T->rchild!=NULL)//递归遍历右子树
MaxPath(T->rchild, sum, a);
//每次遍历完成, 结点出栈
a.pop_back();
//通过line30和line45, 实现:所有的路径都曾在a中出现过
//因此, 每当有新的最大路径和时, 可随时将当前路径存入b中
}
int main()
{
BiTree T;
InitBiTree(T);
vector<int> a;
MaxPath(T, 0, a);
//输出最大路径
for(int i=0;i<b.size();i++)
cout<<b[i]<<" ";
return 0;
}