Update PrintUtil.hpp

Reverse the stack before printing.
This commit is contained in:
Yudong Jin 2022-11-28 20:51:12 +08:00 committed by GitHub
parent 26f9d6363e
commit a8afc963b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -210,34 +210,32 @@ class PrintUtil {
printTree(root->left, trunk, false);
}
/**
* @brief Get the Stack String object
*
* @tparam T
* @param stack
* @return string
*/
template <typename T>
static string getStackString(stack<T> stack) {
ostringstream s;
if(!stack.empty()){
s << stack.top();
stack.pop();
}
while(!stack.empty()){
s << ", " << stack.top();
stack.pop();
}
return "top->" + s.str() + "]";
}
/**
* @brief Print a stack
*
* @tparam T
* @param stack
* @param stk
*/
template <typename T>
static void printStack(stack<T> &stack) {
cout << getStackString(stack) << '\n';
static void printStack(stack<T> &stk) {
// Reverse the input stack
stack<T> tmp;
while(!stk.empty()) {
tmp.push(stk.top());
stk.pop();
}
// Generate the string to print
ostringstream s;
bool flag = true;
while(!tmp.empty()) {
if (flag) {
s << tmp.top();
flag = false;
}
else s << ", " << tmp.top();
tmp.pop();
}
cout << "[" + s.str() + "]" << '\n';
}
};