Uva - 101 - The Blocks Problem

由于每个木块堆的高度不确定，所以用vector保存

AC代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cctype>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <algorithm>

using namespace std;

const int maxn = 30;
int n;
vector<int> pile[maxn];

// 找木块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);
}

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()
{
	// 对流进行加速
	//ios::sync_with_stdio(false);
	int a, b;
	cin >> n;
	string s1, s2;
	for (int i = 0; i < n; i++) {
		pile[i].push_back(i);
	}
	while (cin >> s1 && s1 != "quit" &&cin >> 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;
}