Sponsored
Sponsored
This approach uses Dynamic Programming to determine if Alice can win given n
stones. We can maintain a boolean array dp
where dp[i]
represents whether Alice can win with i
stones. A state is winning if there's any move that leaves the opponent in a losing state.
Time Complexity: O(n*sqrt(n)), Space Complexity: O(n)
1var winnerSquareGame = function(n) {
2 const dp = new Array(n + 1).fill(false);
3 for (let i = 1; i <= n; i++) {
4 for (let k = 1; k * k <= i; k++) {
5 if (!dp[i - k * k]) {
6 dp[i] = true;
7 break;
8 }
9 }
10 }
11 return dp[n];
12};
We use an array dp
to represent if a state is winning. For each number of stones up to n
, we mark it as winning if removing any perfect square results in a losing state for the opponent.
This approach involves a recursive solution with memoization to cache previously computed results. If n
is already solved, we return the cached result. Otherwise, we recursively check if any move leaves the opponent in a losing position.
Time Complexity: O(n*sqrt(n)), Space Complexity: O(n)
1#include <vector>
2#include <cmath>
bool isWinning(int n, std::vector<int>& memo) {
if (n == 0) return false;
if (memo[n] != -1) return memo[n];
for (int k = 1; k * k <= n; ++k) {
if (!isWinning(n - k * k, memo)) {
return memo[n] = true;
}
}
return memo[n] = false;
}
bool winnerSquareGame(int n) {
std::vector<int> memo(n+1, -1);
return isWinning(n, memo);
}
This C++ solution uses a vector to memoize results. For each position, we recursively evaluate if there are moves leading to a losing state for the opponent. If a losing position is found, it is marked as winning in the memo table.