Skip to content

Instantly share code, notes, and snippets.

@vznvzn
Created May 16, 2019 01:50
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/61ce1233f7437b0ec034671b89f8e27d to your computer and use it in GitHub Desktop.
Save vznvzn/61ce1233f7437b0ec034671b89f8e27d to your computer and use it in GitHub Desktop.
def f2(n)
return n.odd? ? (n * 3 + 1) / 2 : n / 2
end
def d(s)
c = s.split('').select { |x| x == '1' }.size
d = c.to_f / s.length
return d
end
def keys(ks2)
ksmn = ks2.select { |x| x.end_with?('-') }
ksmx = ks2 - ksmn
ksmn.each { |x| x.replace(x.chop) }
return {'ks' => ksmn + ksmx, 'ksmx' => ksmx}
end
def order(l, ks2)
l2 = (0...l.size).map { {} }
$k = keys(ks2) if ($k.nil?)
$k['ks'].each \
{
|k|
l1 = (0...l.size).map { |x| {k => l[x][k], 'i' => x} }
s = $k['ksmx'].member?(k) ? -1 : 1
l1.sort_by! { |x| s * x[k] }
(0...l1.size).each { |x| l2[l1[x]['i']][k] = x }
}
l2.each { |x| x['z'] = x.values.inject(0) { |t, x| t + x } }
return (0...l2.size).sort_by { |x| l2[x]['z'] }
end
def out(a, a2 = {}, t = nil)
fn = 'out.txt'
k = a.keys
if ($f1.nil?) then
$f1 = File.open(fn, 'w')
$f1.puts(k.join("\t"))
f2 = File.open('gnuplot.cmd', 'w')
st = t.nil? ? 'unset title; ' : "set title '#{t}'; "
f2.puts("set ytics nomirror; set y2tics; set colors classic; set key top left box opaque; #{st}; plot \\")
(a.keys - ['t']).each \
{
|x|
f2.puts("'#{fn}' using (column('t')):(column('#{x}')) with line title '#{x}' #{a2[x]}, \\")
}
f2.puts
f2.close
end
$f1.puts(a.values_at(*k).join("\t"))
$f1.flush
end
def remove(a, ks)
a = a.dup
ks.each { |x| a.delete(x) }
return a
end
def seq(n)
n1 = n
l = [n]
while (n != 1)
n = f2(n)
l << n
end
return l
end
def sum(l)
return l.inject { |t, x| t + x }
end
def count(ns, x2)
l = seq(ns.to_i(2))
l1 = l.map { |x| x.to_s(2).length }
a, b = [l1[0], l1[-1]]
l2 = []
l1.size.times \
{
|x|
y = a + x.to_f / (l1.size - 1) * (b - a)
dv = (l1[x] - y) / l1[x]
l2 << dv
}
return {'ns' => ns,
'nw' => ns.length,
'dv' => l2.max,
'l1' => l1,
'l2' => l2
}
end
def add(l, ns, x2)
$t = 0 if ($t.nil?)
$t += 1
x = count(ns, x2).merge({'t' => $t})
l << x
x = remove(x, ['l1', 'l2'])
out(remove(x, ['ns']), {'dv' => 'axes x1y2'}) if ($t % $d == 0)
p(x) if ($t % 200 == 0)
end
def adj(ns)
ns1 = ns.dup
ns1[0, 1] = '10'
ns2 = ns.dup
ns2[0, 1] = '11'
return ns1, ns2
end
def opt(ks)
c = 2e3.to_i
$d = c * 2 / 2000
add(l = [], '1', {'cm' => 0, 'cg' => 0})
c.times \
{
l1 = order(l, ks)
i = l1[rand([l1.size, 50].min)]
x = l.delete_at(i)
ns1, ns2 = adj(x['ns'])
add(l, ns1, x)
add(l, ns2, x)
}
l1 = order(l, ks)
return l[l1[0]]
end
ks = ['dv']
x = opt(ks)
l1 = x['l1']
l2 = x['l2']
f = File.open('out2.txt', 'w')
l1.size.times { |x| f.puts([l1[x], l2[x]].join("\t")) }
f.close
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment