木块问题(Th Blocks Problem,UVa 101)

通过这道题学到了把指令的共同点提取出来，编写函数来减少重复代码，以前我都是按整句话来处理的，学到了。

 

最近一直在刷UvaOJ，只不过前几章的题太简单，没有往上写博客…

 

 

 

看了答案后默写代码都会出错Orz

 

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector> 
using namespace std;
const int maxn = 30 ;
int n ;
vector<int> pile[maxn];

void found_pile( int a , int &p , int &h) // 以引用形式返回方块a的位置 
{
    for( p  = 0 ; p < n; p++)
        for( h = 0 ; h < pile[p].size() ; h++)
            if( pile[p][h] == a )
                return ;
}

void clear_above( int p , int h ) // 将p堆高度为h的木块上方的所有木块移回原位
{
    for( int i = h + 1 ; i < pile[p].size() ; i++)
    {
        int b = pile[p][i];
        pile[b].push_back(b);
    }
    pile[p].resize(h+1);
    return ;
}
 
 void pile_onto( int p , int h , int p2) // 把第p堆高度为h及其上方的木块整体移动到p2堆的顶部
 {
     for( int i = h ; i < pile[p].size() ; i++)
         pile[p2].push_back(pile[p][i]);
     pile[p].resize(h);
     return ;
} 
  
void print()
{
    for( int i = 0 ; i < n ; i++)
    {
        printf("%d:",i);
        for( int j = 0 ; j < pile[i].size() ; j++)
            printf(" %d",pile[i][j]);
        printf("\n");
    }
}
  

int main()
{
    string s1,s2;
    int a , b ;
    cin >> n ;
    for( int i = 0 ; i < n ; i++)
        pile[i].push_back(i);
    while( cin >> s1 >> a >> s2 >> b )
    {
        int pa , pb , ha , hb;
        found_pile(a,pa,ha);
        found_pile(b,pb,hb);
        if( pa == pb ) 
            continue ;
        if( s2 == "onto" )
            clear_above(pb,hb);
        if( s1 == "move" )
            clear_above(pa,ha);
        pile_onto(pa,ha,pb);
    }
    print();
    return 0 ; 
}