Pythonで2つのスパースベクトルの内積を見つけるプログラム

2つのリストで表される2つのスパースベクトルがあるとします。 2つのスパースベクトルの内積を返す必要があります。ベクトルはオブジェクトとして表され、リストはオブジェクトのメンバー変数'nums'に格納されます。

したがって、入力がvector1 =[1、0、0、0、1]、vector2 =[0、0、0、1、1]のようである場合、出力は1になります。内積は1 * 0+0です。 * 0 + 0 * 0 + 0 * 1 + 1 * 1=1。

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

• res：=0

• インデックスiごとに、vector2の数の値vを実行します

  • vが0と同じ場合、

    • 続行

  • それ以外の場合、vector1のnums [i]が0と同じ場合、

    • 次のイテレーションに行く

  • それ以外の場合

    • res：=res + v * nums [i] of vector1

• 解像度を返す

例（Python）

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

class Solution:
   def __init__(self, nums):
      self.nums = nums

   def solve(self, vec):
      res = 0
   for i, v in enumerate(vec.nums):
      if v == 0:
         continue
      elif self.nums[i] == 0:
         continue
      else:
         res += v * self.nums[i]
   return res

ob1, ob2 = Solution([1, 0, 0, 0, 1]), Solution([0, 0, 0, 1, 1])
print(ob1.solve(ob2))

入力

[1, 0, 0, 0, 1], [0, 0, 0, 1, 1]

出力

1