Driver Code
if __name__ == ' _ _ main _ _ ' : NEW_LINE INDENT N = 5 ; NEW_LINE color = [ 1 , 0 , 0 , 0 , 2 ] NEW_LINE edge = [ [ 1 , 2 ] , [ 2 , 3 ] , [ 3 , 4 ] , [ 4 , 5 ] ] ; NEW_LINE countSubtree ( color , N , edge ) ; NEW_LINE DEDENT