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+5;
#define LL long long
vector<int> pile[30];
int n;

//查找木块a所在的pile和height
void find_block(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;
//vector就像一个二维数组，只是第一维大小固定（30），二维大小不固定。
        }
    }
}
//把p堆高度为h的木块上方所有的木块移回原位置
void clear_above(int p,int h)
{
    for(int i=h+1;i<pile[p].size();i++)
    {
        int t=pile[p][i];
        pile[t].push_back(t); ////把标号为t的木块放回原位，即t堆
    }
  pile[p].resize(h+1); //重新定义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=0;i<n;i++)
    {
    printf("%d:",i);
    for(int j=0;j<pile[i].size();j++)
    {
        printf(" %d",pile[i][j]);
    }
    printf("
");
    }
}
int main()
{
   int a,b;
   scanf("%d",&n);
   string s1,s2;
   for(int i=0;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 0;
}