数字三角形4

Description

一个数字三角宝塔。

设数字三角形中的数字为绝对值不超过1000的整数。

小K从最顶层走到最底层，每一步可向下或右斜线向下走。

每走过一个节点他会把这个节点的数字加在自己计数器中。

另外他最多只能向下走k次。

现在小K想知道他到达底层后，计数器中可能的最大的值。

Format

Input

输入数据的第1行是数字三角形的行数n和能够向下走的次数k

接下来n行是数字三角形各行中的数字。所有数字都小于1000。

1<=n<=1000，0<=k<=100

Output

如题

Samples

输入数据 1

4 2
1
3 2
40 10 1
100 3 2 20

Copy

输出数据 1

47

思路 

首先，我们发现，这道题和数字三角形1 ，数字三角形2，数字三角形3十分相似，只是向下走的次数有限。

那么，我们就可以建立一个三维数组at[i][j][k]，ij表示方位，k表示向下走的次数。

这样我们可以得出公式：

代码见下：

#include<bits/stdc++.h>
using namespace std;
long long int n,m,x,a,aa[10240][10240],at[1001][1001][101],dd0=0;
int main(){
	cin>>x>>a;
	//memset(at,-10,sizeof(at));
	for(int i=1;i<=x;i++){
		for(int j=1;j<=i;j++){
			cin>>aa[i][j];
//			at[i][j][0]=0;
//			at[i][j][1]=0;
		}
	}
	for(int i=x;i>=1;i--){
		for(int j=i;j>=1;j--){
			at[i][j][0]=at[i+1][j+1][0]+aa[i][j];
			for(int k=1;k<=a;k++){
				at[i][j][k]=max(at[i+1][j][k-1],at[i+1][j+1][k])+aa[i][j];
			}
			//cout<<max(dd0,aa[i][j]+max(at[i+1][j][1],at[i+1][j+1][1]))<<" ";
		}
		//cout<<endl;
	}
	for(int i=0;i<=a;i++){
		dd0=max(dd0,at[1][1][i]);
	}
	cout<<dd0;
	return 0;
}