Та же логика работает для C ++, но не для Python для максимального в стеке, есть ли что-то, что мне не хватает в моем коде - PullRequest
3 голосов
/ 23 марта 2019

Я написал ту же логику на python и c ++ для возврата максимального элемента в стеке за O (1) времени, используя два стека. но когда я представил его на hackerrank, он показывает неправильный ответ для python, но принимает c ++. Я что-то упустил в Python.

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n,q,x;
    stack<int>s1,s2;
    cin>>n;
    for(int i = 0;i<n;i++)
    {
        cin>>q;//here q is a type of query
        switch(q)
        {
            //push in stack
            case 1:
                cin>>x;
                if (s1.empty())
                {
                  s2.push(x);
                }
                else 
                {
                  if (x >= s2.top())
                  {
                    s2.push(x);
                  }
                }
                s1.push(x);
                break;
            //pop from stack
            case 2:
                 if(!s1.empty())
                {
                    if(s1.top()==s2.top())
                    {
                        s2.pop();
                    }
                    s1.pop();
                }
                break;
            //getMax from stack
            case 3:
                if(!s2.empty())
                    cout<<s2.top()<<endl;
        }
    }   
    return 0;
}
stack1 = stack2 = []
N = int(input())
for i in range(N):
    a = list(map(int,input().rstrip().split()))
    if a[0]==1:
        if stack1 == []:
            stack2.append(a[1])
        elif a[1]>=stack2[-1]:
            stack2.append(a[1])
        stack1.append(a[1])
    elif a[0]==2:
        if stack1 != []:
            if stack1[-1] == stack2[-1]:
                stack2.pop()
            stack1.pop()
    elif a[0] == 3:
        if stack2 != []:
            print(stack2[-1])

Мне это кажется одинаковым.

Я пробовал несколько собственных тестов на другом онлайн-компиляторе, они работали одинаково для обоих. Должен ли я использовать LIFO из модуля очереди в Python, но до сих пор я не сталкивался с какими-либо проблемами при использовании списков в качестве стека.

Они оба должны работать одинаково для всех тестовых случаев.

1 Ответ

6 голосов
/ 23 марта 2019

У вас есть только один «стек» питона:

stack1 = stack2 = []     # two names that point to the same list

print(id(stack1))
print(id(stack2))

# vs

stack1 = []              # points to one list
stack2 = []              # points to another list

print(id(stack1))
print(id(stack2))

Выход:

139948335562312
139948335562312

# vs

139948335067208
139948335562696
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...