Python 数据结构之旋转链表

作者:guoziqing506 时间:2021-05-04 15:39:28 

题目描述:给定一个链表,旋转链表,使得每个节点向右移动k个位置,其中k是一个非负数

样例:给出链表1->2->3->4->5->null和k=2;返回4->5->1->2->3->null

首先,观察一下这个题目要达到的目的,其实,换一种说法,可以这样来描述:给出一个k值,将链表从倒数第k个节点处起之后的部分移动到链表前面,就样例来说,其实是将4->5这一部分移动到整个链表前面,变成4->5->1->2->3->null。不过,需要注意的是,题中没有给出k的大小,当k比链表的长度还大的时候,我们就需要先用k对链表的长度求余,比如,如果k = 7,那么上面的例子还是将4->5移动到整个链表前面。

所以说,这个题的思路可以这样来总结:

1. 先求出整个链表的长度
2. 根据k值找到需要移动的部分链表的前驱(样例中的3)
3. 在前驱之后将链表断开,移动后半部分

代码如下:


# Definition for singly-linked list.
# class ListNode:
#   def __init__(self, x):
#     self.val = x
#     self.next = None

class Solution:
 # @param head: the list
 # @param k: rotate to the right k places
 # @return: the list after rotation
 def rotateRight(self, head, k):
   if head is None:
     return head
   cur = head
   count = 1
   # 计算链表长度
   while cur.next:
     cur = cur.next
     count += 1
   # 为节省代码量,这里是一个很有技巧的处理:用尾节点链接头结点
   cur.next = head
   # 此处,k为cur从尾节点到要断开部分的前驱需走的步数
   k = count - k % count
   # 找到前驱
   while k != 0:
     cur = cur.next
     k -= 1
   # 断开
   head = cur.next
   cur.next = None
   # 因为首尾已经相连,所以直接返回前驱后面的那个节点即可,此处引用为head
   return head
   # write your code here

需要注意的是21行首尾相连的技巧,这大大节省了我们的代码量,其实,就按之前思路中所描述的一步步来,也没问题。但是这个技巧确实很棒,值得学习。具体的细节我写在了代码注释里。

感谢阅读,希望能帮助到大家,谢谢大家对本站的支持!

来源:http://blog.csdn.net/guoziqing506/article/details/51365913

标签:Python,链表
0
投稿

猜你喜欢

  • Python向日志输出中添加上下文信息

    2021-01-09 15:04:42
  • 解决Django加载静态资源失败的问题

    2023-09-11 21:10:51
  • web2.0中流行的设计元素:颜色

    2007-12-10 12:16:00
  • 一文教你实现Python重试装饰器

    2022-04-23 08:01:07
  • Go语言reflect.TypeOf()和reflect.Type通过反射获取类型信息

    2024-04-23 09:38:27
  • Python去除、替换字符串空格的处理方法

    2022-07-17 05:38:25
  • Python的Django框架中设置日期和字段可选的方法

    2023-03-11 11:41:51
  • 小议sqlserver数据库主键选取策略

    2011-10-24 19:51:30
  • 使用Tensorboard工具查看Loss损失率

    2022-04-17 12:50:30
  • SQL写法--行行比较

    2024-01-12 18:53:34
  • 解决flask接口返回的内容中文乱码的问题

    2022-12-17 03:00:37
  • SQL查询不重复记录/删除重复记录

    2008-11-18 16:08:00
  • 关于Python中的main方法教程

    2021-12-30 08:31:37
  • python爬虫 批量下载zabbix文档代码实例

    2022-11-07 11:10:29
  • Flask框架之数据交互的实现

    2023-01-25 05:41:10
  • Python Celery定时任务详细讲解

    2022-08-12 08:25:21
  • JavaScript学习心得之如何走出初学困境

    2008-12-24 13:30:00
  • 基于ExtJs框架的B/S高级查询界面

    2008-10-07 18:05:00
  • 利用Python实现Json序列化库的方法步骤

    2023-01-15 07:03:00
  • python实现双链表

    2022-06-20 01:47:48
  • asp之家 网络编程 m.aspxhome.com