fork download
  1. s = input ().split() #numero de sttar recen nacidos
  2. h = int(input()) #numero de horas de experimento
  3.  
  4. #depois tu vê como coloca pra receber dois valores com espaço
  5.  
  6. lista = [] #lista com os dados do experimento (vetor dinâmico)
  7.  
  8. tot = s #saída
  9.  
  10. lista.append(s)
  11. lista.append(0)
  12. lista.append(0)
  13. lista.append(0)
  14. lista.append(0)
  15. #primeiros elementos da lista
  16.  
  17. if h>= 3:
  18. lista[3] = lista[0]*3
  19. tot += lista[3]
  20.  
  21. #for utilizando função range para percorrer a lista com iteradores, ele começa no 4 e para quando for igual a h+1
  22. for x in range(4, h + 1):
  23. lista[x] = lista[x-4]*3 + lista[x-3]*3
  24. tot -= lista[x-4]
  25. tot += lista[x]
  26. lista.append(0) #próximo elemento do vetor a ser calculado alocado em tempo de execução
  27.  
  28. print(tot)
  29.  
  30. #eu tentei copiar a lógica do teu algorítmo, mas as saídas estão dando errado
  31.  
  32.  
  33. """
  34. #include <bits/stdc++.h>
  35. using namespace std;
  36.  
  37. long long int nas[10010];
  38.  
  39. int main (){
  40. int s, h;
  41. cin >> s >> h;
  42. int i;
  43.  
  44. nas[0] = s;
  45.  
  46. long long int tot = s;
  47.  
  48. if(h>=3){
  49. nas[3] = nas[0]*3;
  50. tot += nas[3];
  51. }
  52.  
  53. for(i=4;i<=h;i++){
  54. nas[i] = nas[i-4]*3 + nas[i-3]*3;
  55. tot -= nas[i-4];
  56. tot += nas[i];
  57. }
  58.  
  59. cout << tot << endl;
  60. }
  61. """
Runtime error #stdin #stdout #stderr 0.12s 23628KB
stdin
40 40 
stdout
Standard output is empty
stderr
Traceback (most recent call last):
  File "./prog.py", line 2, in <module>
EOFError: EOF when reading a line