UVa 101 - The Blocks Problem

算不算17第一题呢？

题目大意：搭积木。输入一个 int ，代表积木总数，积木按 0 ~ n-1 编号。四个口令表示不同的搭积木操作。

-	-
move a onto b	a，b上的积木放回原来的位置，然后a 及 a 以上的积木移动到 b 上
move a over b	a 上的积木放回原来的位置，然后a 及 a 以上的积木移动到 b 上
pile a onto b	b 上的积木放回原来的位置，然后a 及 a 以上的积木移动到 b 上
pile**a** over b	a 及 a 以上的积木移动到 b 上

然后总结一下

-	-
move	a 上的积木放回
pile	b 上的积木放回

无论执行何种操作最后 a 及 a 以上的积木都要移动到 b 上

所以就是写两个函数，一个用于将指定积木上的积木放回，另一个用于移动积木。当然还需要一个查找积木位置的函数。

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;

vector<int>block[26];
int n;

void find( int &p, int &h, int num ){
  for( p = 0; p < n; p++) 
     for( h = 0; h < block[p].size(); h++ )
        if( block[p][h] == num ) return ;
 }

void putback(int p, int h, int num) {
  for(int i = block[p].size() - 1 ; i > h; i--) {
      int a = block[p][i];
      block[a].push_back(a);
  }
  block[p].resize(h+1);
}

void puton(int pa, int ha, int pb, int hb, int a, int b ) {
  block[pb].insert( block[pb].end(), block[pa].begin()+ha, block[pa].end() );
  int size = block[pa].size();
  block[pa].erase(block[pa].begin()+ha, block[pa].end());
}

int main() {
  cin >> n;
  for( int i = 0; i < n; i++ )
     block[i].push_back(i);

  string s1, s2;
  int a, b;
  while( cin >> s1) {
    if( s1 == "quit") break;
    cin >> a >> s2 >> b;
  int pa , ha , pb , hb ;
    find(pa, ha, a);
    find(pb, hb, b);
    if(pa == pb) continue;
    if( s1 == "move") putback(pa, ha, a);
    if( s2 == "onto") putback(pb, hb, b);
    puton(pa, ha, pb, hb, a, b );
  }    
  for( int i = 0; i < n; i++ ) {
     cout << i << ":";
     for( int j = 0; j < block[i].size(); j++)
       cout << " " << block[i][j];
     cout << endl;
  }  
  return 0;
}

PS：一个小细节，一开始是每次需要查找指定编号的木块时就调用一次 find ，然后报错。后来发现每次操作其实在需要 a 和 b 的位置的参数时，它们是不变的，那么只要在最开始查找一次即可。