【算法】链表的基本操作和高频算法题

链表的基本操作

链表的基础操作有查找、删除、添加。

查找

先定义一下链表的数据结构:

class DataNode{     int key;     int value;          DataNode pre;     DataNode next;          public DataNode(){};          public DataNode (int key,int value){         this.key = key;         this.value = value;     } } 

【算法】链表的基本操作和高频算法题

​ 其中的key和value就是节点实际存储的值。pre和next分别指向前一个节点和下一个节点。一般的单向链表只有next,我这里定义的是双向链表。查找操作就是从头节点,一直遍历next,直到找到目标节点为止。可以用while循环或者递归实现,找到目标节点就跳出或返回即可,时间复杂度为O(n)。

删除

以上面的双向链表为例,演示一下删除操作。我们假设有三个节点A、B、C,现要删除B节点。把A.next指向C,C.pre指向A。中间的B节点就不在链路上了,会被垃圾收回器给回收掉。

public void delNode(DataNode node){     node.pre.next = node.next;     node.next.pre = node.pre; } 

上面的代码初看可能有点绕,其实链表除了查找操作,都有点绕,建议画图理解。其中node.pre.next就是上一个节点的下一个节点,把它改成node.next,就相当于让上一个节点指向自己的下一个节点。第二行代码就是让下一个节点的pre指向自己的上一个节点。删除操作的时间复杂度为O(1)。

添加

假设有A、C两个节点,现要往中间添加一个B节点。思路看图都能想到,你的写法不一定要和我一样,只是注意别丢失节点了,代码如下:

//写法1 public void addNode(DataNode pre,DataNode node){     //先记录一下pre.next节点,否则下一步会丢失C节点     DataNode next = pre.next; // 记录C     pre.next = node; //A->B     node.next = next; //B->C     next.pre = node; // B<-C      node.pre = pre; // A<-B } //写法2,不用临时变量 public void addNode(DataNode pre,DataNode node){     node.next = pre.next; //B->C     node.pre = pre; //A<-B     pre.next = node; // A->B     node.next.pre = node; //C<-B } 

算法题

LRU缓存

关于链表的算法题中,我觉得最能训练链表操作的就是LRU缓存。即给出已给固定容量的容器,往里put元素时,如果容量到达最大,就删除最久未使用的元素

题目描述:

实现 LRUCache 类:

LRUCache(int capacity) 以 正整数 作为容量 capacity 初始化 LRU 缓存

int get(int key) 如果关键字 key 存在于缓存中,则返回关键字的值,否则返回 -1 。

void put(int key, int value) 如果关键字 key 已经存在,则变更其数据值 value ;如果不存在,则向缓存中插入该组 key-value 。如果插入操作导致关键字数量超过 capacity ,则应该 逐出 最久未使用的关键字。

函数 get 和 put 必须以 O(1) 的平均时间复杂度运行。

思路:根据key找到value,所以肯定要一个Hash表存储值。元素数量超过capacity就要删除最久未使用的关键字。我们就设计一个链表,每次get元素A时,就把A移到链表头部。需要删除元素时,直接删除链表尾部的元素,尾部的就是最久没使用的。

每次get时要把对应的元素移至头部,为了避免遍历链表,设计Hash表的value类型可以设置成DataNode,这样就免去的链表查找的时间。DataNode就复用开头定义的数据结构。

