7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
上面的代码块表示一个5行的数字三角形。假设给定一个n行数字三角形,计算出从三角形顶至底的一条路径,使该路径经过的数字总和最大。
每一步只能由当前位置向下或向右下。
对于刚刚学习动态规划的我这题的第一反应就是搜索,然后抱着试一下的心态就进行搜索了一下,代码如下:
#include<bits/stdc++.h>
using namespace std;
int ans = 0, sum = 0, a[101][101], n , cnt;
int x[2] = {1,1}, y[2] = {0,1};
bool v[101][101];
void dfs(int dx, int dy)
{
if(dx == n - 1)
{
ans = max(ans, sum);
if(dy == n-1)
{
cout << ans <<endl;
}
}
else
{
for(int i = 0; i < 2; i++)
{
if(!v[dx + x[i]][dy + y[i]])
{
v[dx + x[i]][dy + y[i]] = 1;
cnt++;
sum += a[dx + x[i]][dy + y[i]];
dfs(dx + x[i],dy + y[i]);
}
v[dx + x[i]][dy + y[i]] = 0;
sum -= a[dx + x[i]][dy + y[i]];
cnt--;
}
}
}
void insert()
{
memset(v,false,sizeof(v));
for(int i = 0; i < n; i++)
{
for(int j = 0; j <= i;j++)
{
cin >> a[i][j];
v[i][j] = 0;
}
}
ans = 0, sum = a[0][0], cnt = 1;
dfs(0,0);
}
int main()
{
int t;
cin >> t;
while(t--)
{
cin >> n;
insert();
}
return 0;
}
测试题目所给的样例,过了,然后就提交,但是提交却 TLE 了,对于这个搜索,慢慢分析发现搜索太多次了,因为有回溯的部分,再加上出口也不是很好找,要搜索到最后一个元素,才找到出口,显然是不合理的。那么换一种思路,现在假设我们已经知道最大值了,位于 n 行,那它必然是
由 n 行的正上方和左上方两个中最大的一个,慢慢的推导,就能发现其实就是在随着不断的输入,我们需要更新状态,代码如下:
#include<bits/stdc++.h>
using namespace std;
int sum[151][151], date[151][151], n, a, ans;
int main()
{
cin >> n;
while(n--)
{
cin >> a;
ans = 0;
for(int i = 0; i < a; i++)
{
for(int j = 0; j <= i; j++)
{
cin >> date[i][j];
if(i == 0)
{
sum[i][j] = date[i][j];
}
else
{
if(j == 0)
{
sum[i][j] = date[i][j] + sum[i - 1][j];
}
else
{
sum[i][j] =date[i][j] + max(sum[i - 1][j-1],sum[i - 1][j]);
}
}
ans = max(ans,sum[i][j]);
}
}
cout << ans <<endl;
}
return 0;
}
这样的话,对于一个元素我们最多利用两次,而搜索随着数据规模的增大,元素的利用次数也在增加,相比搜索动态规划大大的减少了耗时。
文章可能用词不当,或者代码有些不对的地方,希望大佬指正。