|
| 1 | +#!/usr/bin/env python |
| 2 | +# -*- coding: utf-8 -*- |
| 3 | +""" |
| 4 | +@create: 2022/3/25 17:55 |
| 5 | +@file: leetcode707.py |
| 6 | +@author: imoyao |
| 7 | +@email: immoyao@gmail.com |
| 8 | +@desc: |
| 9 | +""" |
| 10 | +''' |
| 11 | +设计链表的实现。您可以选择使用单链表或双链表。单链表中的节点应该具有两个属性:val 和 next。val 是当前节点的值,next 是指向下一个节点的指针/引用。如果要使用双向链表,则还需要一个属性 prev 以指示链表中的上一个节点。假设链表中的所有节点都是 0-index 的。 |
| 12 | +
|
| 13 | +在链表类中实现这些功能: |
| 14 | +
|
| 15 | +get(index):获取链表中第 index 个节点的值。如果索引无效,则返回-1。 |
| 16 | +addAtHead(val):在链表的第一个元素之前添加一个值为 val 的节点。插入后,新节点将成为链表的第一个节点。 |
| 17 | +addAtTail(val):将值为 val 的节点追加到链表的最后一个元素。 |
| 18 | +addAtIndex(index,val):在链表中的第 index 个节点之前添加值为 val 的节点。如果 index 等于链表的长度,则该节点将附加到链表的末尾。如果 index 大于链表长度,则不会插入节点。如果index小于0,则在头部插入节点。 |
| 19 | +deleteAtIndex(index):如果索引 index 有效,则删除链表中的第 index 个节点。 |
| 20 | +
|
| 21 | +来源:力扣(LeetCode) |
| 22 | +链接:https://leetcode-cn.com/problems/design-linked-list |
| 23 | +著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。 |
| 24 | +''' |
| 25 | + |
| 26 | + |
| 27 | +class Node: |
| 28 | + |
| 29 | + def __init__(self, val, next_node=None): |
| 30 | + self.val = val |
| 31 | + self.next = next_node |
| 32 | + |
| 33 | + # def __repr__(self): |
| 34 | + # return self.val |
| 35 | + |
| 36 | + |
| 37 | +class NodeList: |
| 38 | + |
| 39 | + def __init__(self): |
| 40 | + self.size = 0 |
| 41 | + self.head = Node(0) |
| 42 | + |
| 43 | + def __iter__(self): |
| 44 | + current = self.head |
| 45 | + while current.next is not None: |
| 46 | + yield current.val |
| 47 | + current = current.next |
| 48 | + |
| 49 | + def get(self, index): |
| 50 | + if index < 0 or index >= self.size: |
| 51 | + return -1 |
| 52 | + current = self.head |
| 53 | + for _ in range(index + 1): |
| 54 | + current = current.next |
| 55 | + return current.val |
| 56 | + |
| 57 | + def add_at_index(self, index, val): |
| 58 | + if index > self.size: |
| 59 | + return None |
| 60 | + if index < 0: |
| 61 | + index = 0 |
| 62 | + |
| 63 | + self.size += 1 |
| 64 | + # 找到被放置节点的前一个,在哪里放置,就for几次 |
| 65 | + prev = self.head # 基准节点 |
| 66 | + for _ in range(index): |
| 67 | + prev = prev.next |
| 68 | + # 构造被插入节点 |
| 69 | + add_node = Node(val) |
| 70 | + add_node.next = prev.next |
| 71 | + # 放入断链位置 |
| 72 | + prev.next = add_node |
| 73 | + |
| 74 | + def add_at_head(self, val): |
| 75 | + self.add_at_index(0, val) |
| 76 | + |
| 77 | + def add_at_tail(self, val): |
| 78 | + self.add_at_index(self.size, val) |
| 79 | + |
| 80 | + def delete_at_index(self, index): |
| 81 | + if index < 0 or index > self.size: |
| 82 | + return None |
| 83 | + self.size -= 1 |
| 84 | + prev = self.head |
| 85 | + for _ in range(index): |
| 86 | + prev = prev.next |
| 87 | + prev.next = prev.next.next |
| 88 | +if __name__ == '__main__': |
| 89 | + |
| 90 | + |
| 91 | + nl = NodeList() |
| 92 | + for item in range(10): |
| 93 | + if item % 2 == 0: |
| 94 | + nl.add_at_head(item) |
| 95 | + else: |
| 96 | + nl.add_at_tail(item) |
| 97 | + |
| 98 | + for item in nl: |
| 99 | + print(item) |
0 commit comments