Pythonで指定されたボールを占めるボックスの位置を見つけます

2つの配列AとBがあるとします。Aのサイズは行の数であり、A[i]はi番目の行のボックスの数です。また、Bはボールの配列であり、B[i]はボール上の数字を示します。ボールi（値B [i]）は、開始からの位置がB[i]であるボックスに配置されます。各B[i]に対応するボックスの行と列を見つける必要があります。

したがって、入力がA =[3、4、5、6]、B =[1、3、5、2]の場合、出力は[（1、1）、（1、3）、（ 2、2）、（1、2）] B [0] =1の場合、ボックスの位置は1行目、1列目B [1] =3になり、ボックス位置は1行目、3列目、Bになります。 [2] =5の場合、ボックスの位置は2行、2列、B [3] =2になり、ボックスの位置は1行、2列になります

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

• len_a：=Aのサイズ

• len_b：=Bのサイズ

• 範囲1からlen_aのiの場合、実行します

  • A [i]：=A [i] + A [i-1]

• 0からlen_bの範囲のiの場合、実行

  • row：=Aをソートして維持するためにB[i]を挿入できるインデックス

  • 行>=1の場合、

    • box_num：=B [i] --A [row-1]

  • それ以外の場合

    • box_num：=B [i]

  • ペアを表示する（row + 1、box_num）

例

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

import bisect
def get_position(A, B):
   len_a = len(A)
   len_b = len(B)
   for i in range(1, len_a):
      A[i] += A[i - 1]
   for i in range(len_b):
      row = bisect.bisect_left(A, B[i])
      if row >= 1:
         box_num = B[i] - A[row - 1]
      else:
         box_num = B[i]
      print ((row + 1, box_num))
A = [3, 4, 5, 6]
B = [1, 3, 5, 2]
get_position(A, B)

入力

[3, 4, 5, 6], [1, 3, 5, 2]

出力

(1, 1)
(1, 3)
(2, 2)
(1, 2)