小编程题之书本转移

根据 csdn 的在线编程题改编

题目：

小强有 3 个箱子 A,B,C 用来装书，所有的书(一共n本)都按序号由小到大的顺序堆在 A上，小号在底部，现在他想把所有的书全都放到 C 里面去。每次他从 A 书架拿 a 本书(不够就全拿完)到 B，然后从 B 拿 b 本书(不够就全拿完)到 C。然后重复操作，直到所有书都到了 C，求最后的C 里面书的顺序，

输入描述：

输入数据占一行，3 个数，n(1<=n<=10^5),a(1<=a<=10^9) , b(1<=b<=10^9)。

输出描述：

每组数据输出一行，输出 n 个数，C 中书的排列

首先给出代码吧：

#include<iostream>
#include<vector>
#include<fstream>
using namespace std;
static int num;

void handle(int n,int a,int b)
{	
	vector <int> A;
	vector <int> B;
	vector <int> C;
	vector <int> a_arry;
	vector <int> b_arry;
	int i;
	int j;
	int A_cnt=n;
	int C_cnt=0;
	int B_cnt=0;
	//cout<<"A_cnt "<<A_cnt<<endl;
	//cout<<"a = "<<a<<" b =  "<<b<<endl;
	for(i=0;i<A_cnt;i++)
	{
		A.push_back(n-i); //A[0]存放的是底部书编号
	}
	//for(i=0;i<n;i++)
	//{
	//	cout<<A[i]<<" ";
	//}
	//cout<<endl;
	
	<span style="color:#ff0000;">if(a<=b)</span>
	{
		<span style="color:#000099;">//cout<<"a<=b"<<endl;
		while(A_cnt!=0)
		{
			if(A_cnt>a)
			{
				for(i=0;i<a;i++)
				{
					//cout<<A[A_cnt-a+i]<<" ";
					C.push_back(A[A_cnt-a+i]); //保证A最上面的书还在C的最上面
				}
				A_cnt=A_cnt-a;
				//cout<<endl<<A_cnt<<endl;
			}
			else
			{
				for(i=0;i<A_cnt;i++)
				{
					//cout<<A[i]<<" ";
					C.push_back(A[i]);
				}
				A_cnt = 0;
			}
		}</span>
	}
	<span style="color:#ff0000;">else</span>
	{	<span style="color:#009900;">//cout<<"a>b"<<endl;
		while(C_cnt != n)
		{
			if(A_cnt >= a)
			{
				//cout<<"A_cnt > a"<<endl;
				for(i=0;i<a;i++)
				{
					B.push_back(A[A_cnt-a+i]);  //A中搬出a本书到B
				}
				//cout<<"B ";
				//for(i=0;i<B.size();i++)
				//	cout<<B[i]<<" ";
				//cout<<endl;
				A_cnt = A_cnt - a;          //相应的 A的书本计数要减少a
				B_cnt = B_cnt + a;          //相应的 B的书本计数要增加
				//cout<<"B_cnt "<<B_cnt<<endl;
				//cout<<"A_cnt "<<A_cnt<<endl;
			}
			else
			{
				//cout<<"A_cnt < a"<<endl;
				for(i=0;i<A_cnt;i++)         //A中书本剩余少于a本的情况
				{
					B.push_back(A[i]);
				}
				//cout<<"B ";
				//for(i=0;i<B.size();i++)
				//	cout<<B[i]<<" ";
				//cout<<endl;
				B_cnt = B_cnt + A_cnt;
				A_cnt = 0;
				//cout<<"B_cnt "<<B_cnt<<endl;
				//cout<<"A_cnt "<<A_cnt<<endl;
			}
		
			if(B_cnt >= b)
			{
			   // cout<<"B_cnt >= b"<<endl;
				for(i=0;i<b;i++)
				{
					C.push_back(B[B_cnt-b+i]);	//B中将b本书放入C中
				}
				//cout<<"C ";
			//	for(i=0;i<C.size();i++)
				//	cout<<C[i]<<" ";
				//cout<<endl;
				B.erase(B.end()-b,B.end());            //刷新B中的书本存放容量
				//cout<<"erase B"<<endl;
				//for(i=0;i<B.size();i++)
				//	cout<<B[i]<<" ";
				//cout<<endl;
				B_cnt = B_cnt -b;                    //相应的 减少 B的书本计数， C 中也是
				C_cnt = C_cnt + b;
				//cout<<"B_cnt "<<B_cnt<<endl;
				//cout<<"C_cnt "<<C_cnt<<endl;
			}
			else
			{	//cout<<"B_cnt < b"<<endl;
				for(i=0;i<B_cnt;i++)
				{
					C.push_back(B[i]);    //B中书本少于b本
				}
				//cout<<"C ";
				//for(i=0;i<C.size();i++)
				//	cout<<C[i]<<" ";
				//cout<<endl;
				C_cnt = C_cnt + B_cnt;
				B_cnt = 0;
				//cout<<"B_cnt "<<B_cnt<<endl;
				//cout<<"C_cnt "<<C_cnt<<endl;
			}	
			
		}
	
	}</span>
	cout<<"bottom -> top "<<endl;
	for(i=0;i<n;i++)
	{
		cout<<C[i]<<" ";
	}
	cout<<endl;
}

int main()
{
	freopen("test.txt","r",stdin);
	int n;
	int a,b;
	num = 0;
	while(cin>>n)
	{
		cin>>a>>b;
		num++;
		handle(n,a,b);
	}
	return 0;
}

test.txt

4 2 1
5 2 3

run:

----------------------------------------------------------------------------------------------------

拿书分为两步一个循环，所以A到B，B到C来分析。

对于a和b，又分大小，这是有区别的，如果 b比较大，则在B到C的时候，B中是没有剩余的，反之则有。

condition 1:

当b比较大时，实际就是直接从A中每次取 a本书放到C中去，这个问题很好解决，见代码 蓝色部分；

这里面在最后A中没有a本书，一次性搬完时，需要单独考虑一下；

condition 2：

在b较小时，这样B在搬b本书到C中之后是有剩余的。大的方向还是先从A中搬a本到B中，然后从B中在搬b本到C中。

其实这是两个相似的过程，与condition 1的情况也是一样的。需要处理的就是将B中一部分搬出后，要将B箱子刷新一下，这里采用的vector的erase函数。代码见 绿色部分。