(SCU - 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角度考虑,可设f[i][j]表示从三角形顶部到第i行第j列这个点所经过数字总和的最大值,那么答案就是max{f[n][1],f[n][2],…,f[n][n-1],f[n][n]}。易得状态转移方程为:
f[i][j]=max{f[i-1][j-1],f[i-1][j]}+a[i][j] (其中初值为f[1][1]=a[1][1]) 当然这里还要考虑i,j的位置关系的限制导致转移方程的不同,当(j==1)时,f[i][j]=f[i-1][j]+a[i][j];当(j==i)时,f[i][j]==f[i-1][j-1]+a[i][j]。弄清楚这些就不难解决这个问题了。
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=105;
int a[maxn][maxn],f[maxn][maxn];
int ans[maxn],n;//ans数组用来记录每一行所经过的那个数
void dp()
{
f[1][1]=a[1][1];
for(int i=2;i<=n;i++)
for(int j=1;j<=i;j++)
{
if(j==1) f[i][j]=f[i-1][j]+a[i][j];
else if(j==i) f[i][j]=f[i-1][j-1]+a[i][j];
else f[i][j]=max(f[i-1][j-1],f[i-1][j])+a[i][j];
}
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
scanf("%d",&a[i][j]);
dp();
int tmp,sum=f[n][1];
for(int i=2;i<=n;i++)
if(sum<f[n][i])
{
sum=f[n][i];
tmp=i;
}
ans[n]=a[n][tmp];
for(int i=n-1;i>=1;i--)
{
if(tmp==1) ans[i]=a[i][1];
else if(tmp==i)
{
ans[i]=a[i][i-1];
tmp--;
}
else if(f[i][tmp]>f[i][tmp-1]) ans[i]=a[i][tmp];
else if(f[i][tmp]<f[i][tmp-1])
{
ans[i]=a[i][tmp-1];
tmp--;
}
}
printf("%d\n",sum);
printf("%d",ans[1]);
for(int i=2;i<=n;i++) printf(" %d",ans[i]);
printf("\n");
}
return 0;
}