uva101 木块问题

木块问题，题目：https://onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=37

主要是对vector的应用。每个木块堆的数目是动态变化的，vector数组比较适用。

#include <iostream>
#include <string>
#include <vector>
using namespace std;

const int maxn = 30;
int n;
vector <int> pile[maxn];//每个pile[i]是一个vector,相当于一个二维数组

//找到一个木块所在的堆和高，以引用的形式返回两个参数
void fild_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);//把木块放回原位
	}
	pile[p].resize(h + 1);//只留下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);//保留编号为0-h-1的木块
}

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;
		fild_block(a, pa, ha);
		fild_block(b, pb, hb);
		if (pa == pb)continue;//非法指令
		if (s2 == "onto")clear_above(pb, ha);
		if (s1 == "move")clear_above(pa, ha);
		pile_onto(pa, ha, pb);
	}
	print();
	return 0;
}