uva679 Dropping Balls

球从根节点往下落,初始化所有开关为0,0球向左边,1球向右边落。

输入树的层数和球的个数,问最后一个球落在那个叶子节点上?


思路:由于每层都是满的,可以用一个一维数组建树模拟,每个球的过程。

#include <fstream>
#include <iostream>
#include <string>
#include <cstring>
#include <complex>
#include <math.h>
#include <set>
#include <vector>
#include <map>
#include <queue>
#include <stdio.h>
#include <stack>
#include <algorithm>
#include <list>
#include <ctime>
#include <ctime>
#include <assert.h>

#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define eps 1e-8
#define M_PI 3.141592653589793

typedef long long ll;
const ll mod=1000000007;
const int inf=0x7fffffff;
ll powmod(ll a,ll b) {ll res=1;a%=mod;for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;}
using namespace std;
bool tree[1<<22];
int main()
{
    int T;
    while(scanf("%d",&T)&&T>0){
      int n,t,cur;
      while(T--){
      scanf("%d%d",&n,&t);
      memset(tree,false,sizeof(tree));
      int sum=(1<<n)-1;
          //cout<<sum<<endl;
       while(t--){
             cur=1;
        while(cur*2<=sum){
            int p=cur;
                //cout<<tree[cur]<<" oo "<<endl;
            if(!tree[cur]) cur=cur*2;
            else cur=cur*2+1;
            tree[p]=!tree[p];
            }
        }
        printf("%d\n",cur);
    }
   }
}

TLE


分析,对于每个小球只需知道他是第几个落入这个点的(奇还是偶),就知道他要往左还是右,所以只用考虑最后一个点,对于每层的情况。

#include<iostream>
using namespace std;
int main()
{
    int T;
    while(cin>>T&&T!=-1){
       while(T--){
        int n,m;cin>>n>>m;
        int t=1;
        n--;
        while(n--){
            if(m%2){
                t=t*2;
                m=m/2+1;
            }
            else{
                t=t*2+1;
                m/=2;
            }
        }
        cout<<t<<endl;
      }
    }
}


你可能感兴趣的:(uva679 Dropping Balls)