山野くんは電車のnow両目に乗っていると、トイレに行きたくなってきたよ!
トイレがある場所を『1』、トイレがない場所を『0』とした車両と車両の間のリストがvehiclesのようになっているよ!
一番近くのトイレに行くには『前』と『後ろ』のどちらに移動して、自分から何番目の車両の間に行く必要があるでしょうか?
出力例を参考に表示してみよう!
『前』と『後ろ』のどちらに行っても距離が変わらないなら、自分から何番目の車両の間だけ表示しよう!
電車は8両編成だよ!
1 <= now <= 8
必ず1つはトイレがあるよ!
1 in vehicles
vehicles[0] は1両目と2両目の間だよ!
vehicles[6] は7両目と8両目の間だよ!
len(vehicles) == 7
now = 5 # 今いる車両
vehicles = [1, 0, 0, 1, 0, 1, 0] # トイレのある位置
前に1番目
now = 4 # 今いる車両
vehicles = [1, 0, 0, 0, 1, 0, 1] # トイレのある位置
後ろに2番目
now = 4 # 今いる車両
vehicles = [1, 1, 0, 0, 1, 0, 1] # トイレのある位置
2番目
トイレまでの距離が変わらないから、何番目かだけ表示しているよ!