Pythonでkから開始して、リストの最後に到達できることを確認するプログラム

numsと呼ばれる数のリストと別の数kがあるとします。インデックスkと任意のインデックスiから開始する場合、正確にnums[i]ステップ数だけ左または右に進むことができます。リストの最後に到達できるかどうかを確認する必要があります。

したがって、入力がnums =[0、0、2、1、3、3、1、1] k =2の場合、インデックス2から開始してから、インデックス4にジャンプするかのように、出力はTrueになります。次に、最後のインデックス7にジャンプします。

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

• n：=numsのサイズ

• 訪問：=サイズnのリストで0を入力

• tovisit：=サイズ1のリストで、それにkを挿入します

• tovisitのサイズが<0の場合、実行

  • i：=tovisitの最後の要素とtovisitから削除

  • iがn-1と同じ場合、

  • Trueを返す

• visited [i]が1と同じでない場合、

  • 訪問済み[i]：=1

  • up：=i + nums [i]

  • down：=i --nums [i]

  • up

  • tovisitの最後に挿入

  • ダウン>=0の場合、

  • tovisitの最後に挿入します

• Falseを返す

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

例

class Solution:
   def solve(self, nums, k):
      n=len(nums)
      visited = [0]*n
      tovisit = [k]
      while len(tovisit)>0:
         i=tovisit.pop()
         if i==n-1:
            return True
         if visited[i]!=1:
            visited[i]=1
            up=i+nums[i]
            dn=i-nums[i]
         if up=0:
            tovisit.append(dn)
      return False
ob = Solution()
nums = [0, 0, 2, 1, 3, 3, 1, 1]
k = 2
print(ob.solve(nums, k))

入力

[0, 0, 2, 1, 3, 3, 1, 1], 2

出力

True