Skip to content

Instantly share code, notes, and snippets.

@vznvzn
Created December 3, 2016 02:36
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vznvzn/2d5f47f48058b2589550a133b8ec993d to your computer and use it in GitHub Desktop.
Save vznvzn/2d5f47f48058b2589550a133b8ec993d to your computer and use it in GitHub Desktop.
def d(s)
c = s.split('').select { |x| x == '1' }.size
d = c.to_f / s.length
return d
end
def stat(l)
return 0, 0, 0, 0 if (l.empty?)
t = t2 = 0
l.each \
{
|x|
t += x
t2 += x ** 2
}
c = l.size
a = t.to_f / c
z = t2.to_f / c - a ** 2
sd = Math.sqrt(z < 0 ? 0 : z)
return a, sd, l.max.to_f, l.min.to_f
end
def mono(l)
l = [] if (l.nil?)
m = l[0]
mx = nil
j = 0
l2 = []
(1...l.size).each \
{
|i|
if (l[i] < m) then
l2 << [j.to_f / l.size, (i - j).to_f / l.size] if (i - j > 1)
m, j = [l[i], i]
end
mx = [mx, i - j].compact.max
}
mx = 0 if (mx.nil?)
return {'mx' => mx, 'ls3' => l.size, 'l2' => l2}
end
def adv(x)
n1 = n = x['n']
ns = n.to_s(2)
x['d'] = d(ns)
l = [n]
c = 0
while (n >= n1 && n != 1)
n = (n * 3 + 1) / 2 while (n.odd?)
while (n.even?)
c += 1
n /= 2
end
l << n
end
j = (0...l.size).max_by { |x| l[x] }
x['ns'] = ns.length
x['ls'] = l.size
x['c'] = c
x['ns2'] = c - ns.length
return x.merge(mono(l[j..-1]))
end
def next2(z)
l = [z]
p = z['p'] + 1
l << adv({'n'=>z['n'] + 2**p, 'p'=>p})
l << z.merge({'p'=>p})
return l
end
def insert(l, x)
l << x
end
def delete(l, j)
z = l.delete_at(j)
return z
end
def sum(l)
t = 0
l.each { |x| t += x }
return t
end
def rank(l, w1)
l2 = []
l.size.times \
{
|x|
l1 = l[x][1].values_at(*w1)
a = {}
w1.size.times \
{
|i|
a[w1[i]] = l1[i]
}
l2 << [x, a]
}
w1.each \
{
|k|
l1 = l2.map { |x| x[1][k] }
a, sd, = stat(l1)
sd = 1 if (sd == 0)
l2.each \
{
|x|
x[1][k + 'b'] = (x[1][k] - a) / sd
}
}
b = w1.map { |x| x + 'b' }
l2.each \
{
|x|
x[1]['z'] = sum(x[1].values_at(*b))
}
l2.sort_by! { |x| [x[1]['z'], x[1]['d']] }
l2.reverse!
a = l2[0][0]
j = l2[rand([l2.size, 100].min)][0]
k = l2[-1][0]
return j, k, l[a][1]
end
def opt(w, w1, c)
l = []
insert(l, next2({'n'=>1, 'p'=>0}))
puts('# ' + Time.now.to_s)
t = Time.now.to_i
c.times \
{
|i|
$stderr.puts([i, sprintf('%.1fm', (Time.now.to_i - t) / 60.0), Time.now.to_s].join("\t")) if (i % 100 == 0)
j, k, a = rank(l, w1)
if (l.size > 1000) then
z2 = delete(l, [j, k].max)
z1 = delete(l, [j, k].min)
z = j < k ? z1 : z2
else
z = delete(l, j)
end
# p(z[1]['l2'])
z[1]['l2'].each \
{
|x, y|
puts([x, 0].join("\t"))
puts([x, y].join("\t"))
puts
}
insert(l, next2(z[1]))
insert(l, next2(z[2]))
$stdout.flush
}
puts('# ' + Time.now.to_s)
end
def test(w)
puts(w.join("\t"))
File.open('db.txt').readlines.map { |x| Kernel.eval(x) }.each_with_index \
{
|x, i|
puts(adv(x).values_at(*w).join("\t"))
puts if ((i + 1) % 100 == 0)
}
end
w = ['ns', 'ls', 'c', 'ns2']
w1 = w
# test(w); exit
opt(w, w1, 2000)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment