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

来源：互联网发布：mac 删除键编辑：程序博客网时间：2024/05/21 05:22

题意：

看白书

要点：

学了一下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;}

0 0