Generation

generate functionMon, 10 Jul 2023

La entrada contiene varios casos de prueba, el primer entero indica la cantidad de casos La entrada consiste de un numero c que es el número de casos, luego le siguen c líneas donde cada línea es un caso de prueba que consta de un numero n, (n será mayor que 2 y menor o igual que 500) que es la cantidad de pilas de cajas, posteriormente le siguen n números, donde cada número representa la cantidad de cajas en cada pila de cajas. En este problema se garantiza que cada entrada contiene el número exacto de cajas para formar una escalera perfecta, es decir una escalera donde la diferencia entre cada escalón será 1, por lo que nunca faltaran o sobraran cajas

def escalera(n, cajas): sub = [] cont = 0 for i in range(len(cajas)): sub.append(cajas[i] - cajas[i-1]) cont = sub.count(1) if cont == len(cajas) - 1: print('SI') else: print('NO') if __name__ == '__main__': n = int(input()) cajas = list(map(int, input().split())) escalera(n, cajas) #Input 4 1 3 5 7 #Output SI

Want to kickstart your project?Use the new AI Studio to create your code