1. 程式人生 > 其它 >演算法競賽入門經典P110(vector)

演算法競賽入門經典P110(vector)

//find_block
//找到a所在堆p和高度h
//clear_above
//將p堆h高度上方所有方塊歸位
//pile_onto
//將p堆h高度及上方所有方塊移至p2
//print
//列印
#include<iostream>
#include<cstdio>
#include<vector>
#include<string>
using namespace std;

const int maxn = 30;
int n;
vector<int>pile[maxn];

void find_block(int a,int& p,int& h){
    
for(p=0;p<n;p++){ for(h=0;h<pile[p].size();h++){ if(a==pile[p][h])return; } } } void clear_above(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 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(
"\n"); } } int main(){ int a,b; string s1,s2; cin>>n; for(int i=0;i<n;i++){ pile[i].push_back(i); } while(cin>>s1>>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; }