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