算法练习--小球下落

小球下落： 
完全二叉树，最大深度D。所有节点从上到下从左到右编号为1,2,3...2的D次方-1.在节点1处放一个小球，它会往下落。每个内节点上都由一个开关，初始全部关闭，当每次有小球落到一个开关上时，它的状态都会改变。当小球达到一个内节点时，如果该节点上的开关关闭，往左走，否则往右走，直到遍历到叶子节点。
问题，输入 深度和小球个数
输出节点编号

输入 4 2  输出 12
输入 3 4  输出 7

参考实现

function f(depth,count){

var states = new Array();
var r;

for(var i = 0;i < count;i++){
for(var root = 1;root <Math.pow(2,depth-1);){
if(states[root] && states[root].open){
states[root].open = false;

root = root * 2 + 1;
}
else{
if(!states[root]){states[root]={open:true};}
else if(states[root].open == false){states[root].open = true;}

root = root * 2;

}

r=root;
}

}

return r;
}

console.log(f(4,2));
console.log(f(3,4));
console.log(f(10,1));
console.log(f(2,2));
console.log(f(8,128));
console.log(f(16,12345));