-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathPredict_The_Winner.java
More file actions
60 lines (45 loc) · 2.56 KB
/
Predict_The_Winner.java
File metadata and controls
60 lines (45 loc) · 2.56 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
// You are given an integer array nums. Two players are playing a game with this array: player 1 and player 2.
// Player 1 and player 2 take turns, with player 1 starting first. Both players start the game with a score of 0. At each turn, the player takes one of the numbers from either end of the array (i.e., nums[0] or nums[nums.length - 1]) which reduces the size of the array by 1. The player adds the chosen number to their score. The game ends when there are no more elements in the array.
// Return true if Player 1 can win the game. If the scores of both players are equal, then player 1 is still the winner, and you should also return true. You may assume that both players are playing optimally.
// Example 1:
// Input: nums = [1,5,2]
// Output: false
// Explanation: Initially, player 1 can choose between 1 and 2.
// If he chooses 2 (or 1), then player 2 can choose from 1 (or 2) and 5. If player 2 chooses 5, then player 1 will be left with 1 (or 2).
// So, final score of player 1 is 1 + 2 = 3, and player 2 is 5.
// Hence, player 1 will never be the winner and you need to return false.
// Example 2:
// Input: nums = [1,5,233,7]
// Output: true
// Explanation: Player 1 first chooses 1. Then player 2 has to choose between 5 and 7. No matter which number player 2 choose, player 1 can choose 233.
// Finally, player 1 has more score (234) than player 2 (12), so you need to return True representing player1 can win.
class Solution {
public boolean PredictTheWinner(int[] nums) {
int n = nums.length;
int[][] memo = new int[n][n];
int totalSum = 0;
for (int num : nums) {
totalSum += num;
}
// Calculate the score difference for subarrays of length 1
for (int i = 0; i < n; i++) {
memo[i][i] = nums[i];
}
// Calculate the score difference for subarrays of length 2
for (int i = 0; i < n - 1; i++) {
memo[i][i + 1] = Math.max(nums[i], nums[i + 1]);
}
// Calculate the score difference for subarrays of length >= 3
for (int len = 3; len <= n; len++) {
for (int i = 0; i <= n - len; i++) {
int j = i + len - 1;
int leftSum = nums[i] + Math.min(memo[i + 2][j], memo[i + 1][j - 1]);
int rightSum = nums[j] + Math.min(memo[i + 1][j - 1], memo[i][j - 2]);
memo[i][j] = Math.max(leftSum, rightSum);
}
}
int player1Score = memo[0][n - 1];
int player2Score = totalSum - player1Score;
return player1Score >= player2Score;
}
}