【Python】引数の個数が変わる関数の入力方法がわからない

Question

AOJ Doubly Linked Listについての質問です。
この問題を解くために以下の関数を作りました。

from collections import deque
ans = deque()

def doubly_linked_list(operation, value=None):
    if operation == "insert":
            ans.appendleft(value)
    elif operation == "delete":
        try:
            ans.remove(value)
        except:
            pass
    elif operation == "deleteFirst":
        ans.popleft()
    elif operation == "deleteLast":
        ans.pop()

関数は作れたのですが、指定された入力にうまく関数を合わせられません。
入力は以下の通りです。

9
insert 5
insert 2
insert 3
insert 1
delete 3
insert 6
delete 5
deleteFirst
deleteLast

最初の9は入力されるコマンド(operation, value) の数です。
この回数分forを回そうと思ったのですが、

• 入力が insert 5 のように来るのでmapで受け取ろうとした
• しかし、deleteFirst, deleteLast が来るとエラーになる
• つまり、 insert 5 の場合

operation, value = map(input().split())
doubly_linked_list(operation, value)

とできるが、deleteFirstの時にどう記述すればいいか分からない

という感じで手が止まってしまいました。

どなたかうまく入力できる方法を教えていただけないでしょうか。

Answer 1

作った関数とそれを使って処理をする方法とに食い違いがあるようです。

関数の引数は、insert、delete、deleteFirst、deleteLast のいずれかが入るoperationと、何からの値が入るvalueがあり、valueは与えられなければNoneが入るようになっています。

なので、引数に値が1つか 2つのリストを渡してやればいいのです。

doubly_linked_list(*input_list)

ところが、

入力が insert 5 のように来るのでmapで受け取ろうとした

のでは、食い違っています。。

Comments

[cbb4]

回答ありがとうございます。
doubly_linked_list(operation, *value)とすることで実現できました。

Answer 2

ls = input().split()

などと一旦リストで受けて、lsの長さに応じてvalueの値を
ls[1]かNoneにすれば良いでしょう。

Comments

[cbb4]

回答ありがとうございます。
doubly_linked_list(operation, *value)とすることで実現できたのですが、ozwkさんの方法でもできると思うのでベストアンサーとさせていただきます。