Description
下图是个数字三角,请编写一个程序计算从顶部至底部某处一条路径,使得该路径所经过的数字总和最大。
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的时候记录
也可以使用结束后dfs……
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
using namespace std;
const int M=105;
int num[M][M];
int dp[M][M];
int ans[M];
int f;
void dfs(int i,int j,int tot,int n)
{
//cout<<i<<" "<<j<<endl;
if(f)
return;
if(i==n)
{
for(int i=1;i<n;i++)
printf("%d ",ans[i]);
printf("%d\n",ans[n]);
f=1;
}
int tmp=tot-num[i][j];
if(tmp==dp[i+1][j])
{
ans[i+1]=num[i+1][j];
dfs(i+1,j,tmp,n);
}
else if(tmp==dp[i+1][j+1])
{
ans[i+1]=num[i+1][j+1];
dfs(i+1,j+1,tmp,n);
}
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
scanf("%d",&num[i][j]);
for(int i=n;i>=1;i--)
for(int j=1;j<=i;j++)
dp[i][j]=max(dp[i+1][j],dp[i+1][j+1])+num[i][j];
printf("%d\n",dp[1][1]);
f=0;
dfs(0,0,dp[1][1],n);
}
return 0;
}