>>794のコードを試してみたんですが、入力すると大抵getch()の入力待ちの状態で止まってしまいます。
if文とgetch()の僅かな処理の時間差のせいだと思うんですが、他のやり方ありませんか？