每个木块堆的高度不定,所以用vector存
1、resize(n)
调整容器的长度大小,使其能容纳n个元素。
如果n小于容器的当前的size,则删除多出来的元素。
否则,添加采用值初始化的元素。
2、 resize(n,t)
多一个参数t,将所有新添加的元素初始化为t。
#include <cstdio>
#include <string>
#include <vector>
#include <iostream>
using namespace std;
const int N = 30;
int n;
vector<int> pile[N]; // 每个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); // 把木块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); // pile只保留下标为0 ~ h - 1的元素
}
int main()
{
int a, b;
cin >> n;
string str1, str2;
for (int i = 0; i < n; i ++ ) pile[i].push_back(i);
while (cin >> str1 >> a >> str2 >> b)
{
int pa, pb, ha, hb;
find_block(a, pa, ha);
find_block(b, pb, hb);
if (pa == pb) continue; // 非法指令
if (str2 == "onto") clear_above(pb, hb);
if (str1 == "move") clear_above(pa, ha);
pile_onto(pa, ha, pb);
}
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");
}
return 0;
}