Python Program for Find a triplet from three linked lists with sum equal to a given number

# Python program to find a triplet 
# from three linked lists with 
# sum equal to a given number 

# Link list node 
class Node: 
    def __init__(self, new_data): 
        self.data = new_data 
        self.next = None

# A utility function to insert 
# a node at the beginning of a 
# linked list
def push ( head_ref, new_data) :

    # allocate node 
    new_node = Node(0)

    # put in the data 
    new_node.data = new_data 

    # link the old list of the new node 
    new_node.next = (head_ref) 

    # move the head to point to the new node 
    (head_ref) = new_node
    
    return head_ref;

# A function to check if there are three elements in a, b 
# and c whose sum is equal to givenNumber. The function 
# assumes that the list b is sorted in ascending order 
# and c is sorted in descending order. 
def isSumSorted(headA, headB,headC, givenNumber) :

    a = headA 

    # Traverse through all nodes of a 
    while (a != None) :
    
        b = headB 
        c = headC 

        # For every node of list a, prick two nodes 
        # from lists b abd c 
        while (b != None and c != None) :
        
            # If this a triplet with given sum, print 
            # it and return true 
            sum = a.data + b.data + c.data 
            if (sum == givenNumber) :
            
                print "Triplet Found: " , a.data , " " , b.data , " " , c.data, 
                return True
            
            # If sum of this triplet is smaller, look for 
            # greater values in b 
            elif (sum < givenNumber): 
                b = b.next
            else :# If sum is greater, look for smaller values in c 
                c = c.next
        
        a = a.next # Move ahead in list a 
    
    print("No such triplet") 
    return False

# Driver code

# Start with the empty list 
headA = None
headB = None
headC = None

# create a linked list 'a' 10.15.5.20 
headA = push (headA, 20) 
headA = push (headA, 4) 
headA = push (headA, 15) 
headA = push (headA, 10) 

# create a sorted linked list 'b' 2.4.9.10 
headB = push (headB, 10) 
headB = push (headB, 9) 
headB = push (headB, 4) 
headB = push (headB, 2) 

# create another sorted 
# linked list 'c' 8.4.2.1 
headC = push (headC, 1) 
headC = push (headC, 2) 
headC = push (headC, 4) 
headC = push (headC, 8) 

givenNumber = 25

isSumSorted (headA, headB, headC, givenNumber) 

# This code is contributed by Arnab Kundu