在牛客网的二叉树前序遍历中,需要将ArrayList<Integer>
转换成int[]
数组,ArrayList中存放的是Integer类型数据,如果直接arrayList.toString()的话,得到的是Object[]的结果,不是int[],不满足要求,
两种解决方案如下
方法1:创建新数组并对其遍历赋值
ArrayList<Integer> arrList = new ArrayList<>();
arrList.add(7);
arrList.add(8);
arrList.add(9);
int[] res = new int[arrList.size()];
for (int i = 0; i < res.length; i++) {
res[i] = arrList.get(i);
}
方法2:使用Stream流转换
ArrayList<Integer> arrList = new ArrayList<>();
arrList.add(7);
arrList.add(8);
arrList.add(9);
int[] ints = arrList.stream().mapToInt(i -> i).toArray(); // 关键代码,使用到了Stream转换
//int[] ints = arrList.stream().mapToInt(i -> i.intValue()).toArray(); // 这种写法也可以
System.out.println("结果是" + ints); // 结果是[I@7cca494b, 打印的是int[]的地址
System.out.println("结果内容是" + Arrays.toString(ints)); // 结果内容是[7, 8, 9]
stream()方法可以追溯到Collection接口,ArrayList implements List接口,List接口extends Collection接口,Collection接口中的相关内容如下:
default Stream<E> stream() {
return StreamSupport.stream(spliterator(), false);
}
牛客原题如下:
https://www.nowcoder.com/practice/5e2135f4d2b14eb8a5b06fab4c938635?tpId=295&tqId=2291302&ru=/exam/oj&qru=/ta/format-top101/question-ranking&sourceUrl=%2Fexam%2Foj%3Fpage%3D1%26tab%3D%25E7%25AE%2597%25E6%25B3%2595%25E7%25AF%2587%26topicId%3D295
BM23 二叉树的前序遍历
给你二叉树的根节点 root ,返回它节点值的前序遍历。
数据范围:二叉树的节点数量满足1≤n≤100, 二叉树节点的值满足1≤val≤100, 树的各节点的值各不相同
代码如下:
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型一维数组
*/
ArrayList<Integer> res = new ArrayList<>();
public int[] preorderTraversal (TreeNode root) {
// write code here
if (root == null) return new int[0];
res.add(root.val);
preorderTraversal(root.left);
preorderTraversal(root.right);
// 关键代码,使用到了上面介绍的第二种方法
return res.stream().mapToInt(i -> i.intValue()).toArray();
}
}