SCU 1114 数字三角 dp

http://acm.scu.edu.cn/soj/problem.action?id=1114

下图是个数字三角，请编写一个程序计算从顶部至底部某处一条路径，使得该路径所经过的数字总和最大。

7

3  8

8  1  0

2  7  4  4

 

1．  每一步可沿左斜线向下或右斜线向下走；

2．  1<=三角形行数<=100

3．  三角形中的数字为整数 0，1，……，99。

4．  如果有多种情况结果都最大，任意输出一种即可。

 

输入：

第一行一个整数N，代表三角形的行数。

接下来N行，描述了一个数字三角。

 

输出：

    第一行一个整数，代表路径所经过底数字总和。

    第二行N个数，代表所经过的数字。

 

样例：

输入：

4
7
3 8
8 1 0
2 7 4 4

输出：

25
7 3 8 7

思路：从倒数第二层向上dp，转移方程：dp[i][j]=max(dp[i+1][j],dp[i+1][j+1])。最后一层要赋初始值即可，dp[n][j]=a[n][j]，然后因为这道题还要打印路径，因此还需要一个path[maxn][maxn]数组，path[i][j]存储在第i+1行选择的元素的位置。具体看代码。

#include<iostream>
#include<cstdio>
#include<stack>
#include<cmath>
#include<cstring>
#include<queue>
#include<map>
#include<set>
#include<algorithm>
#include<iterator>
#define INF 0x3f3f3f3f
#define EPS 1e-10
typedef long long ll;
typedef unsigned long long ull;
using namespace std;

int dp[105][105];
int a[105][105];
int path[105][105];

int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=i;j++)
		{
			scanf("%d",&a[i][j]);
			if(i==n)
				dp[i][j]=a[i][j];
		}
	}
	for(int i=n-1;i>=1;i--)
	{
		for(int j=1;j<=i;j++)
		{
			if(dp[i+1][j]>dp[i+1][j+1])
			{
				dp[i][j]=dp[i+1][j]+a[i][j];
				path[i][j]=j;
			}
			else
			{
				dp[i][j]=dp[i+1][j+1]+a[i][j];
				path[i][j]=j+1;
			}
		}
	}
	printf("%d\n",dp[1][1]);
	printf("%d",a[1][1]);
	int cur=path[1][1];
	for(int i=1;i<=n-1;i++)
	{
		printf(" %d",a[i+1][path[i][cur]]);
		cur=path[i][cur];
	}
	return 0;
}