根据 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函数。代码见 绿色部分。