从左到右有n个木块,编号为0-n-1,要求模拟以下4种操作(下面a和b都是木块编号)。
- move a onto b: 把a和b上方的木块全部归位,然后把a摞到b上面。
- move a over b: 把a上方的木块全部归位,然后把a及上面的木块整体摞到b上面。
- pile a onto b: 把b上方的木块全部归位,然后把a及a上面的木块摞在b上面。
- pile a over b: 把a及上面的木块整体摞在b所在木块堆的顶部。
遇到quit 时终止一组数据。a和b在同一堆的指令是非法指令,应当忽略。
所有操作结束后,输出每个位置的木块列表,按照从底部到顶部的顺序排列。
分析:
每个木块堆的高度不确定,所以用vector来保存很合适;而木块堆的个数不超过n,所以用一个数组来存就可以了。
然后,可以把4个操作总结为:
当出现move时,会把a全部归位;当出现onto时,会把b全部归位。
代码如下:
代码语言:javascript复制#include<cstdio>
#include<string>
#include<vector>
#include<iostream>
using namespace std;
const int maxn = 30;
int n;
vector<int> pile[maxn]; //每个pile[i]是一个vector
//找木块a所在的pile和height,以引用的形式返回调用者
void find_block(int a,int& p,int& h) {
for(p=0;p<n;p ){
for(h=0;h<pile[p].size();h ){
if(pile[p][h] == a){
return;
}
}
}
}
//把第p堆高度为h的木块上方的所有木块移回原位
void clear_above(int p,int 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); //pile只应保留下标0-h的元素
}
//把第p堆高度为h及其上方的木块整体移动到p2堆的顶部
void pile_onto(int p,int h,int p2){
for(int i=h;i<pile[p].size();i ){
pile[p2].push_back(pile[p][i]);
}
pile[p].resize(h);
}
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(){
int a,b;
cin>>n;
string s1,s2;
for(int i=0;i<n;i ) pile[i].push_back(i);
while(cin>>s1>>a>>s2>>b){
int pa,pb,ha,hb;
find_block(a,pa,ha);
find_block(b,pb,hb);
if(pa == pb) continue;
if(s2 == "onto") clear_above(pb,hb);
if(s1 == "over") clear_above(pa,ha);
pile_onto(pa,ha,pb);
}
print();
return 0;
}
也可以选择调试模式,来观察下,代码如下:
代码语言:javascript复制 int a,b;
cin>>n;
string s1,s2;
for(int i=0;i<n;i ) pile[i].push_back(i);
while(cin>>s1>>a>>s2>>b){
if(s1 == "quit" || s2 == "quit"){
break;
}
int pa,pb,ha,hb;
find_block(a,pa,ha);
printf("pa:%d,ha:%d n",pa,ha);
find_block(b,pb,hb);
printf("pb:%d,hb:%d n",pb,hb);
if(pa == pb) continue;
if(s2 == "onto"){
clear_above(pb,hb);
printf("onto: n");
print();
}
if(s1 == "move"){
clear_above(pa,ha);
printf("move: n");
print();
}
pile_onto(pa,ha,pb);
printf("pile onto: n");
print();
}
print();
return 0;
题目地址