[英]Check if a stack is palindrome
在最近的一次編碼面試中,我被要求解決一個問題,其中的任務是完成一個函數,該函數通過引用接收堆棧作為參數,並檢查傳遞的堆棧是否為回文。 我確實想出了一個方法,但在我看來這根本不是一個好方法。
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
void copy_it(stack<int> &st, stack<int> &temp) {
if(st.empty())
return;
int element = st.top();
temp.push(element);
st.pop();
copy_it(st, temp);
st.push(element);
}
bool check_palindrome(stack<int> &st, stack<int> &temp) {
if(st.size() != temp.size())
return false;
while(!st.empty()) {
if(st.top() != temp.top())
return false;
st.pop();
temp.pop();
}
return true;
}
int main()
{
vector<int>vec{-1, -2, -3, -3, -2, -1};
stack<int>st;
for(int i = vec.size() - 1; i >= 0; --i) {
st.push(vec[i]);
}
stack<int> temp;
copy_it(st, temp);
cout << check_palindrome(st, temp);
return 0;
}
有一個更好的方法嗎? 我最好尋找遞歸算法/代碼。
一種方法是彈出一半堆棧,推入另一個堆棧並比較它們。 例如:
[A, B, C, B, A] // our stack, where right is top
-> [A, B, C, B], [A] // pop A, push A onto temporary stack
-> [A, B, C], [A, B] // pop B, push B
-> [A, B], [A, B] // pop C, discard C
只有當堆棧大小為奇數時,我們才必須彈出回文的中心 ( C
) 作為最后一步。
bool isPalindrome(std::stack<int> &stack) {
std::stack<int> tmp;
size_t size = stack.size();
size_t halfSize = size / 2;
for (size_t i = 0; i < halfSize; ++i) {
tmp.push(stack.top());
stack.pop();
}
// discard leftover element if the number of original elements is odd
if (size & 1) {
stack.pop();
}
return tmp == s;
}
如果應該恢復原始堆棧的狀態,我們只需要通過從我們的tmp
堆棧中彈出並推回輸入stack
來反轉該過程。 請記住,我們不會丟棄中間元素,而是在再次將其推回之前暫時存儲它。
或者,如果這不被視為“作弊”,我們可以簡單地接受stack
作為值而不是左值引用。 然后我們只需在進行任何更改之前復制它。
// balance() does the job of popping from one stack and pushing onto
// another, but recursively.
// For simplicity, it assumes that a has more elements than b.
void balance(std::stack<int> &a, std::stack<int> &b) {
if (a.size() == b.size()) {
return;
}
if (a.size() > b.size()) {
b.push(a.top());
a.pop();
if (a.size() < b.size()) {
// we have pushed the middle element onto b now
b.pop();
return;
}
}
return balance(a, b);
}
bool isPalindrome(std::stack<int> &stack) {
std::stack<int> tmp;
balance(stack, tmp);
return stack == tmp;
}
bool check_palindrome(std::stack<int> &st) {
std::stack<int> temp;
auto initialSize = st.size();
for(size_t i{}; i < initialSize/2; ++i) { temp.push(st.top()); st.pop(); }
if(temp.size() < st.size()) st.pop();
return st == temp;
}
我認為您在這里不需要遞歸算法,只需將堆棧中的一半元素推入第二個堆棧,然后彈出兩個堆棧的元素並檢查它們是否相同:
int main()
{
vector<int>vec{-1, -2, -3, -3, -2, -1};
stack<int>st;
for(int i = vec.size() - 1; i >= 0; --i) {
st.push(vec[i]);
}
stack<int> temp;
for (size_t i = 0; i < st.size() / 2; i++)
{
temp.push(st.top());
st.pop();
}
if (st.size() != temp.size()) st.pop();
while (!st.empty())
{
if (st.top() != temp.top()) return 1;
st.pop();
temp.pop();
}
return 0;
}
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.