我一直在研究代码战的难题,可以在这里找到:
http://www.codewars.com/kata/can-you-get-the-loop
基本上,输入是链表的第一个节点,保证有一些长度的尾部和一些长度的循环。 (图片见链接。)
我的解决方案是让两个迭代器逐步遍历列表,一个访问每个节点,一个跳过每个节点。一旦它们命中,我知道我在循环中,所以我只计算一个循环并返回计数。
这是我的代码:
def loop_size(node):
size = 1
onestep = node
twostep = node.next
while(onestep != twostep):
twostep = twostep.next.next
onestep = onestep.next
#we are inside the loop
#onestep == twostep
onestep = node.next
size += 1
while(onestep != twostep):
size += 1
onestep = onestep.next
return size
出于某种原因,我得到了奇怪的结果。每当尾部小于环路时,我得到正确的结果。但只要尾部长度大于或等于循环的大小,我的函数就会获得更高的计数。
以下是一些例子:
Tail length = 1 Loop Length = 3
###result 3 - correct
Tail length = 999 Loop Length = 1000
###result 1000 - correct
Tail length = 1000 Loop Length = 999
###result 1998 - incorrect
Tail length = 50 Loop Length = 3
###result 51 - incorrect
Tail length = 3 Loop Length = 3
###result 6 - incorrect
Tail length = 3 Loop Length = 4
###result 4 - correct
答案 0 :(得分:1)
行
onestep = onestep.next
应该是
class Node(object):
def __init__(self, next_node=None):
self.next_node = next_node
@property
def next(self):
return self.next_node
def set_next(self, new_next):
self.next_node = new_next
return new_next
def loop_size(node):
onestep = node
twostep = node.next
while(onestep != twostep):
twostep = twostep.next.next
onestep = onestep.next
onestep = onestep.next
size = 1
while(onestep != twostep):
size += 1
onestep = onestep.next
return size
def test_ll(tail, loop):
head = Node()
nodes = [head]
for i in range(2, tail+loop+1):
head = head.set_next(Node())
nodes.append(head)
nodes[-1].set_next(nodes[tail])
size = loop_size(nodes[0])
print "Tail: {}, Loop: {}, Size: {}".format(tail, loop, size)
test_ll(1, 3)
test_ll(999, 1000)
test_ll(1000, 999)
test_ll(50, 3)
test_ll(3, 3)
test_ll(3, 4)
否则你再次从头开始并重新进入循环,这样你的结果将是尾长太长。另外,我认为你的大小应该从1开始,而不是2开始(大小= 1,大小+ = 1,然后第二个循环开始)。
此代码适用于您的所有示例:
Tail: 1, Loop: 3, Size: 3
Tail: 999, Loop: 1000, Size: 1000
Tail: 1000, Loop: 999, Size: 999
Tail: 50, Loop: 3, Size: 3
Tail: 3, Loop: 3, Size: 3
Tail: 3, Loop: 4, Size: 4
<强>输出强>
{{1}}