Akuna Capital 2025 Quant OA test

Movie Ratings 参考答案

1. def dp_func(ratings):  # all ratings are negative
   
2.     n = len(ratings)
   
3.     if n==1: return 0
   
4.     if n==2: return max(ratings[0], ratings[1])
   
5.     dp = [0] * n
   
6.     dp[0] = ratings[0]
   
7.     dp[1] = ratings[1]
   
8.     for i in range(2, n):
   
9.         dp[i] = max(dp[i-2], dp[i-1]) + ratings[i]
   
10.     return max(dp[n-2], dp[n-1])
    
11. #
    
12. def maximizeRatings(s):
    
13.     ans = 0
    
14.     i = 0
    
15.     n = len(ratings)
    
16.     while i < n:
    
17.         if s[i] < 0 :
    
18.             j = i+1
    
19.             while j<n and s[j] < 0: j += 1
    
20.             ans += dp_func(s[i:j])
    
21.             i = j
    
22.         else:
    
23.             ans += s[i]
    
24.             i += 1
    
25.     return ans
    
26. # Example usage:
    
27. ratings = [-3, 2, 4, -1, -2, -5]  # 4
    
28. print(maximizeRatings(ratings))  # Output: 4

复制代码