UVa101 The Blocks Problem

VJ链接 （紫书110）

vector应用
题目不难，好好分析题意就能写出来。虽然是照着lrj的板子抄的…

1. 函数可以使用指针来改变参数变量。
   eg：Find(int a, int &p, int &h)在函数中即可改变p和h的量
2. vector的用法
   vector.resize(n)改变容量的大小为n，如果n<size，删除多余元素
   vector.resize(n, t)把新加入的元素初始化为t

#include<iostream>
#include<vector>
using namespace std;
const int maxn=50;
vector<int>pile[maxn];
int n;

void Find(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;
        }
    }
}

void Clear(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);
}

void Move(int p1,int h1,int p2){
    for(int i=h1;i<pile[p1].size();i++){
        pile[p2].push_back(pile[p1][i]);
    }
    pile[p1].resize(h1);
}

int main()
{

    cin>>n;
    for(int i=0;i<n;i++)pile[i].push_back(i);
    int a,b;
    string s1,s2;
    while(cin>>s1){
        if(s1[0]=='q')break;
        cin>>a>>s2>>b;
        //find木块位置
        int pa,pb,ha,hb;
        Find(a,pa,ha);
        Find(b,pb,hb);
        if(pa==pb)continue;
        if(s2=="onto"){//b上木块归位
            Clear(pb,hb);
        }
        if(s1=="move"){//a上木块归位
            Clear(pa,ha);
        }
        Move(pa,ha,pb);

    }
    for(int i=0;i<n;i++){
            cout<<i<<":";
            for(int j=0;j<pile[i].size();j++){
                cout<<" "<<pile[i][j];
            }
            cout<<endl;
        }
}