先程の方のをプログラムにするとこんな感じです。