例题5-2 UVa101 The Blocks Problem(STL：vector)

题意：

看白书

要点：

学了一下vector中的resize()的用法，就是把resize(n)，n及以上的都丢掉。这道题的归位的意思是放回对应的第i个堆。

#include<iostream>
#include<vector>
#include<algorithm>
#include<string>
using namespace std;
int n;
vector<int> pile[30];

void FindPile(int a, int &p, int &h)
{
	for(int i=0;i<n;i++)
		for(int j=0;j<pile[i].size();j++)
			if (pile[i][j] == a)
			{
				p = i;
				h = j;
				return;
			}
}
void ClearPile(int p, int h)//注意这里要放回原来的位置，也就是b放回堆b
{
	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);
}
void MovePile(int pa, int ha, int pb)
{
	for (int i = ha; i < pile[pa].size(); i++)
		pile[pb].push_back(pile[pa][i]);
	pile[pa].resize(ha);
}
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;
	string s1, s2;
	cin >> n;
	for (int i = 0; i < n; i++)
		pile[i].push_back(i);
	while (1)
	{
		cin >> s1;
		if (s1 == "quit")
			break;
		cin >> a >> s2 >> b;
		int pa, pb, ha, hb;
		FindPile(a, pa, ha);
		FindPile(b, pb, hb);
		if (pa == pb)
			continue;//如果在同一个堆则忽略
		if (s1 == "move")
			ClearPile(pa, ha);//如果第一个字符串是move就清空a上方的木块
		if (s2 == "onto")
			ClearPile(pb, hb);//如果第二个字符串是onto就清空b上方的木块
		MovePile(pa, ha, pb);
	}
	print();
	return 0;
}