リストを使用してスタックを実装します.

問題文:プッシュ、ポップ、ピーク、isempty、サイズのメソッドを使用してスタックを実装します.

テストケース

プッシュ

は、空のスタックを押します.

は、非空のスタック

を押します

ポップ

空っぽのスタックをオンにします.

非空のスタックをポップする値

ピーク

空のスタックのPEEK -> None.

非空のスタックを覗く値

isempty

空っぽのスタックの上でisempty -> true

非空のスタックで空です.

サイズ

空のスタックの

サイズ.

非空のスタックのサイズ.値

アルゴリズム

プッシュ

新しい値をリストの末尾に追加します.

ポップ

スタックが空ならば

None

を返す
他の

.

リストの末尾からの要素をポップします.

要素を返します.

ピーク

スタックが空ならば

None

を返す
他の

.

リストの最後の要素を返します.

isempty

スタックが空ならば

真の

を返す
他の

.

虚偽

を返します
サイズ

リストの長さを返します.

時間と空間の複雑さ

プッシュ

時間の複雑さ: O ( 1 )

スペース複雑さ:o ( 1 )

ポップ

時間の複雑さ: O ( 1 )

スペース複雑さ:o ( 1 )

ピーク

時間の複雑さ: O ( 1 )

スペース複雑さ:o ( 1 )

isempty

時間の複雑さ: O ( 1 )

スペース複雑さ:o ( 1 )

サイズ

時間の複雑さ: O ( 1 )

スペース複雑さ:o ( 1 )

コード

class Stack(object):
    def __init__(self):
        self.items = []

    def isEmpty(self):
        return self.items == []

    def push(self, data):
        self.items.append(data)

    def pop(self):
        if self.isEmpty():
            return None
        return self.items.pop()

    def peek(self):
        if self.isEmpty():
            return None
        return self.items[-1]

    def size(self):
        return len(self.items)

単体テスト

import unittest
from stackLists import Stack


class TestStackLists(unittest.TestCase):
    def test_end_to_end(self):
        print('Test: Empty stack')
        stack = Stack()
        self.assertEqual(stack.peek(), None)
        self.assertEqual(stack.pop(), None)

        print('Test: One element')
        stack = Stack()
        stack.push(5)
        self.assertEqual(stack.size(), 1)
        self.assertEqual(stack.pop(), 5)
        self.assertEqual(stack.peek(), None)

        print('Test: More than one element')
        stack = Stack()
        stack.push(1)
        stack.push(2)
        stack.push(3)
        self.assertEqual(stack.size(), 3)
        self.assertEqual(stack.pop(), 3)
        self.assertEqual(stack.peek(), 2)
        self.assertEqual(stack.pop(), 2)
        self.assertEqual(stack.peek(), 1)
        self.assertEqual(stack.isEmpty(), False)
        self.assertEqual(stack.pop(), 1)
        self.assertEqual(stack.peek(), None)
        self.assertEqual(stack.isEmpty(), True)
        print('Success: test_end_to_end')


def main():
    test = TestStackLists()
    test.test_end_to_end()


if __name__ == '__main__':
    main()

Github repo

ハッピーコーディング!😃