-
Notifications
You must be signed in to change notification settings - Fork 37
/
Copy path486. Predict the Winner.cpp
46 lines (44 loc) · 1.32 KB
/
486. Predict the Winner.cpp
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
class Solution {
public:
bool PredictTheWinner(vector<int>& nums) {
vector<int>score(2);
return solve(nums, score, 0, 0, nums.size() - 1);
}
bool solve(vector<int>& nums, vector<int>& score, int player, int l, int r) {
if (l >= r) {
return score[0] >= score[1];
}
if (player == 0) {
// Pick left
score[0] += nums[l];
if (solve(nums, score, player ^ 1, l + 1, r)) {
score[0] -= nums[l];
return true;
}
score[0] -= nums[l];
// Pick right
score[0] += nums[r];
if (solve(nums, score, player ^ 1, l, r - 1)) {
score[0] -= nums[r];
return true;
}
score[0] -= nums[r];
} else {
// Pick left
score[1] += nums[l];
if (!solve(nums, score, player ^ 1, l + 1, r)) {
score[1] -= nums[l];
return false;
}
score[1] -= nums[l];
// Pick right
score[1] += nums[r];
if (!solve(nums, score, player ^ 1, l, r - 1)) {
score[1] -= nums[r];
return false;
}
score[1] -= nums[r];
}
return player;
}
};