题目:木块问题
不定长数组vector的应用
算法竞赛入门经典第二版 刘汝佳 P110例题5-2
思路:
- 木块堆高度:使用不定长数组vector存储
木块堆个数(n):用一个数组存储 - 提取指令间的共同点:
含有“onto”:b上方的木块都要归位
含有“move”:a上方的木块都要归位
最后一步都是将a放在b顶部 或者 a及上面的放在b顶部(both a to b) - 根据共同点写函数:
find_block:定位要操作的木块(所在堆的编号及高度)
clear_above:上方木块归位
pile_onto:移到顶部
print:输出结果
vector的常用操作:https://blog.csdn.net/weixin_42324771/article/details/87863706
代码:
(就是教材上的喽)
#include <bits/stdc++.h>
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;//vector就像一个二维数组,第一维大小固定(不超过maxn),第二维大小不固定
}
//把第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);//把木块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=="move")clear_above(pa,ha);
pile_onto(pa,ha,pb);
}
print();
return 0;
}
ctrl+z 后回车结束输入
收获:
- 以引用的形式返回调用者
- 使用while()可以连续输入多组数据,continue忽略非法指令
- vector pile[maxn]; 二维数组