例题5-2 木块问题 vector

题目：木块问题

不定长数组vector的应用
算法竞赛入门经典第二版 刘汝佳 P110例题5-2

思路:

1. 木块堆高度：使用不定长数组vector存储
   木块堆个数(n)：用一个数组存储
2. 提取指令间的共同点：
   含有“onto”：b上方的木块都要归位
   含有“move”：a上方的木块都要归位
   最后一步都是将a放在b顶部 或者 a及上面的放在b顶部（both a to b）
3. 根据共同点写函数：
   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 后回车结束输入

收获：

1. 以引用的形式返回调用者
2. 使用while()可以连续输入多组数据，continue忽略非法指令
3. vector pile[maxn]; 二维数组