1242: 又见数塔
时间限制: 1 Sec 内存限制: 128 MB提交: 10 解决: 5
[提交][状态][论坛]
题目描述
在讲述DP算法的时候,一个经典的例子就是数塔问题,它是这样描述的:
有如下所示的数塔,要求从顶层走到底层,若每一步只能走到相邻的结点,则经过的结点的数字之和最大是多少?并输出得到最大值的路径。
输入
输入数据首先包括一个整数C,表示测试实例的个数,每个测试实例的第一行是一个整数N(1 <= N <= 100),表示数塔的高度,接下来用N行数字表示数塔,其中第i行有个i个整数,且所有的整数均在区间[0,99]内。
输出
对于每个测试实例,输出可能得到的最大和,并输出其路径。如果有多条路径则输出最左边的那条路径。
样例输入
1 5 7 3 8 8 1 0 2 7 4 4 4 5 2 6 5
样例输出
30 (0,0)-->(1,0)-->(2,0)-->(3,1)-->(4,1)
刚刚接触动态规划不久, 做出来了前面的找最大值的问题, 后面的寻找路径没有做出来。。。。
寻找路径大概思路是:dp数组保存了各个状态的值, 减去数塔中的数a数组,找到每个通过判断一步步找出路径。。。
#include<cstdio> #include<algorithm> using namespace std; int main() { int a[105][105], dp[105][105]; int t, i, j; scanf("%d", &t); while(t--){ int n; scanf("%d", &n); for(i = 1; i <= n; i++) for(j = 1; j <= i; j++) scanf("%d", &a[i][j]); for(j = 1; j <= n; j++) dp[n][j] = a[n][j]; for(i = n - 1; i >= 1; i--) for(j = 1; j <= i; j++){ dp[i][j] = a[i][j] + max(dp[i + 1][j], dp[i + 1][j + 1]); } printf("%d\n", dp[1][1]); printf("(0,0)"); //寻找路径 int m = dp[1][1],d = a[1][1],ti=1,tj=1; for(i = 2; i<=n; i++){ m -= d; ti = i; if(m == dp[ti][tj]){ printf("-->(%d,%d)", ti - 1, tj - 1); d = a[ti][tj]; } else if(m == dp[ti][tj+1]){ tj += 1; printf("-->(%d,%d)", ti - 1, tj - 1); d = a[ti][tj]; } } printf("\n"); } return 0; }