UVa712 S-Trees

// UVa712 S-Trees
// Rujia Liu
// 題意：給一棵滿二叉樹，每一層代表一個01變量，取0時往左走，取1時往右走。給出所有葉子的值，以及一些查詢（即每個變量的值），求最後到達的葉子的值
// 算法：結點從上到下編号為1, 2, 3, ...則左走就是乘以2，右走是乘以2加1。第一個葉子的編号是2^n
#include<iostream>
#include<string>
using namespace std;

const int maxn = 10;
int n, v[maxn];//映射表
string leaves;

int solve(const string& q) {
  int u = 1;
  for(int i = 0; i < n; i++) {
    if(q[v[i]] == '0') u *= 2; else u = u*2+1;
  }
  return leaves[u-(1<<n)] - '0';
}

int main() {
  int kase = 0;
  while(cin >> n && n) {
    string s;
    cout << "S-Tree #" << ++kase << ":\n";
    for(int i = 0; i < n; i++) { cin >> s; v[i] = s[1] - '1'; }
    int m;
    cin >> leaves >> m;
    while(m--) {
      string q;
      cin >> q;
      cout << solve(q);
    }
    cout << "\n\n";
  }
  return 0;
}      

我的解答：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
//0 left, 1 right

const int N=10;
char leaves[1<<N];
//順序映射表
int order[N];
int n;

void solve(char* buf)
{
        //最終葉節點的編号(從1開始)
        int j=1;
        for(int i=0;i<n;i++)
        {
                if(buf[order[i]]=='0')
                        j*=2;
                else
                        j=j*2+1;
        }
        printf("%c", leaves[j-(1<<n)]);
}

void getorder(char* s)
{
        for(int i=0;i<n;i++)
        {
                order[i]=s[1+3*i]-'1';
        }
}

int main()
{
        //freopen("./uva712.in", "r", stdin);
        char buf[N*3+1];
        int cnt=0;
        while(scanf("%d", &n)==1 && n)
        {
                printf("S-Tree #%d:\n", ++cnt);
                gets(buf);//read line

                gets(buf);//x1 x2 ...
                getorder(buf);

                gets(leaves);
                int m;
                scanf("%d", &m); gets(buf);//read line
                for(int i=0;i<m;i++)
                {
                        gets(buf);
                        solve(buf);
                }
                printf("\n\n");
        }


    return 0;
}