例题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;
}