345 Matrix Sum -- v3.sf 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. #!/usr/bin/ruby
  2. # Author: Trizen
  3. # Date: 24 March 2022
  4. # https://github.com/trizen
  5. # Matrix Sum
  6. # https://projecteuler.net/problem=345
  7. func matrix_sum(arr, size, callback, value=0, sum=0, columns = Hash()) is cached {
  8. var (i, j) = divmod(value, size)
  9. for k in (^size) {
  10. next if columns.has(k)
  11. callback(var new_sum = sum+arr[i*size + j])
  12. if ((i+1 < size) && !((columns+j).has(k))) {
  13. __FUNC__(arr, size, callback, (i+1)*size + k, new_sum, columns+j)
  14. }
  15. }
  16. if ((j+1 < size) && !columns.has(j+1)) {
  17. __FUNC__(arr, size, callback, i*size + (j+1), sum, columns)
  18. }
  19. }
  20. var arr = [
  21. 7 53 183 439 863
  22. 497 383 563 79 973
  23. 287 63 343 169 583
  24. 627 343 773 959 943
  25. 767 473 103 699 303
  26. ]
  27. var max = 0
  28. matrix_sum(arr, arr.len.isqrt, {|sum|
  29. if (sum > max) {
  30. max = sum
  31. say "Best sum so far: #{sum}"
  32. }
  33. })
  34. say "Best sum: #{max}"
  35. arr = <<'EOT'.lines.map { .nums }.flat if true
  36. 7 53 183 439 863 497 383 563 79 973 287 63 343 169 583
  37. 627 343 773 959 943 767 473 103 699 303 957 703 583 639 913
  38. 447 283 463 29 23 487 463 993 119 883 327 493 423 159 743
  39. 217 623 3 399 853 407 103 983 89 463 290 516 212 462 350
  40. 960 376 682 962 300 780 486 502 912 800 250 346 172 812 350
  41. 870 456 192 162 593 473 915 45 989 873 823 965 425 329 803
  42. 973 965 905 919 133 673 665 235 509 613 673 815 165 992 326
  43. 322 148 972 962 286 255 941 541 265 323 925 281 601 95 973
  44. 445 721 11 525 473 65 511 164 138 672 18 428 154 448 848
  45. 414 456 310 312 798 104 566 520 302 248 694 976 430 392 198
  46. 184 829 373 181 631 101 969 613 840 740 778 458 284 760 390
  47. 821 461 843 513 17 901 711 993 293 157 274 94 192 156 574
  48. 34 124 4 878 450 476 712 914 838 669 875 299 823 329 699
  49. 815 559 813 459 522 788 168 586 966 232 308 833 251 631 107
  50. 813 883 451 509 615 77 281 613 459 205 380 274 302 35 805
  51. EOT
  52. var max = 0
  53. matrix_sum(arr, arr.len.isqrt, {|sum|
  54. if (sum > max) {
  55. max = sum
  56. say "Best sum so far: #{sum}"
  57. }
  58. })
  59. say "Best sum: #{max}"