何も考えずに返してた

そうなるね、ごめん
(1)でＡの要素数えてるから(2)で同じ要領でＢの要素数出して解いてもいいね