tags: [javascript]

    闲暇之余写了这个工具类，可以使用这个工具方法为链表及树相关的算法题快速构造参数。

    1. class ListNode {
    2.   constructor(val) {
    3.     this.val = val;
    4.     this.next = null;
    5.   }
    6. }
    8. class TreeNode {
    9.   constructor(val) {
    10.     this.val = val;
    11.     this.left = this.right = null;
    12.   }
    13. }
    15. function buildList(arr) {
    16.   return arr.reduceRight((pre, curr) => {
    17.     const currNode = new ListNode(curr);
    18.     currNode.next = pre;
    20.     return currNode;
    21.   }, new ListNode(arr[arr.length - 1]));
    22. }
    24. function buildTree(arr) {
    25.   if (!arr.length) return null;
    26.   if (arr.length === 1) return parent;
    27.   const parent = new TreeNode(arr[0]);
    28.   const queue = [parent];
    29.   let index = 1;
    31.   while(queue.length) {
    32.     const p = queue.shift();
    33.     if (p === null) {
    34.       index++;
    35.       continue;
    36.     }
    37.     if (index < arr.length) {
    38.       p.left = arr[index] === null ? null : new TreeNode(arr[index]);
    39.       queue.push(p.left);
    40.       index++;
    41.     }
    42.     if (index < arr.length) {
    43.       p.right = arr[index] === null ? null : new TreeNode(arr[index]);
    44.       queue.push(p.right);
    45.       index++;
    46.     }
    47.   }
    48.   return parent;
    49. }
    51. buildTree([6,2,8,0,4,7,9,null,null,3,5]);

    注:

    1. 代码中链表的构造思路是使用了 reduceRight 由最后一个节点向前构造
    2. 代码中树的构造思路是根据层次遍历的思想一层一层的构造