bitwise56.rb

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) }
	
	x = {'ks' => ksmn + ksmx, 'ksmx' => ksmx, 'ksmn' => ksmn}
	
	$stderr.puts(x.inspect)

	return x
	
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, ct = "(column('t')):", w = ' with line ')

	fn = $fn.nil? ? 'out.txt' : $fn
	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|
			next if (a2.member?(x) && a2[x].nil?)
			m = a2.fetch(x, '')
			m += w if (!m.include?('with'))
			f2.puts("'#{fn}' using #{ct}(column('#{x}')) title '#{x}' #{m}, \\")
		}
		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 = []
	begin
		l << n
		n = f2(n)
	end while (n != 1)
	
	return l
end

def seq2(n)

	n1 = n
	l = []
	begin
		l << n
		n = f2(n)
	end while (n != 1 && n >= n1)
	
	return l
end

def sum(l)
	return l.inject { |t, x| t + x }
end

def avg(l)
	return l.empty? ? 0 : sum(l).to_f / l.size
end

def count(ns)

	nw = ns.length

	l = seq(n = ns.to_i(2))

	cm1 = (0...l.size).max_by { |x| l[x] }
	cg = l.index { |x| x < l[0] }
	cg = l.size - 1 if (cg.nil?)
	
	cg1 = l.rindex { |x| x >= l[0] }
	cm = (0..cg).max_by { |x| l[x] }
	c = l.size

	da = (0.5 - d(ns)).abs
	
	return {
		'ns' => ns, 
		'n' => n,
		'd' => d(ns),
		'l' => l,
		'cg' => cg, 
		'cg1' => cg1,
		'c' => c, 
		'cm1' => cm1, 
		'cm' => cm,
		'nw' => nw, 
		'da' => da
		}
end

def fmt(x)
	return remove(x, ['l'])
end

def axes2(x2)
	return x2.inject({}) { |h, x| h.merge({x => 'axes x1y2' }) }
	
end

def edge(x, x2, f)
	$a = [] if ($a.nil?)
	
	$a << x
	
	return if (!f)

	mn = {}
	mx = {}
	$a[0].keys.each \
	{
		|k|
		l = $a.map { |x| x[k] }
		mn[k] = l.min
		mx[k] = l.max
	}

	$a.clear


	out(mn, x2)
	out(mx, x2)
end

def add(l, ns, x2)
	
	$t = 0 if ($t.nil?)
	$t += 1
	
	x = count(ns).merge({'t' => $t})
	l << x
	
	x = fmt(x)
	
	edge(remove(x, ['ns']), x2, $t % $d == 0)
		
	$stderr.puts(x.inspect) 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, x2)

	$d = c / 2000
	$d = 1 if ($d == 0)
	
	add(l = [], '1', x2)

	(c / 2).times \
	{
		l1 = order(l, ks)

		i = l1[rand([l1.size, 50].min)]
		j = l1[-1]
		
		x = l.delete_at(i)

		l.delete_at(j < i ? j : j - 1) if (l.size > 2000)
		
		ns1, ns2 = adj(x['ns'])
		
		add(l, ns1, x2)
		add(l, ns2, x2)
		
	}

	return l[order(l, ks)[0]]
end

ks = ['nw', 'c', 'cm', 'cm1', 'cg', 'cg1', 'da-']

x2 = axes2(['d'])
	.merge({'n' => nil})

x = opt(ks, 1e4.to_i, x2)
$stderr.puts("\n" + fmt(x).inspect)