そのプログラムでいいと思います。ただし 1 は素数でも合成数でもないので、2以上の自然数を入力させるのがいいでしょう。