算法练习--小球下落

小球下落：
完全二叉树，最大深度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));