Python: encuentre la suma de la longitud de las strings en los índices dados

Dada la lista de strings, escriba un programa de Python para calcular la suma de las longitudes de los índices personalizados de la lista.

Ejemplos:

Entrada : test_list = [“gfg”, “is”, “best”, “for”, “geeks”], idx_list = [0, 1, 4] 
Salida : 10 
Explicación : 3 + 2 + 5 = 10. (Tamaños de strings en idx.)
 

Entrada : test_list = [“gfg”, “is”, “best”, “for”, “geeks”], idx_list = [0, 2, 4] 
Salida : 12 
Explicación : 3 + 4 + 5 = 12. 

Método #1: Usar len() + loop 

En esto, iteramos para todos los índices y verificamos si ocurren en la lista de índices, si es así, incrementamos la frecuencia en el contador de suma.

# Python3 code to demonstrate working of
# Length sum of custom indices Strings
# Using len() + loop
 
# initializing list
test_list = ["gfg", "is", "best", "for", "geeks"]
 
# printing original lists
print("The original list is : " + str(test_list))
 
# initializing idx list
idx_list = [0, 1, 4]
 
res = 0
for idx, ele in enumerate(test_list):
 
    # adding length if index in idx_list
    if idx in idx_list:
        res += len(ele)
 
# printing result
print("Computed Strings lengths sum : " + str(res))

The original list is : ['gfg', 'is', 'best', 'for', 'geeks']
Computed Strings lengths sum : 10

Método #2: Usar sum() + len() + comprensión de lista

En esto, realizamos la tarea de realizar la suma usando sum(), el resto todas las funcionalidades se realizan según el método anterior, solo como una sola línea.

# Python3 code to demonstrate working of
# Length sum of custom indices Strings
# Using sum() + len() + list comprehension
 
# initializing list
test_list = ["gfg", "is", "best", "for", "geeks"]
 
# printing original lists
print("The original list is : " + str(test_list))
 
# initializing idx list
idx_list = [0, 1, 4]
 
# performing summation using sum()
# len() used to get strings lengths
res = sum([len(ele) for idx, ele in enumerate(test_list) if idx in idx_list])
 
# printing result
print("Computed Strings lengths sum : " + str(res))

The original list is : ['gfg', 'is', 'best', 'for', 'geeks']
Computed Strings lengths sum : 10

La complejidad de tiempo y espacio para todos los métodos es la misma:

Complejidad de tiempo: O(n)

Complejidad espacial: O(n)