配列に答えを予め格納しておいて配列のインデックスと与えられる整数とを結びつける
return ans[seisu];
で一発や