LintCode:重哈希

LintCode:重哈希

"""
Definition of ListNode
class ListNode(object):

    def __init__(self, val, next=None):
        self.val = val
        self.next = next
"""
class Solution:
    """
    @param hashTable: A list of The first node of linked list
    @return: A list of The first node of linked list which have twice size
    """
    def rehashing(self, hashTable):
        # write your code here
        m = len(hashTable)
        ans = [None for i in range(2 * m)]
        for head in hashTable:
            if head == None:
                pass
            else:
                p1 = head
                while p1 != None:
                    n = p1.val % (2 * m)
                    if ans[n] == None:
                        ans[n] = ListNode(p1.val)
                    else:
                        p2 = ans[n]
                        tmp_node = ListNode(p1.val)
                        while p2.next != None:
                            p2 = p2.next
                        p2.next = tmp_node
                    p1 = p1.next
        return ans