Created
July 28, 2020 23:45
-
-
Save vznvzn/93c0dd7333d0bb84f07883be1cba4a47 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
def f2(n) | |
return n.odd? ? (n * 3 + 1) / 2 : n / 2 | |
end | |
def seq(n) | |
l = [n] | |
while (n != 1) | |
n = f2(n) | |
l << n | |
end | |
return l | |
end | |
def addgap(a, b) | |
d = b - a | |
$g[d] = $g.fetch(d, []) + [[a, b]] | |
$mx = [$mx, d].compact.max | |
end | |
def rmgap(a, b) | |
d = b - a | |
$g[d] -= [[a, b]] | |
if ($g[d].empty?) then | |
$g.delete(d) | |
$mx = $g.keys.max if (d == $mx) | |
end | |
end | |
def gap(z) | |
i = (0...$x.size).bsearch { |x| $x[x] >= z } | |
if (i.nil?) then | |
$x << z | |
return if ($x.size == 1) | |
addgap(*$x[-2, 2]) | |
return | |
end | |
return if ($x[i] == z) | |
rmgap(*$x[i - 1, 2]) if (i > 0) | |
$x[i, 0] = z | |
addgap(*$x[i, 2]) | |
addgap(*$x[i - 1, 2]) if (i > 0) | |
end | |
def gapmx(z) | |
gap(z) | |
return $mx.nil? ? z : $mx | |
end | |
def count(ns, l2 = []) | |
l = seq(n = ns.to_i(2)) | |
cg = l.index { |x| x < l[0] } | |
cg = l.size - 1 if (cg.nil?) | |
cm = (0..cg).max_by { |x| l[x] } | |
c = l.size | |
return {'ns' => ns, | |
'cm' => cm, | |
'c' => c, | |
'l' => l | |
}.merge({'g' => gapmx(c)}) | |
end | |
def out(fn, a, d, a2 = {}, t = nil) | |
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}')) every #{d} with line title '#{x}' #{a2[x]}, \\") | |
} | |
f2.puts | |
f2.close | |
end | |
$f1.puts(a.values_at(*k).join("\t")) | |
$f1.flush | |
end | |
def rank(x) | |
return -x['z'] | |
end | |
def adj(x, wt, km) | |
z = 0 | |
km.keys.each { |k| z += (x[k] - wt["#{k}_a"]) / wt["#{k}_sd"] * km[k] } | |
x['z'] = z | |
raise [x, wt].to_s if (!z.finite?) | |
return x | |
end | |
def fmt(a) | |
['ns', 'l'].each { |x| a.delete(x) } | |
return a | |
end | |
def add(bs, km, k, h, ns, l1, wt, t, d) | |
j = ns.length | |
h[j] = [] if (!h.member?(j)) | |
x = adj(count(ns), wt, km) | |
a, b = [h[j][0], h[j][-1]] | |
return if (h[j].map { |x| x['ns'] }.member?(x['ns'])) | |
x['t'] = t | |
h[j] << x | |
h[j].sort_by! { |x| rank(x) } | |
h[j][bs..-1] = [] if (h[j].size > bs) | |
out('out.txt', fmt(x.dup), d, {k => 'axis x1y2', 'z' => 'axes x1y2'}) | |
l1 << x if (!l1.nil?) | |
end | |
def init(w, m) | |
return '1' * w | |
end | |
def initd(w, m) | |
d = m.to_f / w | |
w2 = w - 1 | |
a = (0...w2).to_a | |
s = '0' * w2 | |
(1..(d * w - 1)).map { a.delete_at(rand(a.size)) }.each { |x| s[x, 1] = '1' } | |
return ('1' + s) | |
end | |
def initb(w, m) | |
r = m.to_f / w | |
return '1' + (1...w).map { rand < r ? '0' : '1' }.join | |
end | |
def initw(w, m) | |
s = initd(w, w / 2) | |
i = rand(w - m) + 1 | |
m = [w - i, m].max | |
s[i, m] = rand(2).to_s * m | |
return s | |
end | |
def mut(ns) | |
ns = ns.dup | |
i = rand(ns.length - 1) + 1 | |
ns[i, 1] = (ns[i, 1].to_i ^ 1).to_s | |
return ns | |
end | |
def inner(ns1, ns2) | |
a, b = [ns1, ns2].sort_by { |x| x.length } | |
d = b.length - a.length | |
i = rand(d + 1) | |
c = b.dup | |
c[i, a.length] = a | |
ns1.replace(b) | |
ns2.replace(c) | |
end | |
def mix(ns1, ns2) | |
ns1 = ns1.dup | |
ns2 = ns2.dup | |
inner(ns1, ns2) | |
return (0...ns1.length).map { |x| [ns1, ns2][rand(2)][x, 1] }.join | |
end | |
def cut(ns1, ns2) | |
ns1 = ns1.dup | |
ns2 = ns2.dup | |
inner(ns1, ns2) | |
i = rand(ns1.length) | |
return ns1[0...i] + ns2[i..-1] | |
end | |
def ext(ns) | |
ns = ns.dup | |
(rand(3) + 1).times { ns[rand(ns.length) + 1, 0] = rand(2).to_s } | |
return ns | |
end | |
def short(ns) | |
return ns if (ns.length == 1) | |
ns = ns.dup | |
(rand([ns.length - 1, 3].min) + 1).times { ns[rand(ns.length - 1) + 1, 1] = '' } | |
return ns | |
end | |
def top(h1) | |
return h1[rand([h1.size / 4, h1.size].max)] | |
end | |
def stat(l) | |
return 0, 1.0 if (l.empty?) | |
t = l.inject { |a, x| a + x } | |
t2 = l.inject(0) { |a, x| a + x ** 2 } | |
c = l.size | |
a = t.to_f / c | |
z = t2.to_f / c - a ** 2 | |
sd = Math.sqrt(z < 0 ? 0 : z) | |
raise if (sd.nan?) | |
return a, sd | |
end | |
def dist(km, h, wt) | |
l = h.values.flatten | |
km.keys.each \ | |
{ | |
|k| | |
a, sd = stat(l.map { |x| x[k] }) | |
sd = 1.0 if (sd == 0) | |
wt.merge!({"#{k}_a" => a, "#{k}_sd" => sd}) | |
} | |
end | |
def nonnil(x, y) | |
return x.nil? ? y : x | |
end | |
def opt(km, w1, k, l1, n) | |
h = {} | |
ops = ['init0', 'initd0', 'initb0', 'initw0', 'mut1', 'mix2', 'cut2'] | |
# ops += ['ext1', 'short1'] | |
bs = 1000 | |
bc = 50 | |
t = 0 | |
d = n / 2000 | |
d = 1 if (d == 0) | |
wt = {} | |
n.times \ | |
{ | |
if (t % 100 == 0) then | |
dist(km, h, wt) | |
h.values.flatten.each { |x| adj(x, wt, km) } | |
end | |
opc = ops[rand(ops.size)] | |
c = opc[-1..-1].to_i | |
op = opc[0..-2] | |
ks = h.keys.sort_by { |x| rank(h[x][0]) } | |
ks = ks[0...[ks.size, ks.size / 4].max] | |
case (c) | |
when 0 | |
w = ks.empty? ? w1 : ks[rand(ks.size)] | |
ns = send(op, w, rand(w)) | |
when 1, 2 | |
redo if (h.empty?) | |
h.delete(ks.pop) if (ks.size > bc) | |
arg = (0...c).map { ks[rand(ks.size)] } | |
l = arg.map { |x| top(h[x])['ns'] } | |
ns = send(op, *l) | |
ns[0, 1] = '1' | |
end | |
add(bs, km, k, h, ns, l1, wt, t, d) | |
if (t % 100 == 0) then | |
kmx = nonnil(h.values.flatten.max_by { |x| x[k] }, {}).dup | |
h0 = nonnil(h.fetch(ks[0], [])[0], {}).dup | |
k0 = "#{k}_0" | |
msg = Hash[[['t', 'hmn', 'hmx', 'hc', 'h_0', k0], [t, ks.min, ks.max, h.size, h0, kmx]].transpose] | |
['h_0', k0].each { |x| fmt(msg[x]) } | |
$stderr.puts(msg) | |
end | |
t += 1 | |
} | |
return h | |
end | |
def outa(f, l, a = {}, t = '') | |
f.puts('$dat << eof') | |
k = l[0].keys | |
f.puts(k.join("\t")) | |
l.each { |x| f.puts(x.values.join("\t")) } | |
f.puts('eof') | |
f.puts("set colors classic; set key top left opaque; set title '#{t}'; ") | |
# f.puts("set ytics nomirror; set y2tics;") | |
f.puts("plot \\") | |
k, ct = [k - ['t'], "(column('t')):"] if (k.member?('t')) | |
k.each \ | |
{ | |
|x| | |
next if (a.member?(x) && a[x].nil?) | |
opt = a.fetch(x, '') | |
opt += ' lw 2 ' if (!opt.include?('lw')) | |
opt += ' with line ' if (!opt.include?('with') && !opt.include?('pt')) | |
f.puts("'$dat' using #{ct}(column('#{x}')) #{opt} title '#{x}',\\") | |
} | |
f.puts | |
# f.puts("reset; pause -1;") | |
end | |
def outafn(l, a = {}, fn = nil) | |
outa(f = File.open(fn.nil? ? 'gnuplot.cmd' : fn, 'w'), l, a) | |
f.close | |
end | |
def d(s) | |
c = s.split('').select { |x| x == '1' }.size | |
d = c.to_f / s.length | |
return d | |
end | |
def len(ns, p) | |
l = ns.split(p) | |
l = [] if (l.nil?) | |
l.shift if (l[0] == '') | |
return l.map { |x| x.length } | |
end | |
def len01(ns) | |
return len(ns, /0+/), len(ns, /1+/) | |
end | |
def e(ns) | |
return len01(ns).flatten.size.to_f / ns.length | |
end | |
def fmt1(l, f) | |
w = l.max_by { |x| x.length }.length | |
f.puts("set colors classic; set ytics nomirror; set y2tics; set key top left opaque;\\") | |
f.puts("plot \\") | |
f.puts("[0:#{l.size - 1}][0:#{w}] '-' matrix using 2:1:3 with image title '', \\") | |
f.puts("'-' using (column('d')) with line axes x1y2 dt 2 lw 3, \\") | |
f.puts("'-' using 1 with line title 'e' axes x1y2 dt 2 lw 3") | |
l1 = [] | |
l.each_with_index \ | |
{ | |
|z, i| | |
l1 << [d(z), e(z)] | |
c = w - z.length | |
z[z.length...w] = '0' * c | |
f.puts(z.split('').join(' ')) | |
} | |
f.puts("eof") | |
f.puts("eof") | |
f.puts("d") | |
l1.each { |x| f.puts(x[0]) } | |
f.puts("eof") | |
l1.each { |x| f.puts(x[1]) } | |
f.puts("eof") | |
end | |
def grid(l) | |
f = File.open(fn = "gnuplot3.cmd", 'w') | |
f.puts("set palette negative grayscale; unset colorbox;") | |
fmt1(l.map { |x| x.reverse }, f) | |
f.close | |
end | |
def sample(l, c) | |
return l if (l.size < c) | |
r = l.size / c | |
l2 = *(0...l.size).select { |x| x % r == 0 } | |
l2[c..-1] = [] | |
return l.values_at(*l2) | |
end | |
def g(x) | |
gapmx(x) | |
p({'x' => x, '$x' => $x, '$g' => $g, '$mx' => $mx}) | |
end | |
def testgap() | |
[5, 6, 1, 2, 9, 1, 3].each { |x| g(x) } | |
exit | |
end | |
$x = [] | |
$g = {} | |
$mx = nil | |
# testgap() | |
c = 2e3 | |
h = opt({'c' => 1, 'g' => -1}, 200, nil, l1 = [], c.to_i) | |
outafn(h.values[0].map { |x| fmt(x.dup) }.sort_by { |x| x['t']}, {}, 'gnuplot1.cmd') | |
a = {'t1' => 'pt 5 ps 0.25'} | |
a = {'t1' => nil} | |
outafn(l1.sort_by! { |x| x['c'] }.map { |x| x1 = fmt(x.dup); x1['t1'] = x1.delete('t'); x1 }, a, 'gnuplot2.cmd') | |
grid(l = sample(l1, 250).map { |x| x['ns'] }) | |
f = File.open('out2.txt', 'w') | |
l.each_with_index \ | |
{ | |
|ns, j| | |
c = 0 | |
ns.reverse.split('').each_with_index { |x1, i| c += [-1, 1][x1.to_i]; f.puts([i, c, j].join("\t")) } | |
f.puts | |
} | |
f.close | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment