紫书例题5-2 木块问题

Uva 101 繁琐的模拟题，有时候对于重复的动作，我们可以编写一个函数去实现他.

1. int &a,int &b的妙用，查找各种木块的编号.

void find(int a,int &p,int &h){
	for(p=0;p<n;p++){
		for(h=0;h<q[p].size();h++){
			if(q[p][h]==a) return ;
		}
	}
}

传参的时候就直接

	int pa,pb,ha,hb;
find(a,pa,ha);find(b,pb,hb);

2.vector的删除 用resize()函数，如果保存0-h 就resize（h+1），0——h-1，就resize（h）总之是加1.

3.完整代码

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#define maxn 300
using namespace std;
int n;vector<int> q[maxn];string s1,s2;
void find(int a,int &p,int &h){
	for(p=0;p<n;p++){
		for(h=0;h<q[p].size();h++){
			if(q[p][h]==a) return ;
		}
	}
}
void clea_r(int p,int h){ int temp,i,j;
	for(i=h+1;i<q[p].size();i++){ temp=q[p][i];
	q[temp].push_back(temp);
	}
	q[p].resize(h+1);
}
void mo_ve(int pa,int ha,int pb,int hb){ int temp,i,j;
	for(i=ha;i<q[pa].size();i++){
		temp=q[pa][i]; q[pb].push_back(temp);
	}
	q[pa].resize(ha);
}
void print(){ int  j;
	for(int i=0;i<n;i++){
		printf("%d:",i);
		for(j=0;j<q[i].size();j++) printf(" %d",q[i][j]); cout<<endl;
	}
}
int main(){ int i,j,k;cin>>n; int a,b;
for(i=0;i<n;i++) q[i].push_back(i);
while(cin>>s1&&(s1!="quit")&&cin>>a>>s2>>b){
	int pa,pb,ha,hb;
find(a,pa,ha);find(b,pb,hb);
if(pa==pb) continue;
if(s2=="onto") clea_r(pb,hb);
if(s1=="move") clea_r(pa,ha);
mo_ve(pa,ha,pb,hb);
} print();
return 0;}