harem.vim

scriptencoding utf-8


function! harem#now(list, n)
  return harem#_loop(a:list, [], 1, a:n)
endfunction


function! harem#_loop(list, result, count, n)
  if len(a:list) > 0
    let result = harem#_loop(a:list[1:], a:result + ['.'], a:count, a:n)
    if a:count <= a:n
      let result += harem#_loop(a:list[1:], a:result + [a:list[0]], a:count + 1, a:n)
    endif
    return result
  else
    return [a:result]
  endif
endfunction


" vim-prettyprint (by thinca)
echo PrettyPrint(harem#now(split('ABCDE', '\zs'), 2))
" =>
"[
"  ['.', '.', '.', '.', '.'],
"  ['.', '.', '.', '.', 'E'],
"  ['.', '.', '.', 'D', '.'],
"  ['.', '.', '.', 'D', 'E'],
"  ['.', '.', 'C', '.', '.'],
"  ['.', '.', 'C', '.', 'E'],
"  ['.', '.', 'C', 'D', '.'],
"  ['.', 'B', '.', '.', '.'],
"  ['.', 'B', '.', '.', 'E'],
"  ['.', 'B', '.', 'D', '.'],
"  ['.', 'B', 'C', '.', '.'],
"  ['A', '.', '.', '.', '.'],
"  ['A', '.', '.', '.', 'E'],
"  ['A', '.', '.', 'D', '.'],
"  ['A', '.', 'C', '.', '.'],
"  ['A', 'B', '.', '.', '.']
"]


echo len(harem#now(split('ABCDE', '\zs'), 2))
" => 16
" C(5,2)+C(5,1)+C(5,0)=10+5+1=16 ... Ok