Commit 5c50572

AasheeshLikePannerpre-commit-ci[bot]cclaussMaximSmolskiy

authored

Fixing stock_span_problem.py (#10540)

* Adding doctests in simpson_rule.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update stock_span_problem.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update subset_generation.py * Update subset_generation.py * Update data_structures/stacks/stock_span_problem.py Co-authored-by: Christian Clauss <cclauss@me.com> * Update stock_span_problem.py * Update data_structures/stacks/stock_span_problem.py Co-authored-by: Christian Clauss <cclauss@me.com> * Update stock_span_problem.py * Update stock_span_problem.py * updating DIRECTORY.md * Update stock_span_problem.py * Update stock_span_problem.py * Update stock_span_problem.py * Update stock_span_problem.py --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Christian Clauss <cclauss@me.com> Co-authored-by: Maxim Smolskiy <mithridatus@mail.ru> Co-authored-by: MaximSmolskiy <MaximSmolskiy@users.noreply.github.com>

1 parent 4394fd9 commit 5c50572Copy full SHA for 5c50572

File tree

2 files changed

+29

-6

lines changed

DIRECTORY.md
data_structures/stacks
- stock_span_problem.py

2 files changed

+29

-6

lines changed

`‎DIRECTORY.md`

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -725,6 +725,7 @@`
`725`	`725`	`* [Secant Method](maths/numerical_analysis/secant_method.py)`
`726`	`726`	`* [Simpson Rule](maths/numerical_analysis/simpson_rule.py)`
`727`	`727`	`* [Square Root](maths/numerical_analysis/square_root.py)`
	`728`	`+ * [Weierstrass Method](maths/numerical_analysis/weierstrass_method.py)`
`728`	`729`	`* [Odd Sieve](maths/odd_sieve.py)`
`729`	`730`	`* [Perfect Cube](maths/perfect_cube.py)`
`730`	`731`	`* [Perfect Number](maths/perfect_number.py)`

`‎data_structures/stacks/stock_span_problem.py`

Lines changed: 28 additions & 6 deletions

Original file line number	Diff line number	Diff line change
`@@ -8,8 +8,29 @@`
`8`	`8`	`"""`
`9`	`9`
`10`	`10`
`11`		`-def calculation_span(price, s):`
	`11`	`+def calculate_span(price: list[int]) -> list[int]:`
	`12`	`+ """`
	`13`	`+ Calculate the span values for a given list of stock prices.`
	`14`	`+ Args:`
	`15`	`+ price: List of stock prices.`
	`16`	`+ Returns:`
	`17`	`+ List of span values.`
	`18`	`+`
	`19`	`+ >>> calculate_span([10, 4, 5, 90, 120, 80])`
	`20`	`+ [1, 1, 2, 4, 5, 1]`
	`21`	`+ >>> calculate_span([100, 50, 60, 70, 80, 90])`
	`22`	`+ [1, 1, 2, 3, 4, 5]`
	`23`	`+ >>> calculate_span([5, 4, 3, 2, 1])`
	`24`	`+ [1, 1, 1, 1, 1]`
	`25`	`+ >>> calculate_span([1, 2, 3, 4, 5])`
	`26`	`+ [1, 2, 3, 4, 5]`
	`27`	`+ >>> calculate_span([10, 20, 30, 40, 50])`
	`28`	`+ [1, 2, 3, 4, 5]`
	`29`	`+ >>> calculate_span([100, 80, 60, 70, 60, 75, 85])`
	`30`	`+ [1, 1, 1, 2, 1, 4, 6]`
	`31`	`+ """`
`12`	`32`	`n = len(price)`
	`33`	`+ s = [0] * n`
`13`	`34`	`# Create a stack and push index of fist element to it`
`14`	`35`	`st = []`
`15`	`36`	`st.append(0)`
`@@ -21,18 +42,20 @@ def calculation_span(price, s):`
`21`	`42`	`for i in range(1, n):`
`22`	`43`	`# Pop elements from stack while stack is not`
`23`	`44`	`# empty and top of stack is smaller than price[i]`
`24`		`- while len(st) > 0 and price[st[0]] <= price[i]:`
	`45`	`+ while len(st) > 0 and price[st[-1]] <= price[i]:`
`25`	`46`	`st.pop()`
`26`	`47`
`27`	`48`	`# If stack becomes empty, then price[i] is greater`
`28`	`49`	`# than all elements on left of it, i.e. price[0],`
`29`	`50`	`# price[1], ..price[i-1]. Else the price[i] is`
`30`	`51`	`# greater than elements after top of stack`
`31`		`- s[i] = i + 1 if len(st) <= 0 else (i - st[0])`
	`52`	`+ s[i] = i + 1 if len(st) <= 0 else (i - st[-1])`
`32`	`53`
`33`	`54`	`# Push this element to stack`
`34`	`55`	`st.append(i)`
`35`	`56`
	`57`	`+ return s`
	`58`	`+`
`36`	`59`
`37`	`60`	`# A utility function to print elements of array`
`38`	`61`	`def print_array(arr, n):`
`@@ -42,10 +65,9 @@ def print_array(arr, n):`
`42`	`65`
`43`	`66`	`# Driver program to test above function`
`44`	`67`	`price = [10, 4, 5, 90, 120, 80]`
`45`		`-S = [0 for i in range(len(price) + 1)]`
`46`	`68`
`47`		`-# Fill the span values in array S[]`
`48`		`-calculation_span(price, S)`
	`69`	`+# Calculate the span values`
	`70`	`+S=calculate_span(price)`
`49`	`71`
`50`	`72`	`# Print the calculated span values`
`51`	`73`	`print_array(S, len(price))`

0 commit comments

Comments

(0)

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Uh oh!

Commit 5c50572

File tree

2 files changed

2 files changed

`‎DIRECTORY.md`

`‎data_structures/stacks/stock_span_problem.py`

0 commit comments