Java-数字三角形(回溯法)
题目描述:
上图给出了一个数字三角形。从三角形的顶部到底部有很多条不同的路径。对于每条路径,把路径上面的数加起来可以得到一个和,你的任务就是找到最大的和。
路径上的每一步只能从一个数走到下一层和它最近的左边的那个数或者右 边的那个数。此外,向左下走的次数与向右下走的次数相差不能超过 1。
输入描述
输入的第一行包含一个整数N (1≤N≤100),表示三角形的行数。
下面的 N 行给出数字三角形。数字三角形上的数都是 0 至 100 之间的整数。
输出描述
输出一个整数,表示答案。
输入输出样例
示例
输入
5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
输出
27
逻辑很简单,利用遍历二叉树的方法,层层递归下去,记录左右行走次数把符合条件的赋值并比较找出权值最大的路
代码如下:
import java.util.Scanner;
public class Num_Triangle {
public static int sum = 0;
public static int temp = 0;
public static int l = 0;//向左走的次数
public static int r = 0;//向右走的次数
public static void dfs(int a[][], int row, int col) {
//遍历到叶子节点终止
if (row == a.length - 1) {
if (temp >= sum && Math.abs(l - r) <= 1) {
sum = temp;
}
return;
}
//模拟二叉树向左下走
l++;
temp += a[row + 1][col];
dfs(a, row + 1, col);
//回溯
l--;
temp -= a[row + 1][col];
//向右下走
r++;
temp += a[row + 1][col + 1];
dfs(a, row + 1, col + 1);
//回溯
r--;
temp -= a[row + 1][col + 1];
}
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int n = scan.nextInt();
int[][] a = new int[n][100];
for (int i = 0; i < n; i++) {
for (int j = 0; j <= i; j++) {
a[i][j] = scan.nextInt();
}
}
temp = a[0][0];//根结点赋值
dfs(a, 0, 0);
System.out.println(sum);
}
}