题目
题目:题目链接
方法一:杨辉三角
这题我其实第一反应是排列组合的数学题,就是一共要走n+m-2次,然后其中要走n-1次右行。也就是组合数C(n-1,n+m-2),当然如果m<n是用C(m-1,n+m-2)的,减少计算量嘛。然后呢组合数就是杨辉三角嘛,这题就算完了。那先打印个杨辉三角吧:
#include<iostream>
#include<vector>
using namespace std;
typedef long long ll;
const int max_size = 200;
vector<vector<ll> > arr(max_size + 5, vector<ll>(max_size + 5, 0));
void YangHuiTri() {
for (int i = 1; i <= max_size; ++i) {
arr[i][1] = arr[i][i] = 1;
}
for (int i = 3; i <= max_size; ++i) {
for (int j = 2; j <= i-1; ++j) {
arr[i][j] = arr[i - 1][j - 1] + arr[i - 1][j];
}
}
}
int main() {
YangHuiTri();
int n =
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= i; ++j) {
cout << arr[i][j] << " ";
}
cout << endl;
}
return 0;
}
根据这个想法可以写出AC代码:
vector<vector<int> > arr = vector<vector<int> >(105, vector<int>(105, 1));
void YangHuiTri(int n) {
for (int i = 3; i <= n; ++i) {
for (int j = 2; j <= i - 1; ++j) {
if (arr[i - 1][j - 1] <= 2e9 - arr[i - 1][j])
arr[i][j] = arr[i - 1][j - 1] + arr[i - 1][j];
else break;
}
}
}
int slove1(int m, int n) {
if (m == 1 || n == 1) return 1;
if (m == 2 || n == 2) return m + n - 2;
if (m < n) swap(m, n);
YangHuiTri(m + n - 1);
return arr[m + n - 1][n];
}
注: if (arr[i - 1][j - 1] <= 2e9 - arr[i - 1]这部分代码很关键,因为n,m最大可以是100,当算到100时哪怕是long long 都不放不下了,而题目限定答案<=2*10^9,所以超过的肯定不是答案。而直接相加判断也会溢出,所以才用减法表示。
方法二:动态规划
后面看答案发现DP,然后我的妈,真蠢。整得那么复杂干嘛啊。其实我们会发现到(m,n)这个位置只能从(m-1,n)或(m,n-1)这两个位置出发。那递推方程不就出来了吗!即:
d
p
[
m
]
[
n
]
=
d
p
[
m
?
1
]
[
n
]
+
d
p
[
m
]
[
n
?
1
]
dp[m][n] = dp[m-1][n] +dp[m][n-1]
dp[m][n]=dp[m?1][n]+dp[m][n?1] 完事了,然后直接两个for循环就可以了。看代码吧:
int slove2(int m, int n) {
int dp[m][n];
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
if (i == 0 || j == 0) dp[i][j] = 1;
else dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
}
}
return dp[m - 1][n - 1];
}
其实知道怎么做了,去实现它还是需要考虑很多细枝末节的问题,并不是知道怎么做就会做了的。我在杨辉三角花了很多时间去找bug,开始由于溢出一直RE,哪怕换成long long 还是一样,最后审题才有了上面的判断语句。 感谢您的查看!加油! 最后附完整代码:
class Solution {
public:
vector<vector<int> > arr = vector<vector<int> >(105, vector<int>(105, 1));
void YangHuiTri(int n) {
for (int i = 3; i <= n; ++i) {
for (int j = 2; j <= i - 1; ++j) {
if (arr[i - 1][j - 1] <= 2e9 - arr[i - 1][j])
arr[i][j] = arr[i - 1][j - 1] + arr[i - 1][j];
else break;
}
}
}
int slove1(int m, int n) {
if (m == 1 || n == 1) return 1;
if (m == 2 || n == 2) return m + n - 2;
if (m < n) swap(m, n);
YangHuiTri(m + n - 1);
return arr[m + n - 1][n];
}
int slove2(int m, int n) {
int dp[m][n];
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
if (i == 0 || j == 0) dp[i][j] = 1;
else dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
}
}
return dp[m - 1][n - 1];
}
int uniquePaths(int m, int n) {
return slove1(m, n);
return slove2(m, n);
}
};
加油!!
|