Mohit and Swiggy

  • + 1 comment

    import math import os import random import re import sys import numpy as np

    def solve(n, roads): # Write your code here a = np.array(roads) b = a[:,n] c = np.sum(b)

    return (c)
    

    if name == 'main':

    ## fptr = open(os.environ['OUTPUT_PATH'], 'w')

    first_multiple_input = input().rstrip().split()
    
    n = int(first_multiple_input[0])
    
    
    m = int(first_multiple_input[1])
    
    
    roads = []
    
    for _ in range(m):
        roads.append(list(map(int, input().rstrip().split())))
    
    
    result = solve(n, roads)
    print(result)
    

    **## fptr.write(str(result) + '\n')

    #

    fptr.close()**