プログラマ脳を鍛える数学パズルをひたすらPythonで：Q3[カードを裏返せ]

Q3:カードを裏返せ

今回もプログラマ脳を鍛える数学パズルの問題をPythonで解いてみました。 ちなみに、こちらのブログでも同じようにPythonで問題を解いてる方がいました！

pythonista.hateblo.jp

目標としては全問Pythonで解いてブログにアップしていきたい。

問題文

1～100までの番号が書かれた100枚のカードが順番に並べられています。 最初、すべてのカードは裏返しの状態で置かれています。 ある人が2番のカードから、１つおきにカードを裏返していきます。 すると、2, 4, 6, … , 100番のカードが表を向いています。 次に、別の人が、3番のカードから2つおきにカードを裏返していきます。 （裏向きのカードは表を向き、表向いているカードは裏返しされます。） また、別の人が、4番のカードから3つおきに、カードを裏返してきます。 このようにn番目のカードからn-1つおきにカードを裏返す操作を、 どのカードの向きも変わらなくなるまで続けたとします。 裏向きになっているカードの番号をすべて求めてください。

Code

まずはrubyでの模範回答

N = 100
cards = Array.new(N, false)

(2..N).each{|i|
    j = i - 1
    while (j < cards.size) do
        cards[j] = !cards[j]
        j += i
    end
}

N.times{|i|
    puts i + 1 if !cards[i]
}

rubyの模範回答をpythonで

N = [True]*100
for i in range(1, 101):
    j = i - 1
    while j < 100:
        N[j] = not N[j]
        j += i

print([k + 1 for k in range(100) if N[k] == False])
#1, 4, 9, 16, 25, 36, 49, 64, 81, 100

まぁそこまで長くなく無難なCodeになったのではないでしょうか？？

rubyのように簡単に

• cards[j] = !cards[j]
• puts i + 1 if !cards[i]

と書くにはどうしたらいいのかとかなり悩みました...。