代码的鲁棒性
链表中倒数第k个结点
分析
我在这里使用了栈,本题主要注意几个问题
- 输入的链表为空
- k为0
- k的值大于链表长度
代码
import java.util.Stack;
public class Solution {
public ListNode FindKthToTail(ListNode head,int k) {
if ( head == null || k == 0 ){
return null;
}
int length = 0;
Stack<ListNode> stack = new Stack();
ListNode node = head;
while( node != null ){
stack.push(node);
node = node.next;
length++;
}
if( k > length )
return null;
for( int i = 1; i < k; i++ ){
stack.pop();
}
return stack.pop();
}
}
反转链表
分析
反转链表一直绕的头晕,这次看着大佬的代码,画图理解,终于有些头绪
代码
public class Solution {
public ListNode ReverseList(ListNode head) {
if( head == null || head.next == null ){
return head;
}
ListNode pre = null;
ListNode next = null;
while( head != null ){
next = head.next;
head.next = pre;
pre = head;
head = next;
}
return pre;
}
}
合并两个排序链表
分析
数据结构中很常见的题目
设置head和一个哨兵cur,对比两个链表的结点通过哨兵将结点连接到head上。
代码
public class Solution {
public ListNode Merge(ListNode list1,ListNode list2) {
if ( list1 == null && list2 == null ) return null;
if ( list1 == null && list2 != null ) return list2;
if ( list1 != null && list2 == null ) return list1;
ListNode head = new ListNode(0);
ListNode cur = head;
while( list1 != null && list2 != null ){
if ( list1.val <= list2.val ){
cur.next = list1;
list1 = list1.next;
}else{
cur.next = list2;
list2 = list2.next;
}
cur = cur.next;
}
if ( list1 != null ) cur.next = list1;
if ( list2 != null ) cur.next = list2;
return head.next;
}
}
树的子结构
分析
这题需要注意一下子树与子结构的区别(能力有限感觉网上解释不太好,待找到好的解释再来补上)
代码
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
if ( root1 == null || root2 == null ) return false;
return subTree(root1, root2)
|| subTree(root1.left, root2)
|| subTree(root1.right, root2);
}
public boolean subTree ( TreeNode root1, TreeNode root2 ){
if ( root2 == null ) return true;
if ( root1 == null ) return false;
if ( root1.val != root2.val )
return subTree(root1.left, root2)
|| subTree(root1.right, root2);
else
return subTree(root1.left, root2.left)
&& subTree(root1.right, root2.right);
}
}