m-chrzan.xyz
aboutsummaryrefslogtreecommitdiff
path: root/23/b.lua
diff options
context:
space:
mode:
Diffstat (limited to '23/b.lua')
-rw-r--r--23/b.lua99
1 files changed, 99 insertions, 0 deletions
diff --git a/23/b.lua b/23/b.lua
new file mode 100644
index 0000000..6d504b0
--- /dev/null
+++ b/23/b.lua
@@ -0,0 +1,99 @@
+Node = {}
+Node.__index = Node
+function Node:new(value)
+ object = {}
+ object.value = value
+ object.right = nil
+ object.left = nil
+ return setmetatable(object, self)
+end
+
+function Node:insert(left)
+ self.left = left
+ self.right = left.right
+ self.right.left = self
+ self.left.right = self
+end
+
+function Node:detach()
+ self.left.right = self.right
+ self.right.left = self.left
+ self.left = nil
+ self.right = nil
+ return self
+end
+
+CyclicList = {}
+CyclicList.__index = CyclicList
+
+function CyclicList:new()
+ object = {}
+ object.head = nil
+ object.last = nil
+ object.index = {}
+ return setmetatable(object, self)
+end
+
+function CyclicList:insert(value)
+ node = Node:new(value)
+ self.index[value] = node
+ if self.head == nil then
+ self.head = node
+ node.left = node
+ node.right = node
+ else
+ node:insert(self.head.left)
+ end
+end
+
+function CyclicList:move_head(node)
+ self.head = node
+end
+
+input = '716892543'
+cups = CyclicList:new()
+for i = 1, #input do
+ cups:insert(tonumber(input:sub(i,i)))
+end
+for i = 10, 1000000 do
+ cups:insert(i)
+end
+
+function next_target(n)
+ n = n - 1
+ if n == 0 then
+ n = 1000000
+ end
+ return n
+end
+
+for round = 1, 10000000 do
+ current_value = cups.head.value
+
+ next_node = cups.head.right
+ next_values = {}
+ for i = 1, 3 do
+ next_values[next_node.value] = true
+ next_node = next_node.right
+ end
+
+ target = next_target(current_value)
+ while next_values[target] do
+ target = next_target(target)
+ end
+
+ target_node = cups.index[target]
+
+ for i = 1, 3 do
+ moved_node = cups.head.right:detach()
+ moved_node:insert(target_node)
+ target_node = moved_node
+ end
+
+ cups:move_head(cups.head.right)
+end
+
+one_node = cups.index[1]
+a = one_node.right.value
+b = one_node.right.right.value
+print(a * b)