fork download
  1. import numpy as np
  2. import copy
  3.  
  4.  
  5. if __name__ == '__main__':
  6. N = int(input())
  7. action_values = np.array([list(map(int, input().split())) for _ in range(N)]) # [[a1, b1, c1], [a2, b2, c2], ...]
  8. dp = np.zeros((N+1, 3), dtype=int)
  9. # print(dp)
  10. for n in range(N+1):
  11. if n == 0:
  12. dp[n, :] = 0
  13. else:
  14. for i in range(3):
  15. dp[n, i] = max([dp[n-1, k] + action_values[n-1, i] for k in range(3) if k != i])
  16. # print(actions)
  17. # print(dp)
  18. print(max(dp[N]))
Success #stdin #stdout 0.14s 28916KB
stdin
3
10 40 70
20 50 80
30 60 90
stdout
210