コードの欠陥が分かりません

Question

n,m = input().split()
height = list(input().split())
list = []
for i in range(1,int(n)+1):
    list.append(i)
for i in range(int(m)):
    a,b = input().split()
    a = int(a)
    b = int(b)
    if int(height[a-1]) < int(height[b-1]):
        if int(a) in list:
            list.remove(a)
    elif int(height[a-1]) > int(height[b-1]):
        if int(b) in list:
            list.remove(b)
    elif int(height[a-1]) == int(height[b-1]):
        if int(b) in list:
            list.remove(b)
        elif int(a) in list:
            list.remove(a)
print(len(list))

https://atcoder.jp/contests/abc166/tasks/abc166_c

Atcoderの問題で一つだけWAが出てしまうのですがどこが問題なのか分かりません。
もしわかる方がいらっしゃったら教えていただけると助かります。

Answer 1

とりあえず元のコードが読みづらかったので、同じロジックのまま直しました。

# 修正点
# ・list→lst
# ・読み込み時にintに変換するのが常道なのでそうする
# ・PEP8に準拠させる
# 
# コメント
# ・組み込みの名前と被る変数名を使ってはいけないのは有名な話：https://qiita.com/k-nakamura/items/e43338e0007d79316c02
# ・無駄な記述を減らさないと見通しが悪くなります。型変換とか先に済ませましょう
# ・a-1とか気持ち悪いので何かしらの工夫があるといいのかも
# ・heightも本当はheightsの方が良い気がしますが、そこまでは直していません
# ・このロジックでTLEにならないのかどうかは知らない。速くはなさそう

n, m = map(int, input().split())
height = [int(x) for x in input().split()]
lst = []
for i in range(1, n + 1):
    lst.append(i)

for i in range(m):
    a, b = map(int, input().split())
    if height[a-1] < height[b-1]:
        if a in lst:
            lst.remove(a)
    elif height[a-1] > height[b-1]:
        if b in list:
            lst.remove(b)
    elif height[a-1] == height[b-1]:
        if b in lst:
            lst.remove(b)
        elif a in lst:
            lst.remove(a)

print(len(lst))

あー、これはあれですね、

    elif height[a-1] == height[b-1]:
        if b in lst:
            lst.remove(b)
        elif a in lst:
            lst.remove(a)

aとb両方消さないといけないのに、上が通った時点で下はelifなのでスルーされますね。ifに変えましょう。

Comments

[bluer]

なるほど、elifだとそうなってしまうのですね。
アドバイスまで下さり本当に助かります。
ありがとうございます。