AtCoder ABC134_C pythonについて

Question

AtCoder ABC134_Cの解法として以下のコードを見つけたのですが、for文の print(b[(x < b[-1]) - 2])の箇所が理解できません。
どなたか教えていただけないでしょうか。

n = int(input())
a = [int(input()) for _ in range(n)]
b = sorted(a)
for x in a:
    print(b[(x < b[-1]) - 2])

Answer 1

入力が 5 1 9 3 7 5 だったとします。
a = [1, 9, 3, 7, 5]
b = [1, 3, 5, 7, 9]
です。
b[-1] は 9 で、a の中の最大値です。

for文で a の各要素 x を順番に見ていきます。

x が a[1] 以外の時、x は最大値より小さいので、(x < b[-1]) は True です。
- 2 の演算をするとき、True が 1 となり、(x < b[-1]) - 2 は -1 です。
b[-1] は「a から x を除いたもの」の最大値です。

x が a[1] の時、x は最大値なので、(x < b[-1]) は False です。
- 2 の演算をするとき、False が 0 となり、(x < b[-1]) - 2 は -2 です。
b[-2] は最大値 x を除いたものの中で最大の値です。

Comments

[kitsune_42]

n=5ですね。
Trueが1でFalseが0となることを利用しているということですね、理解できました、ありがとうございます！

[kazuma-s]

すみません。入力の値が間違っていたので訂正しました。