Skumulowaną sumę wektora oblicza się po prostu biorąc sumę wszystkich poprzednich elementów. Na przykład:
vec = [1 1 1 -1 -1 -1 -1 -1 1 1 1 1 -1]
cum_vec = [1 2 3 2 1 0 -1 -2 -1 0 1 2 1]
Teraz nałóż górną i dolną granicę, co oznacza, że przestaniesz zwiększać sumę skumulowaną, jeśli jest na górnej granicy, i przestaniesz zmniejszać sumę skumulowaną, jeśli będzie na dolnej granicy. Prosty przykład:
upper_lim = 2
lower_lim = -1
vec = [1 1 1 -1 -1 -1 -1 -1 1 1 1 1 -1]
cum_vec = [1 2 2 1 0 -1 -1 -1 0 1 2 2 1]
Wektor wejściowy składa się z liczb całkowitych, niekoniecznie tylko 1
i -1
dodatnich i ujemnych. Załóżmy, że upper_lim >= lower_lim
. Jeśli pierwszy element wektora znajduje się poza granicą, przeskocz bezpośrednio do granicy (patrz ostatni przykład).
Napisz funkcję, która pobiera wektor liczb całkowitych jako dane wejściowe i dwie liczby całkowite reprezentujące górną i dolną granicę. Wyprowadza ograniczony wektor skumulowany, jak zdefiniowano powyżej. Dane wejściowe mogą być argumentami funkcji lub ze STDIN.
Obowiązują standardowe zasady gry w golfa.
Przykłady:
upper_lim = 6
lower_lim = -2
vec = [1 4 3 -10 3 2 2 5 -4]
cum_vec = [1 5 6 -2 1 3 5 6 2]
upper_lim = 100
lower_lim = -100
vec = [1 1 1 1 1 1]
cum_vec = [1 2 3 4 5 6]
upper_lim = 5
lower_lim = 0
vec = [10 -4 -3 2]
cum_vec = [5 1 0 2]
upper_lim = 0
lower_lim = 0
vec = [3 5 -2 1]
cum_vec = [0 0 0 0]
upper_lim = 10
lower_lim = 5
vec = [1 4 6]
cum_vec = [5 9 10]
|
Note, jumped to 5, because 5 is the lower bound.
źródło