Commitebc8de4

siderism

authored

Add sum of sub squares of a given nxn matrix (keon#631)

* Added function to calculate sum of all sub-squares in a square matrix* Added test for sum_sub_squares function* Updated README

1 parent38173ae commitebc8de4Copy full SHA for ebc8de4

File tree

+42

-1

lines changed

+42

-1

lines changed

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -236,6 +236,7 @@ If you want to uninstall algorithms, it is as simple as:`
`236`	`236`	`-[spiral_traversal](algorithms/matrix/spiral_traversal.py)`
`237`	`237`	`- [crout_matrix_decomposition](algorithms/matrix/crout_matrix_decomposition.py)`
`238`	`238`	`- [cholesky_matrix_decomposition](algorithms/matrix/cholesky_matrix_decomposition.py)`
	`239`	`+- [sum_sub_squares](algorithms/matrix/sum_sub_squares.py)`
`239`	`240`	`-[queues](algorithms/queues)`
`240`	`241`	`-[max_sliding_window](algorithms/queues/max_sliding_window.py)`
`241`	`242`	`-[moving_average](algorithms/queues/moving_average.py)`

Lines changed: 24 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,24 @@`
	`1`	`+# Function to find sum of all`
	`2`	`+# sub-squares of size k x k in a given`
	`3`	`+# square matrix of size n x n`
	`4`	`+defsum_sub_squares(matrix,k):`
	`5`	`+n=len(matrix)`
	`6`	`+result= [[0foriinrange(k)]forjinrange(k)]`
	`7`	`+`
	`8`	`+ifk>n:`
	`9`	`+return`
	`10`	`+foriinrange(n-k+1):`
	`11`	`+l=0`
	`12`	`+forjinrange(n-k+1):`
	`13`	`+sum=0`
	`14`	`+`
	`15`	`+# Calculate and print sum of current sub-square`
	`16`	`+forpinrange(i,k+i):`
	`17`	`+forqinrange(j,k+j):`
	`18`	`+sum+=matrix[p][q]`
	`19`	`+`
	`20`	`+result[i][l]=sum`
	`21`	`+l+=1`
	`22`	`+`
	`23`	`+returnresult`
	`24`	`+`

Lines changed: 17 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,8 @@`
`7`	`7`	`rotate_image,`
`8`	`8`	`sparse_dot_vector,`
`9`	`9`	`spiral_traversal,`
`10`		`-sudoku_validator`
	`10`	`+sudoku_validator,`
	`11`	`+sum_sub_squares`
`11`	`12`	`)`
`12`	`13`	`importunittest`
`13`	`14`
`@@ -272,6 +273,21 @@ def test_sudoku_validator(self):`
`272`	`273`	`[3,0,0,4,8,1,1,7,9]`
`273`	`274`	`]))`
`274`	`275`
	`276`	`+classTestSumSubSquares(unittest.TestCase):`
	`277`	`+"""[summary]`
	`278`	`+ Test for the file sum_sub_squares.py`
	`279`	`+`
	`280`	`+ Arguments:`
	`281`	`+ unittest {[type]} -- [description]`
	`282`	`+ """`
	`283`	`+deftest_sum_sub_squares(self):`
	`284`	`+mat= [[1,1,1,1,1],`
	`285`	`+ [2,2,2,2,2],`
	`286`	`+ [3,3,3,3,3],`
	`287`	`+ [4,4,4,4,4],`
	`288`	`+ [5,5,5,5,5]]`
	`289`	`+self.assertEqual(sum_sub_squares.sum_sub_squares(mat,3),`
	`290`	`+ [[18,18,18], [27,27,27], [36,36,36]])`
`275`	`291`
`276`	`292`	`if__name__=="__main__":`
`277`	`293`	`unittest.main()`

Comments

(0)