复杂链表的复制
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
'''
牛客网@Yannyezixin代码思路:
1、在每个节点后面复制当前节点;
2、将原节点的链接关系复制到复制节点上;
3、将此链表按奇偶位置拆分成两个链表。
'''
# -*- coding:utf-8 -*-
# class RandomListNode:
# def __init__(self, x):
# self.label = x
# self.next = None
# self.random = None
class Solution:
# 返回 RandomListNode
def Clone(self, pHead):
# write code here
if not pHead:
return None
head=pHead
# 这一步在每个节点后一步复制其链接关系,即将原来的A->B->...变为A->A->B->B->...
while head:
# 注意此处一定要保留next节点,否则经过下面的赋值会导致next节点变化
next_node=head.next
temp=RandomListNode(head.label)
temp.next=next_node
head.next=temp
head=next_node
# 这一步复制random链接关系
head=pHead
while head:
temp=head.next
random_node=head.random
if random_node:
# 指向random节点后一位(即上一个循环中复制的random节点)
temp.random=random_node.next
head=temp.next
# 分离两个链表
result=pHead.next
head=pHead
while head:
copy_node=head.next
next_node=copy_node.next
head.next=copy_node.next
# 判断链表是否结束
if next_node:
copy_node.next=next_node.next
else:
# 不可缺少这一步,否则会报错
copy_node.next=None
head=next_node
return result
二叉搜索树与双向链表
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def Convert(self, pRootOfTree):
# write code here
if not pRootOfTree:
return None
nodelist=[]
templist=[]
tempnode=pRootOfTree
while tempnode or templist:
if tempnode:
templist.append(tempnode)
tempnode=tempnode.left
else:
popnode=templist.pop()
nodelist.append(popnode)
tempnode=popnode.right
result=nodelist[0]
while nodelist:
head=nodelist.pop(0)
if nodelist:
head.right=nodelist[0]
nodelist[0].left=head
return result