1假设有排成一行的N个位置记为1~N,N一定大于或等于2
开始时机器人在其中的M位置上(M一定是1~N中的一个)
如果机器人来到1位置,那么下一步只能往右来到2位置;
如果机器人来到N位置,那么下一步只能往左来到N-1位置;
如果机器人来到中间位置,那么下一步可以往左走或者往右走;
规定机器人必须走K步,最终能来到P位置(P也是1~N中的一个)的方法有多少种
给定四个参数 N、M、K、P,返回方法数
动态规划:如果在调用过程中有重复调用过程就将他存起来
比如递归求斐波那契数列,中间有不少重复过程
递归策略:
#include <vector>
using namespace std;
class Solution {
public:
int ways1(int N, int start, int aim, int K) {
if (N < 2 || start < 1 || start > N || aim < 1 || aim > N || K < 1) {
return -1;
}
return process1(start, K, aim, N);
}
/ 机器人当前来到的位置是cur,
// 机器人还有rest步需要去走,
// 最终的目标是aim,
// 有哪些位置?1~N
// 返回:机器人从cur出发,走过rest步之后,最终停在aim的方法数,是多少?
private:
int process1(int cur, int rest, int aim, int N) {
if (rest == 0) {
return cur == aim ? 1 : 0;
}
if (cur == 1) {
return process1(2, rest - 1, aim, N);
}
if (cur == N) {
return process1(N - 1, rest - 1, aim, N);
}
return process1(cur - 1, rest - 1, aim, N) + process1(cur + 1, rest - 1, aim, N);
}
};
这个递归来进行优化
该递归中出现了重复解
cur范围1到n
rest范围0到k
准备一张dp表,先全设为-1,表示没算过这个过程
#include <vector>
using namespace std;
class Solution {
public:
int ways2(int N, int start, int aim, int K) {
if (N < 2 || start < 1 || start > N || aim < 1 || aim > N || K < 1) {
return -1;
}
vector<vector<int>> dp(N + 1, vector<int>(K + 1, -1));
return process2(start, K, aim, N, dp);
}
private:
int process2(int cur, int rest, int aim, int N, vector<vector<int>>& dp) {
if (dp[cur][rest] != -1) {
return dp[cur][rest];
}//算过的值就塞入缓存表
int ans = 0;
if (rest == 0) {
ans = cur == aim ? 1 : 0;
}
else if (cur == 1) {
ans = process2(2, rest - 1, aim, N, dp);
}
else if (cur == N) {
ans = process2(N - 1, rest - 1, aim, N, dp);
}
else {
ans = process2(cur - 1, rest - 1, aim, N, dp) + process2(cur + 1, rest - 1, aim, N, dp);
}
dp[cur][rest] = ans;//没算过的算出来了也塞入缓存表
return ans;
}
};
再往下优化,直接填这张动态规划表
不要硬憋状态转移,要去想尝试策略
#include <vector>
using namespace std;
class Solution {
public:
int ways3(int N, int start, int aim, int K) {
if (N < 2 || start < 1 || start > N || aim < 1 || aim > N || K < 1) {
return -1;
}
vector<vector<int>> dp(N + 1, vector<int>(K + 1, 0));
dp[aim][0] = 1;
for (int rest = 1; rest <= K; rest++) {
dp[1][rest] = dp[2][rest - 1];
for (int cur = 2; cur < N; cur++) {
dp[cur][rest] = dp[cur - 1][rest - 1] + dp[cur + 1][rest - 1];
}
dp[N][rest] = dp[N - 1][rest - 1];
}
return dp[start][K];
}
};
2给定一个整型数组arr,代表数值不同的纸牌排成一条线
玩家A和玩家B依次拿走每张纸牌
规定玩家A先拿,玩家B后拿
但是每个玩家每次只能拿走最左或最右的纸牌
玩家A和玩家B都绝顶聪明
请返回最后获胜者的分数
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int f1(vector<int>& arr, int L, int R);
int g1(vector<int>& arr, int L, int R);
int win1(vector<int>& arr) {
if (arr.empty()) {
return 0;
}
int first = f1(arr, 0, arr.size() - 1);
int second = g1(arr, 0, arr.size() - 1);
return max(first, second);
}
int f1(vector<int>& arr, int L, int R) {
if (L == R) {
return arr[L];
}
int p1 = arr[L] + g1(arr, L + 1, R);
int p2 = arr[R] + g1(arr, L, R - 1);
return max(p1, p2);
}
int g1(vector<int>& arr, int L, int R) {
if (L == R) {
return 0;
}
int p1 = f1(arr, L + 1, R); // 对手拿走了L位置的数
int p2 = f1(arr, L, R - 1); // 对手拿走了R位置的数
return min(p1, p2);
}
int main() {
vector<int> arr = {3, 9, 1, 2};
cout << win1(arr) << endl; // 示例输入,输出结果根据实际情况可能不同
return 0;
}
弄两张表进行记忆化缓存
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int f2(vector<int>& arr, int L, int R, vector<vector<int>>& fmap, vector<vector<int>>& gmap);
int g2(vector<int>& arr, int L, int R, vector<vector<int>>& fmap, vector<vector<int>>& gmap);
int win2(vector<int>& arr) {
if (arr.empty()) {
return 0;
}
int N = arr.size();
vector<vector<int>> fmap(N, vector<int>(N, -1));
vector<vector<int>> gmap(N, vector<int>(N, -1));
int first = f2(arr, 0, arr.size() - 1, fmap, gmap);
int second = g2(arr, 0, arr.size() - 1, fmap, gmap);
return max(first, second);
}
int f2(vector<int>& arr, int L, int R, vector<vector<int>>& fmap, vector<vector<int>>& gmap) {
if (fmap[L][R] != -1) {
return fmap[L][R];
}
int ans = 0;
if (L == R) {
ans = arr[L];
} else {
int p1 = arr[L] + g2(arr, L + 1, R, fmap, gmap);
int p2 = arr[R] + g2(arr, L, R - 1, fmap, gmap);
ans = max(p1, p2);
}
fmap[L][R] = ans;
return ans;
}
int g2(vector<int>& arr, int L, int R, vector<vector<int>>& fmap, vector<vector<int>>& gmap) {
if (gmap[L][R] != -1) {
return gmap[L][R];
}
int ans = 0;
if (L != R) {
int p1 = f2(arr, L + 1, R, fmap, gmap); // 对手拿走了L位置的数
int p2 = f2(arr, L, R - 1, fmap, gmap); // 对手拿走了R位置的数
ans = min(p1, p2);
}
gmap[L][R] = ans;
return ans;
}
int main() {
vector<int> arr = {3, 9, 1, 2};
cout << win2(arr) << endl; // 示例输入,输出结果根据实际情况可能不同
return 0;
}
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int win3(vector<int>& arr) {
if (arr.empty()) {
return 0;
}
int N = arr.size();
vector<vector<int>> fmap(N, vector<int>(N));
vector<vector<int>> gmap(N, vector<int>(N));
for (int i = 0; i < N; i++) {
fmap[i][i] = arr[i];
}
for (int startCol = 1; startCol < N; startCol++) {
int L = 0;
int R = startCol;
while (R < N) {
fmap[L][R] = max(arr[L] + gmap[L + 1][R], arr[R] + gmap[L][R - 1]);
gmap[L][R] = min(fmap[L + 1][R], fmap[L][R - 1]);
L++;
R++;
}
}
return max(fmap[0][N - 1], gmap[0][N - 1]);
}
int main() {
vector<int> arr = {3, 9, 1, 2};
cout << win3(arr) << endl; // 示例输入,输出结果根据实际情况可能不同
return 0;
}