ボールが円管内で衝突する回数を調べるPythonプログラム

円管にn個のボールがあるとします。チューブの長さは100メートルで、最初は、チューブ内の各ボールは、開始点と呼ばれる点から1メートル離れています。これで、ボールはチューブ内をさまざまな方向に循環して移動し始めます。ボールはチューブ内を毎秒0.1メートル移動します。 2つのボールが一点で出会うと、衝突が発生し、ボールの進行方向が変わります。このプロセスが長時間続く場合は、10 ^ 9+6秒としましょう。ボールが衝突した回数を調べる必要があります。開始点からのボールの初期距離が入力として与えられます。

したがって、入力がinput_array =[0、10]のような場合、出力は400000

ボールは2つあり、スタートラインからの初期距離が入力として与えられます。方向が同じであれば、衝突することはありません。しかし、それらの方向が異なる場合、それらは時々衝突します。 1つのボールが別のボールと正確に400000回衝突します。

これを解決するには、次の手順に従います-

• リストinput_arrayを並べ替える
• size：=input_arrayのサイズ
• lap_count：=（10 ^ 5）* 2
• output：=2 * lap_count *（size / 2）のフロア値*（size-（size / 2）のフロア値）
• 停止：=0
• 0からサイズ-1の範囲のiの場合、実行します
• 停止が1と同じでない場合、
  • input_array [i]+1がinput_array[i+ 1]と同じ場合、
    • 出力：=出力+ 2
    • 停止：=1
  • それ以外の場合、
    • 停止：=0
• それ以外の場合、
  • 停止：=0
• 出力を返す

例

理解を深めるために、次の実装を見てみましょう-

def solve(input_array):
   input_array.sort()
   size = len(input_array)
   lap_count = (10**5)*2
   output = 2*lap_count*(size//2)*(size - size//2)
   stop = 0
   for i in range(size - 1):
      if stop != 1:
         if input_array[i] + 1 == input_array[i+1]:
            output+=2
            stop = 1
         else:
            stop = 0
      else:
         stop = 0
   return output
print(solve([0, 10]))

入力

[0, 10]

出力

400000