Skip to content

Instantly share code, notes, and snippets.

@vznvzn
Created December 3, 2016 02:33
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/ffa9fe0af42740c6b3c8c58eea0e6a4b to your computer and use it in GitHub Desktop.
Save vznvzn/ffa9fe0af42740c6b3c8c58eea0e6a4b 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 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
x['ns'] = ns.length
x['ls'] = l.size
x['c'] = c
x['ns2'] = c - ns.length
x['ns2a'] = x['ns2'].abs
return x
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]['ls'], x[1]['d']] }
l2.reverse!
j = l2[rand([l2.size, 100].min)][0]
k = l2[-1][0]
return j, k
end
def opt(w, w1, c)
l = []
insert(l, next2({'n'=>1, 'p'=>0}))
puts('# ' + Time.now.to_s)
puts(w.join("\t"))
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 = 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
insert(l, next2(z[1]))
insert(l, next2(z[2]))
puts([z[1].values_at(*w)].join("\t"))
$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 = ['ns', 'ls', 'c', 'ns2']
# test(w); exit
opt(w, w1, 20000)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment