CODEWITHSUNDEEP
pythonoptimizationsum
rasen58
rasen58

Reputation: 5081

Best way to speed up a double summation

My code works, but I need it to run in under 1 second. How do I do that?

For this summation

import math
import time

t=time.time()
line = input().split(' ')
Alen = int(line[0])
Blen = int(line[1])

A = input().split(' ')
B = input().split(' ')

total=0
for i, Ai in enumerate(A):
    for j, Bj in enumerate(B):
       total+= abs((int(Ai)-int(Bj)))*(int(i)-int(j))

print(total)#, time.time()-t)

Upvotes: 0

Views: 556

Answers (1)

Hugh Bothwell
Hugh Bothwell

Reputation: 56624

This should be a bit faster; no way of telling how much without some sample data.

import time

def get_ints(s):
    return [int(i) for i in s.split()]

t = time.time()

Alen, Blen = get_ints(input())
A = get_ints(input())
B = get_ints(input())

total = sum(abs(Ai - Bj) * (i - j) for i,Ai in enumerate(A) for j,Bj in enumerate(B))

print(total)#, time.time()-t)

Upvotes: 2

Related Questions