|
| 1 | +use "collections" |
| 2 | + |
| 3 | +// based on 1.go and 1.zig |
| 4 | +// multi-threaded implementation (Go/Zig based) |
| 5 | + |
| 6 | +actor Main |
| 7 | + new create(env: Env) => |
| 8 | + let n: USize = try env.args(1)?.usize()? else 12 end |
| 9 | + let coordinator = Coordinator(env, n) |
| 10 | + coordinator.start() |
| 11 | + |
| 12 | +class val Result |
| 13 | + let max_flips: USize |
| 14 | + let check_sum: ISize |
| 15 | + |
| 16 | + new create(mf: USize, cs: ISize) => |
| 17 | + max_flips = mf |
| 18 | + check_sum = cs |
| 19 | + |
| 20 | +actor Coordinator |
| 21 | + let _env: Env |
| 22 | + let _n: USize |
| 23 | + let _workers: USize = 4 |
| 24 | + var _results_received: USize = 0 |
| 25 | + var _max_flips: USize = 0 |
| 26 | + var _check_sum: ISize = 0 |
| 27 | + let _fact: Array[USize] val |
| 28 | + let _chunks: USize = 720 |
| 29 | + var _chunk_size: USize = 0 |
| 30 | + var _tasks: USize = 0 |
| 31 | + |
| 32 | + new create(env: Env, n: USize) => |
| 33 | + _env = env |
| 34 | + _n = n |
| 35 | + _fact = recover val |
| 36 | + var a = Array[USize](_n + 1) |
| 37 | + var prev: USize = 1 |
| 38 | + a.push(prev) |
| 39 | + for i in Range[USize](1, _n + 1) do |
| 40 | + prev = prev * i |
| 41 | + a.push(prev) |
| 42 | + end |
| 43 | + a |
| 44 | + end |
| 45 | + |
| 46 | + try |
| 47 | + let factn = _fact(_n)? |
| 48 | + let t1 = factn + _chunks |
| 49 | + let t2 = t1 - 1 |
| 50 | + _chunk_size = t2 / _chunks |
| 51 | + _chunk_size = _chunk_size + (_chunk_size % 2) |
| 52 | + let t3 = factn + _chunk_size |
| 53 | + let t4 = t3 - 1 |
| 54 | + _tasks = t4 / _chunk_size |
| 55 | + end |
| 56 | + |
| 57 | + be start() => |
| 58 | + for i in Range[USize](0, _tasks) do |
| 59 | + let worker = Worker(this, _n, _fact, i * _chunk_size, _chunk_size) |
| 60 | + worker.compute() |
| 61 | + end |
| 62 | + |
| 63 | + be result(r: Result val) => |
| 64 | + _results_received = _results_received + 1 |
| 65 | + if r.max_flips > _max_flips then |
| 66 | + _max_flips = r.max_flips |
| 67 | + end |
| 68 | + _check_sum = _check_sum + r.check_sum |
| 69 | + |
| 70 | + if _results_received == _tasks then |
| 71 | + _env.out.print(_check_sum.string()) |
| 72 | + _env.out.print("Pfannkuchen(" + _n.string() + ") = " + _max_flips.string()) |
| 73 | + end |
| 74 | + |
| 75 | +actor Worker |
| 76 | + let _coordinator: Coordinator |
| 77 | + let _n: USize |
| 78 | + let _fact: Array[USize] val |
| 79 | + let _idx_min: USize |
| 80 | + let _chunk_size: USize |
| 81 | + |
| 82 | + new create(coordinator: Coordinator, n: USize, fact: Array[USize] val, |
| 83 | + idx_min: USize, chunk_size: USize) => |
| 84 | + _coordinator = coordinator |
| 85 | + _n = n |
| 86 | + _fact = fact |
| 87 | + _idx_min = idx_min |
| 88 | + _chunk_size = chunk_size |
| 89 | + |
| 90 | + be compute() => |
| 91 | + try |
| 92 | + var idx_max = _idx_min + _chunk_size |
| 93 | + if idx_max > _fact(_n)? then |
| 94 | + idx_max = _fact(_n)? |
| 95 | + end |
| 96 | + |
| 97 | + let p = Array[USize](_n) |
| 98 | + let pp = Array[USize](_n) |
| 99 | + let count = Array[USize](_n) |
| 100 | + |
| 101 | + // Initialize arrays |
| 102 | + for i in Range[USize](0, _n) do |
| 103 | + p.push(i) |
| 104 | + pp.push(0) |
| 105 | + count.push(0) |
| 106 | + end |
| 107 | + |
| 108 | + // Generate first permutation |
| 109 | + var idx = _idx_min |
| 110 | + var i = _n - 1 |
| 111 | + while i > 0 do |
| 112 | + let d = idx / _fact(i)? |
| 113 | + count(i)? = d |
| 114 | + idx = idx % _fact(i)? |
| 115 | + |
| 116 | + // Copy p to pp |
| 117 | + for j in Range[USize](0, _n) do |
| 118 | + pp(j)? = p(j)? |
| 119 | + end |
| 120 | + |
| 121 | + // Rotate |
| 122 | + for j in Range[USize](0, i + 1) do |
| 123 | + if (j + d) <= i then |
| 124 | + p(j)? = pp(j + d)? |
| 125 | + else |
| 126 | + p(j)? = pp((j + d) - i - 1)? |
| 127 | + end |
| 128 | + end |
| 129 | + i = i - 1 |
| 130 | + end |
| 131 | + |
| 132 | + var max_flips: USize = 1 |
| 133 | + var check_sum: ISize = 0 |
| 134 | + var sign = true |
| 135 | + idx = _idx_min |
| 136 | + |
| 137 | + while idx < idx_max do |
| 138 | + // Count flips |
| 139 | + let first = p(0)? |
| 140 | + if first != 0 then |
| 141 | + var flips: USize = 1 |
| 142 | + if p(first)? != 0 then |
| 143 | + // Copy p to pp |
| 144 | + for ii in Range[USize](0, _n) do |
| 145 | + pp(ii)? = p(ii)? |
| 146 | + end |
| 147 | + |
| 148 | + var p0 = first |
| 149 | + while pp(p0)? != 0 do |
| 150 | + flips = flips + 1 |
| 151 | + // Reverse |
| 152 | + var ii: USize = 1 |
| 153 | + var j = p0 - 1 |
| 154 | + while ii < j do |
| 155 | + let tmp = pp(ii)? |
| 156 | + pp(ii)? = pp(j)? |
| 157 | + pp(j)? = tmp |
| 158 | + ii = ii + 1 |
| 159 | + j = j - 1 |
| 160 | + end |
| 161 | + let t = pp(p0)? |
| 162 | + pp(p0)? = p0 |
| 163 | + p0 = t |
| 164 | + end |
| 165 | + end |
| 166 | + |
| 167 | + if flips > max_flips then |
| 168 | + max_flips = flips |
| 169 | + end |
| 170 | + |
| 171 | + if sign then |
| 172 | + check_sum = check_sum + flips.isize() |
| 173 | + else |
| 174 | + check_sum = check_sum - flips.isize() |
| 175 | + end |
| 176 | + end |
| 177 | + |
| 178 | + idx = idx + 1 |
| 179 | + if idx == idx_max then break end |
| 180 | + |
| 181 | + // Generate next permutation |
| 182 | + if sign then |
| 183 | + let tmp = p(0)? |
| 184 | + p(0)? = p(1)? |
| 185 | + p(1)? = tmp |
| 186 | + else |
| 187 | + let tmp = p(1)? |
| 188 | + p(1)? = p(2)? |
| 189 | + p(2)? = tmp |
| 190 | + |
| 191 | + var k: USize = 2 |
| 192 | + while true do |
| 193 | + let c = count(k)? + 1 |
| 194 | + count(k)? = c |
| 195 | + if c <= k then break end |
| 196 | + |
| 197 | + count(k)? = 0 |
| 198 | + // rotate left p[0..k] |
| 199 | + let p0 = p(0)? |
| 200 | + for j in Range[USize](0, k) do |
| 201 | + p(j)? = p(j + 1)? |
| 202 | + end |
| 203 | + p(k)? = p0 |
| 204 | + k = k + 1 |
| 205 | + end |
| 206 | + end |
| 207 | + sign = not sign |
| 208 | + end |
| 209 | + |
| 210 | + _coordinator.result(Result(max_flips, check_sum)) |
| 211 | + end |
0 commit comments