南陽OJ-63 小猴子下落(數據結構-二叉樹)

題目63
小猴子下落
時間限制：3000 ms | 內存限制：65535 KB
難度：3
描述
有一顆二叉樹，最大深度爲D,且全部葉子的深度都相同。全部結點從左到右從上到下的編號爲1,2,3，·····，2的D次方減1。在結點1處放一個小猴子，它會往下跑。每一個內結點上都有一個開關，初始所有關閉，當每次有小猴子跑到一個開關上時，它的狀態都會改變，當到達一個內結點時，若是開關關閉，小猴子往左走，不然往右走，直到走到葉子結點。

一些小猴子從結點1處開始往下跑，最後一個小猴兒會跑到哪裏呢？

輸入
輸入二叉樹葉子的深度D,和小猴子數目I，假設I不超過整棵樹的葉子個數，D<=20.最終以 0 0 結尾
輸出
輸出第I個小猴子所在的葉子編號。
樣例輸入
4 2
3 4
0 0
樣例輸出
12
7

數據結構基礎題

#include<iostream>
#include<cstring>
using namespace std;
const int N = 20;
int vis[1<<N];//標記數組，最大結點個數爲2^max-1 
int main() {
  int D, I;
  while (cin >> D >> I) {
    if (!D && !I) break;//D和I爲零結束 
    memset(vis, 0, sizeof(vis));
    int n = (1<<D)-1, k;
    for (int i = 0; i < I; i++) {//連續讓I個球下落 
      k = 1;//每次操做都從根節點開始 
      while (k <= n) {//判斷是否出界 
        vis[k] = !vis[k];
        k = vis[k] ? 2*k : 2*k+1;//根據開關狀態選擇路徑 
      }
    }
    cout << k / 2 << endl; //出界以前的葉子編號 
  }
}