例题5-2 木块问题 vector

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/weixin_42324771/article/details/87906444

题目:木块问题

不定长数组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]; 二维数组
展开阅读全文

没有更多推荐了,返回首页