LeetCode算法题-Pascal's Triangle II(Java实现)
这是悦乐书的第171次更新,第173篇原创
01 看题和准备
今天介绍的是LeetCode算法题中Easy级别的第30题(顺位题号是119)。给定非负索引k,其中k≤33,返回Pascal三角形的第k个索引行。行索引从0开始。在Pascal的三角形中,每个数字是它上面两个数字的总和。例如:
输入: 2
输出: [1,2,1]
输入: 3
输出: [1,3,3,1]
本次解题使用的开发工具是eclipse,jdk使用的版本是1.8,环境是win7 64位系统,使用Java语言编写和测试。
02 第一种解法
昨天的那道题,是要求输出所有层的数据,今天这题只要求具体某一行的数据,那是不是完全可以借用昨天的代码,最后返回具体某一层即可?这种解法当然是可行的,但是题目有个提示,要求空间复杂度是O(k),其中k代表某一行。而昨天的解法,最后的空间复杂度是O(numRows^2),那能不能把空间复杂度再缩小点?答案是可以的。
我们可以先将list中的每个元素初始值设为1,然后开始循环处理。外层循环控制层数,进入内层循环,改变具体位置元素的值。我们知道第k层第j(j>=1)个元素的值是第k-1层第j-1个元素和第j个元素之和,如果按照从左至右的顺序,后面元素的值需要借助前面元素的值的时候,是已经被改变过的值,最后的结果会失真,所以我们采用从右至左的顺序设值。
注意:list.set(j, value)此方法是找到索引j所在元素,将value设值给它,不会新增元素,而add(value)是会新增元素的。
public List<Integer> getRow(int rowIndex) {
List<Integer> list = new ArrayList<Integer>();
for (int k = 0; k <= rowIndex; k++) {
list.add(1);
}
for (int i = 1; i < rowIndex; i++) {
for (int j = i; j > 0; j--) {
list.set(j, list.get(j - 1) + list.get(j));
}
}
return list;
}
03 第二种解法
同样是第一种解法的思路,但是我想从左至右设值怎么办?
既然想要从左至右设值,那就需要左边的数据是新的,而不是已经被计算过的新值,就需要把每次计算过的值往后移动一位,此时我们就无法在开始时就将list的初始元素值设为1,二是要在循环中添加新的元素,从而把旧的元素往后挪一位。
外层循环一方面是控制层数,另外一个方面就是每次在list的索引0位置新增元素1。只有当r等于2时,才会进入内层循环,此时list中至少有三个元素。
因为首尾元素都是1,所以内层循环的索引起始位置是1(第二位),依次往右重新设值。
public List<Integer> getRow2(int rowIndex) {
List<Integer> list = new ArrayList<>();
for (int r = 0; r <= rowIndex; r++) {
list.add(0, 1);
for (int i = 1; i < r; i++) {
list.set(i, list.get(i) + list.get(i + 1));
}
}
return list;
}
04 第三种解法
我们还可以使用数组来操作此题,思路与昨天的那道题大致类似。
外层循环每次创建一个新的数组,数组长度是i+1,并且此数组的首尾元素是为1,然后开始内层循环,新数组的第j(j>=1)位元素是上一数组中的第j-1位和第j位元素之和。在结束内层循环后,还要将旧数组指向经过内层循环赋值完后的新数组。最后,将数组里面的元素遍历赋值给list。
public List<Integer> getRow3(int rowIndex) {
List<Integer> list = new ArrayList<Integer>();
int[] result = new int[1];
for (int i = 0; i <= rowIndex; i++) {
int[] next = new int[i + 1];
next[0] = 1;
next[i] = 1;
for (int j = 1; j < i; j++) {
next[j] = result[j - 1] + result[j];
}
result = next;
}
for (int in : result) {
list.add(in);
}
return list;
}
05 验证与测试
针对上面三种解法,选取部分随机数字并测试不同解法耗时,代码如下:
public static void main(String[] args) {
Easy_119_PascalTriangleII instance = new Easy_119_PascalTriangleII();
int rowIndex = 3;
long start = System.nanoTime();
List<Integer> list = instance.getRow(rowIndex);
long end = System.nanoTime();
System.out.println("getRow---输入:"+rowIndex+" , 输出:"+list+" , 用时:"+(end-start)/1000+"微秒");
long start2 = System.nanoTime();
List<Integer> list2 = instance.getRow2(rowIndex);
long end2 = System.nanoTime();
System.out.println("getRow2---输入:"+rowIndex+" , 输出:"+list2+" , 用时:"+(end2-start2)/1000+"微秒");
long start3 = System.nanoTime();
List<Integer> list3 = instance.getRow3(rowIndex);
long end3 = System.nanoTime();
System.out.println("getRow3---输入:"+rowIndex+" , 输出:"+list3+" , 用时:"+(end3-start3)/1000+"微秒");
}
测试结果如下:
getRow---输入:10 , 输出:[1, 10, 45, 120, 210, 252, 210, 120, 45, 10, 1] , 用时:73微秒
getRow2---输入:10 , 输出:[1, 10, 45, 120, 210, 252, 210, 120, 45, 10, 1] , 用时:48微秒
getRow3---输入:10 , 输出:[1, 10, 45, 120, 210, 252, 210, 120, 45, 10, 1] , 用时:24微秒
06 小结
解法一和解法二的思路类似,但是第二种解法稍快于解法一,第三种解法耗时最少,但是三种解法的空间复杂度都是O(k)。
以上就是全部内容,如果大家有什么好的解法思路、建议或者其他问题,可以下方留言交流,点赞、留言、转发就是对我最大的回报和支持!