-
Notifications
You must be signed in to change notification settings - Fork 1
/
948.bag-of-tokens.cpp
108 lines (107 loc) · 2.66 KB
/
948.bag-of-tokens.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/*
* @lc app=leetcode id=948 lang=cpp
*
* [948] Bag of Tokens
*
* https://leetcode.com/problems/bag-of-tokens/description/
*
* algorithms
* Medium (46.34%)
* Likes: 950
* Dislikes: 331
* Total Accepted: 48.4K
* Total Submissions: 101.2K
* Testcase Example: '[100]\n50'
*
* You have an initial power of power, an initial score of 0, and a bag of
* tokens where tokens[i] is the value of the i^th token (0-indexed).
*
* Your goal is to maximize your total score by potentially playing each token
* in one of two ways:
*
*
* If your current power is at least tokens[i], you may play the i^th token
* face up, losing tokens[i] power and gaining 1 score.
* If your current score is at least 1, you may play the i^th token face down,
* gaining tokens[i] power and losing 1 score.
*
*
* Each token may be played at most once and in any order. You do not have to
* play all the tokens.
*
* Return the largest possible score you can achieve after playing any number
* of tokens.
*
*
* Example 1:
*
*
* Input: tokens = [100], power = 50
* Output: 0
* Explanation: Playing the only token in the bag is impossible because you
* either have too little power or too little score.
*
*
* Example 2:
*
*
* Input: tokens = [100,200], power = 150
* Output: 1
* Explanation: Play the 0^th token (100) face up, your power becomes 50 and
* score becomes 1.
* There is no need to play the 1^st token since you cannot play it face up to
* add to your score.
*
*
* Example 3:
*
*
* Input: tokens = [100,200,300,400], power = 200
* Output: 2
* Explanation: Play the tokens in this order to get a score of 2:
* 1. Play the 0^th token (100) face up, your power becomes 100 and score
* becomes 1.
* 2. Play the 3^rd token (400) face down, your power becomes 500 and score
* becomes 0.
* 3. Play the 1^st token (200) face up, your power becomes 300 and score
* becomes 1.
* 4. Play the 2^nd token (300) face up, your power becomes 0 and score becomes
* 2.
*
*
*
* Constraints:
*
*
* 0 <= tokens.length <= 1000
* 0 <= tokens[i], power < 10^4
*
*
*/
// @lc code=start
class Solution
{
public:
uint16_t bagOfTokensScore(vector<int> &tokens, uint16_t power)
{
uint16_t score = 0;
sort(tokens.begin(), tokens.end());
for (int16_t i = 0, j = tokens.size() - 1; i <= j;)
{
if (power >= tokens[i])
{
power -= tokens[i++];
score++;
}
else if (score > 0 && j > i)
{
power += tokens[j--];
score--;
}
else
break;
}
return score;
}
};
// @lc code=end