class LRUCache {     int capacity;     HashMap<Integer,DataNode> map;          //定义一个虚拟的头节点和尾节点,方便删除尾节点和往头节点添加元素     DataNode head;     DataNode tail;      //1.初始化相关属性     public LRUCache(int capacity) {         this.capacity = capacity;         map = new HashMap<>();         head = new DataNode();         tail = new DataNode();         head.next = tail;         tail.pre = head;     }     //2.实现get逻辑,里面的moveToHead可以先不实现     public int get(int key) {         DataNode node = map.get(key);         if(node==null){             return -1;         }         //把node节点移至头部         moveToHead(node);         return node.value;     }          //3.实现put逻辑     public void put(int key, int value) {         if(map.containsKey(key)){             //如果当前key已经存在             DataNode node = map.get(key);             moveToHead(node);             node.value = value;         }else{             //不存在就新建一个node,如果超过capacity就删除尾部节点             DataNode node = new DataNode(key,value);             map.put(key,node);             if(map.size()>capacity){                 //因为还要从map中删除元素,所以removeTail要有返回值                 DataNode delNode = removeTail();                 map.remove(delNode.key);             }             addHead(node);         }     }          //4.最后一步,实现上面所需的链表操作方法     private void moveToHead(DataNode node){         //先删除,再移至头部         removeNode(node);         addHead(node);     }          private void addHead(DataNode node){         node.next = head.next;         node.pre = head;         head.next = node;         node.next.pre = node;     }          private DataNode removeTail(){         DataNode delNode = tail.pre;         removeNode(delNode);         return delNode;     }          //removeTail和moveToHead都有删除元素的操作,所以再提取一个删除方法     private void removeNode(DataNode node){         node.pre.next = node.next;         node.next.pre = node.pre;     } } 

反转链表

反转链表也是面试中出现频率比较高的,反转链表是个单向链表,它的操作比双向链表更简单。

题目描述:

给你单链表的头节点 head ,请你反转链表,并返回反转后的链表。

【算法】链表的基本操作和高频算法题

思路:定义两个指针(变量),一个指向当前节点,一个指向前一个节点。每次反转指针指向的两个节点,然后指针往后移一位。

/**  * Definition for singly-linked list.  * public class ListNode {  *     int val;  *     ListNode next;  *     ListNode() {}  *     ListNode(int val) { this.val = val; }  *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }  * }  */ class Solution {     public ListNode reverseList(ListNode head) {         if(head==null||head.next==null){             return head;         }         ListNode pre = null;         ListNode node = head;         while(node!=null){             ListNode temp = node.next;             //反转node和pre             node.next = pre;             //node和pre往后移一位             pre = node;             node = temp;         }         //因为node最终会移到尾节点的next上,也就是null         //所以pre才是真正的尾节点,也就是反转后的头节点         return pre;     } } 

环形链表

题目描述:

出一个链表的head,判断该链表是否是环形链表。如果是,就返回环形的入口。如果不是,就返回null。

【算法】链表的基本操作和高频算法题

如上图,入口节点就是2。

思路1:要做出这个题不难,第一下就能想到:边遍历链表,边往Hash表存储节点,每次遍历前判断Hash表是否存在当前节点,如果存在,这个节点就是环形入口。如果遍历完了,还没有重复节点,就说明没有环形。时间复杂度:O(n),空间复杂度:O(n)。

/**  * Definition for singly-linked list.  * class ListNode {  *     int val;  *     ListNode next;  *     ListNode(int x) {  *         val = x;  *         next = null;  *     }  * }  */ public class Solution {     public ListNode detectCycle(ListNode head) {         HashSet<ListNode> set = new HashSet<>();         while(head!=null){             if(set.contains(head)){                 return head;             }             set.add(head);             head = head.next;         }         return null;     } } 

思路2:优化链表的空间复杂度常用手段就是用指针,思路2就是定义两个快慢指针,属于数学逻辑范畴了。我们定义一个慢指针slow,一个快指针fast。slow一次移动一位,fast一次移动两位。

  • 如果fast移到了null节点,说明链表无环,直接返回null

  • fast和slow相遇

    • 此时fast和slow一定在环形内,否则不可能相遇。我们假设head到环形入口(不含入口)的长度为x,环形长度为y
    • 然后假设slow走了s步,则fast走了2s步(fast是slow的两倍速)
    • fast和slow相遇时,fast在环内比slow多走了ny步(关键点,可以画图理解一下)
      • 所以fast=2s=s+ny(s是slow走的步数,ny是fast比slow多走的步数)
      • 所以s=ny
    • 根据上面的推测s=ny,接着可以推算出,入口点就是x+ny。因为y是环形长度,n是正整数,所以ny实际上和y没区别,无非就是多绕了几圈。(关键点,也可以画图理解一下)。
    • 此时slow已经走了ny步,所以再走x步就是入口点了。但是我们不知道x等于多少,那我们就让一个指针从head再走一遍,一次走一步,和slow相遇点就是入口点。为了少创建一个变量,可以让fast指针回到head节点重新走。
public class Solution {     public ListNode detectCycle(ListNode head) {         if(head==null||head.next==null){             return null;         }         ListNode slow = head;         ListNode fast = head;         while(true){             if(fast == null||fast.next == null){                 return null;             }             slow = slow.next;             fast = fast.next.next;             //第一次相遇             if(fast==slow){                 break;             }         }         fast = head;         while(fast!=slow){             fast = fast.next;             slow = slow.next;         }         return slow;     } } 

总结

链表必须要掌握它的删除、添加、查找三个基础操作。链表的类型还分为:单向链表、循环链表(头尾相连,或者带环的)、双向链表。只要掌握了双向链表的基础操作,其他链表都不在话下。

关于链表的算法中,因为不能像数组那样,通过下标随机访问,所以一般会把节点存进Hash表。如果Hash表也不想存,想优化空间复杂度,一般的做法是定义指针。单向链表一般要定义双指针,一个指向当前,一个指向前一个,如果是双向链表,只用定义一个。但是在算法题中,单纯只考链表的题目比较少,很多都会带一些其他知识点。比如链表的排序、链表的二分查找等。只要熟练掌握链表的插入、删除,只用考虑排序、查找的逻辑就行了,跟数组的排序、二分查找没啥区别。

发表评论

相关文章