POJ 1208 The Blocks Problem --vector

http://poj.org/problem?id=1208

晚点仔细看

https://blog.csdn.net/yxz8102/article/details/53098575

 #include<stdio.h>
 #include<string.h>
 #include<math.h>
 #include<iostream>
 #include<stdlib.h>
 #include<algorithm>
 #include<queue>
 #include<vector>
 #include<string>
 #define mem(a) memset(a,0,sizeof(a))
 using namespace std;
 const long long  N=1e9+;
 #define LL long long
 vector<int> pile[];
 int n;

 //查找木块a所在的pile和height
 void find_block(int a,int &p,int &h) //引用，相当于直接用实参
 {
     for(p=;p<n;p++)
     {
         for(h=;h<pile[p].size();h++)
         {
             if(pile[p][h]==a)  return;
 //vector就像一个二维数组，只是第一维大小固定（30），二维大小不固定。
         }
     }
 }
 //把p堆高度为h的木块上方所有的木块移回原位置
 void clear_above(int p,int h)
 {
     for(int i=h+;i<pile[p].size();i++)
     {
         int t=pile[p][i];
         pile[t].push_back(t); ////把标号为t的木块放回原位，即t堆
     }
   pile[p].resize(h+); //重新定义p堆的长度
 }
  //把p堆高度为h及其上方的木块整体移到p2的得尾部
 void pile_onto(int p,int h,int p2)
 {
     for(int i=h;i<pile[p].size();i++)
         pile[p2].push_back(pile[p][i]); //??
        pile[p].resize(h);
 }
 void print()
 {
     for(int i=;i<n;i++)
     {
     printf("%d:",i);
     for(int j=;j<pile[i].size();j++)
     {
         printf(" %d",pile[i][j]);
     }
     printf("\n");
     }
 }
 int main()
 {
    int a,b;
    scanf("%d",&n);
    string s1,s2;
    for(int i=;i<n;i++)
    {
        pile[i].push_back(i);
    }
    while(cin>>s1)
    {
        if (s1=="quit") break;
        cin>>a>>s2>>b;
        int pa,pb,ha,hb;
        find_block(a,pa,ha);
        find_block(b,pb,hb);
        if(pa==pb) continue;
        if(s2=="onto") clear_above(pb,hb);
        if(s1=="move") clear_above(pa,ha);
        pile_onto(pa,ha,pb);
    }
    print();
     return ;
 }